Searched defs:user_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()
60 …StreamController::QueryTaskIdOnStream(const DeviceContext *device_context, uint32_t user_stream_id, in QueryTaskIdOnStream()
90 uint32_t user_stream_id, in RecordEvent()
108 uint32_t user_stream_id, uint32_t memory_stream_id) { in WaitEvent()
123 uint32_t user_stream_id) { in WaitEvent()
133 …amController::DispatchRecordWaitEvent(const DeviceContext *device_context, uint32_t user_stream_id, in DispatchRecordWaitEvent()
198 int64_t TaskIdOnStreamManager::Query(uint32_t user_stream_id, uint32_t memory_stream_id) { in Query()
203 bool TaskIdOnStreamManager::Update(int64_t task_id_on_stream, uint32_t user_stream_id, uint32_t mem… in Update()
|
| D | memory_manager.h | 84 bool RecordEvent(int64_t task_id_on_stream, uint32_t user_stream_id, in RecordEvent()
93 bool WaitEvent(int64_t task_id_on_stream, uint32_t user_stream_id, uint32_t memory_stream_id) { in WaitEvent()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/ |
| D | mem_dynamic_allocator.cc | 1093 bool DynamicMemPoolBestFit::RecordEvent(int64_t task_id_on_stream, uint32_t user_stream_id, in RecordEvent()
1120 bool DynamicMemPoolBestFit::WaitEvent(int64_t task_id_on_stream, uint32_t user_stream_id, uint32_t … in WaitEvent()
1299 bool DynamicMemBuf::RecordEvent(int64_t task_id_on_stream, uint32_t user_stream_id, const DeviceEve… in RecordEvent()
1317 bool DynamicMemBuf::WaitEvent(uint32_t task_id_on_stream, uint32_t user_stream_id) { in WaitEvent()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | kernel_actor.cc | 1126 auto user_stream_id = kernel_mod_->record_stream_id(); in ProcessMultiStreamBeforeKernelLaunch() local
1185 auto user_stream_id = stream_id; in ProcessMultiStreamBeforeKernelLaunch() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ccsrc/backend/common/ |
| D | mem_dynamic_allocator_test.cc | 279 uint32_t user_stream_id = 1; in TEST_F() local
|