Home
last modified time | relevance | path

Searched refs:SetStreamId (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc146 AnfAlgo::SetStreamId(independent_switch_stream_id, independent_switch_app.get()); in InsertIndepentParallel()
166 AnfAlgo::SetStreamId(*fpbp_switch_stream_id, fpbp_switch_app.get()); in InsertFpBpLoopStreamSwitch()
204 AnfAlgo::SetStreamId(fpbp_stream_id, eos_done_recv.get()); in InsertEosDoneRecv()
228 AnfAlgo::SetStreamId(fpbp_stream_id, fpbp_start_recv.get()); in InsertFpBpStartRecv()
241 AnfAlgo::SetStreamId(fpbp_stream_id, assign_add_one.get()); in InsertNextLoopAssignAdd()
304 AnfAlgo::SetStreamId(*getnext_switch_stream_id, getnext_switch_app.get()); in InsertGetNextLoopStreamSwitch()
320 AnfAlgo::SetStreamId(getnext_stream_id, (*exec_order)[(*exec_order).size() - 1].get()); in SetBeforeGetNextStreamID()
337 AnfAlgo::SetStreamId(getnext_stream_id, fpbp_start_send.get()); in InsertGetNextLoopFpBpStartSend()
351 AnfAlgo::SetStreamId(getnext_stream_id, eos_start_send.get()); in InsertGetNextLoopEosStartSend()
369 AnfAlgo::SetStreamId(*eos_switch_stream_id, eos_switch_app.get()); in InsertEosStreamSwitch()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_stream_assign.cc652 AnfAlgo::SetStreamId(cur_common_stream_id, cur_cnode_ptr.get()); in AssignCommonStreamId()
656 AnfAlgo::SetStreamId(it->first, cur_cnode_ptr.get()); in AssignCommonStreamId()
660 AnfAlgo::SetStreamId(cur_common_stream_id, cur_cnode_ptr.get()); in AssignCommonStreamId()
731 AnfAlgo::SetStreamId(cur_hcom_stream_id, cur_cnode_ptr.get()); in AssignHcomStreamId()
735 AnfAlgo::SetStreamId(it->first, cur_cnode_ptr.get()); in AssignHcomStreamId()
739 AnfAlgo::SetStreamId(cur_hcom_stream_id, cur_cnode_ptr.get()); in AssignHcomStreamId()
791 AnfAlgo::SetStreamId(cur_independent_stream_id, cur_cnode_ptr.get()); in AssignIndependentStreamId()
795 AnfAlgo::SetStreamId(it->first, cur_cnode_ptr.get()); in AssignIndependentStreamId()
799 AnfAlgo::SetStreamId(cur_independent_stream_id, cur_cnode_ptr.get()); in AssignIndependentStreamId()
816 AnfAlgo::SetStreamId(AnfAlgo::GetStreamId(cur_cnode_ptr), cnode_ptr_list[i - 1].get()); in UpdateAtomicAddrCleanStreamId()
[all …]
Dascend_kernel_runtime.cc755 AnfAlgo::SetStreamId(id, node.get()); in SetKernelModStream()
760 AnfAlgo::SetStreamId(id, node.get()); in SetKernelModStream()
765 AnfAlgo::SetStreamId(1, node.get()); in SetKernelModStream()
769 AnfAlgo::SetStreamId(0, node.get()); in SetKernelModStream()
776 AnfAlgo::SetStreamId(stream_id, kernels[i - 1].get()); in SetKernelModStream()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc787 TEST_F(AnfRuntimeAlgorithmTest, SetStreamId) { in TEST_F() argument
792 AnfAlgo::SetStreamId(0, add.get()); in TEST_F()
793 EXPECT_THROW(AnfAlgo::SetStreamId(0, nullptr), std::runtime_error); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.h250 static void SetStreamId(uint32_t stream_id, AnfNode *node);
Danf_runtime_algorithm.cc1491 void AnfRuntimeAlgorithm::SetStreamId(uint32_t stream_id, AnfNode *node) { in SetStreamId() function in mindspore::session::AnfRuntimeAlgorithm
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/
Dprofiling_utils.cc313 AnfAlgo::SetStreamId(AnfAlgo::GetStreamId(anf_node), profiling_node.get()); in CreateProfilingCNodeWithStream()