Home
last modified time | relevance | path

Searched refs:IsCommunicationOp (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_tensor_move_for_cascade.cc38 if (!AnfAlgo::IsCommunicationOp(prev_node)) { in IsPartOutputsOfHcclOp()
116 if (!AnfAlgo::IsCommunicationOp(node)) { in Process()
Dinsert_tensor_move_for_hccl_op.cc60 if (AnfAlgo::IsRealKernel(node) && !AnfAlgo::IsCommunicationOp(node)) { in IsNodeOutPutUsedByOtherRealKernel()
154 if (!AnfAlgo::IsCommunicationOp(node)) { in Process()
Dgetnext_tensor_move_elimination.cc71 if (AnfAlgo::IsCommunicationOp(next_node)) { in Process()
Dsplit_n_optimizer.cc149 if (InvalidOps.find(op_name) != InvalidOps.end() || AnfAlgo::IsCommunicationOp(item)) { in OutputCheck()
/third_party/mindspore/mindspore/ccsrc/profiler/device/gpu/
Dgpu_profiling_utils.cc74 if (AnfAlgo::IsCommunicationOp(node)) { in GetTraceHccl()
123 if (AnfAlgo::IsCommunicationOp(*iter)) { in SetTraceBpEnd()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse.cc51 bool is_comm_op = AnfAlgo::IsCommunicationOp(kernel_cnode); in InitDynamicOutputKernelRef()
185 bool is_comm_op = AnfAlgo::IsCommunicationOp(kernel); in SetInputMap()
299 if (AnfAlgo::IsCommunicationOp(kernel)) { in SetKernelDefMap()
Dmem_swap_manager.cc178 if (AnfAlgo::IsCommunicationOp(kernel)) { in IsCommunicationRelevantOp()
192 …[](const std::pair<AnfNodePtr, int> &node_pair) { return AnfAlgo::IsCommunicationOp(node_pair.firs… in IsCommunicationRelevantOp()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/
Dprofiling_utils.cc87 if (AnfAlgo::IsCommunicationOp(node)) { in GetTraceHccl()
170 if (AnfAlgo::IsCommunicationOp(*iter)) { in GetTraceBpEnd()
210 AnfAlgo::IsCommunicationOp(*iter)) { in GetGraphLastKernelName()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc362 [](const AnfNodePtr &kernel) { return AnfAlgo::IsCommunicationOp(kernel); }); in IsDistributedTraining()
377 if (AnfAlgo::IsCommunicationOp(kernel)) { in FetchMemUnitSize()
1141 if (AnfAlgo::IsCommunicationOp(kernel)) { in AllocKernelOutputDynamicRes()
1195 if (AnfAlgo::IsCommunicationOp(kernel) && AnfAlgo::GetCNodeName(kernel) != kHcomSendOpName && in AllocCommunicationOpDynamicRes()
1276 if (AnfAlgo::IsCommunicationOp(kernel)) { in FreeKernelDynamicRes()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc145 if (!AnfAlgo::IsCommunicationOp(node)) { in RunOpAssignCommunicationInput()
199 if (!AnfAlgo::IsCommunicationOp(node)) { in RunOpAssignCommunicationOutput()
643 if (AnfAlgo::IsCommunicationOp(kernel_with_index.first)) { in AssignStaticMemoryOutput()
705 if (!AnfAlgo::IsCommunicationOp(kernel)) { in GenKernelEvents()
725 if (AnfAlgo::IsCommunicationOp(child)) { in GenKernelEvents()
1088 if (AnfAlgo::IsCommunicationOp(node)) { in AssignDynamicMemory()
1423 if (AnfAlgo::IsCommunicationOp(kernel)) { in LaunchKernel()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc298 if (mindspore::session::AnfRuntimeAlgorithm::IsCommunicationOp(kernel)) { in GetCommunicationOpInputInfo()
308 if (AnfAlgo::IsCommunicationOp(cnode) || AnfAlgo::IsIndependentNode(cnode) || in GetCommunicationOpInputInfo()
Dascend_kernel_runtime.cc743 if (AnfAlgo::IsCommunicationOp(node)) { in SetKernelModStream()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.h270 static bool IsCommunicationOp(const AnfNodePtr &node);
Dgpu_session.cc601 if (node->isa<CNode>() && !AnfAlgo::IsCommunicationOp(node) && !ps_mode) { in UpdateOutputTensors()
Danf_runtime_algorithm.cc1648 bool AnfRuntimeAlgorithm::IsCommunicationOp(const AnfNodePtr &node) { in IsCommunicationOp() function in mindspore::session::AnfRuntimeAlgorithm
1661 if (!IsCommunicationOp(node)) { in IsFusedCommunicationOp()
Dkernel_graph.cc196 bool is_comm_node = AnfAlgo::IsCommunicationOp(next_node); in EnqueueActiveNodes()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_scheduler.cc446 if (AnfAlgo::IsCommunicationOp(kernel)) { in Link()
679 if (!AnfAlgo::IsCommunicationOp(kernel)) { in BuildDataPrepareActor()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc513 bool IsCommunicationOp(const PrimitivePtr &prim) { in IsCommunicationOp() function
533 if (IsCommunicationOp(prim) && cnode->in_forward_flag()) { in FindCommunicationOp()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/
Dsomas.cc409 if (AnfAlgo::IsCommunicationOp(kernel)) { in InitSomasStreamAndNode()