Home
last modified time | relevance | path

Searched refs:get_cur_stream_num (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_stream_assign.cc614 auto common_stream_num = resource_manager.get_cur_stream_num(); in AssignAllNodesStream()
619 auto hcom_stream_num = resource_manager.get_cur_stream_num() - common_stream_num; in AssignAllNodesStream()
624 …auto independent_stream_num = resource_manager.get_cur_stream_num() - common_stream_num - hcom_str… in AssignAllNodesStream()
626 …resource_manager.get_cur_stream_num() + Uint32tMulWithOverflowCheck(hcom_stream_num, kHcomSecondar… in AssignAllNodesStream()
636 …MS_LOG(INFO) << "After stream assign, total stream nums:" << resource_manager.get_cur_stream_num(); in AssignAllNodesStream()
643 uint32_t cur_stream_num = resource_manager.get_cur_stream_num(); in AssignCommonStreamId()
1867 uint32_t assigned_stream_num = resource_manager.get_cur_stream_num(); in CheckStreamAssign()
2031 uint32_t total_stream_num = resource_manager.get_cur_stream_num(); in GetWaitStreams()
2141 auto stream_num = resource_manager.get_cur_stream_num(); in FindStreamRelations()
Dascend_stream_assign.h87 uint32_t get_cur_stream_num() { return cur_stream_num_; } in get_cur_stream_num() function
Dascend_kernel_runtime.cc497 MS_LOG(INFO) << "Call DavinciModel total stream num:" << resource_manager.get_cur_stream_num() in GenTask()
504 …resource_manager.get_cur_stream_num(), graph.label_num(), resource_manager.get_cur_event_num(), 0); in GenTask()