Searched defs:memory_stream_id (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | multi_stream_controller.cc | 56 … uint32_t user_stream_id, uint32_t memory_stream_id) { in UpdateTaskIdOnStream()
61 uint32_t memory_stream_id) { in QueryTaskIdOnStream()
108 uint32_t user_stream_id, uint32_t memory_stream_id) { in WaitEvent()
134 uint32_t memory_stream_id) { in DispatchRecordWaitEvent()
198 int64_t TaskIdOnStreamManager::Query(uint32_t user_stream_id, uint32_t memory_stream_id) { in Query()
203 …eamManager::Update(int64_t task_id_on_stream, uint32_t user_stream_id, uint32_t memory_stream_id) { in Update()
|
| D | memory_manager.h | 93 bool WaitEvent(int64_t task_id_on_stream, uint32_t user_stream_id, uint32_t memory_stream_id) { in WaitEvent()
100 bool WaitEvent(int64_t task_id_on_stream, uint32_t memory_stream_id) { in WaitEvent()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | kernel_actor.cc | 1127 auto memory_stream_id = stream_id; in ProcessMultiStreamBeforeKernelLaunch() local
1186 auto memory_stream_id = cross_stream_kernel_tensor->stream_id(); in ProcessMultiStreamBeforeKernelLaunch() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ccsrc/backend/common/ |
| D | mem_dynamic_allocator_test.cc | 285 uint32_t memory_stream_id = 0; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/ |
| D | mem_dynamic_allocator.cc | 1120 …BestFit::WaitEvent(int64_t task_id_on_stream, uint32_t user_stream_id, uint32_t memory_stream_id) { in WaitEvent()
1149 bool DynamicMemPoolBestFit::WaitEvent(int64_t task_id_on_stream, uint32_t memory_stream_id) { in WaitEvent()
|