Searched refs:root_id_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel.cc | 68 …: hccl_count_(0), op_type_(::HcclReduceOp::HCCL_REDUCE_SUM), root_id_(0), src_rank_(0), dest_rank_… in HcclKernel()
76 root_id_ = 0; in ~HcclKernel()
127 if (!HcomUtil::GetHcomRootId(anf_node, &root_id_)) { in Init()
245 …) << "HCCL Task : stream_id=" << stream_id << ", count=" << hccl_count_ << ", root_id=" << root_id_ in GenTask()
269 … hccl_count_, root_id_, op_type_, data_type, group_, NeedDump())); in GenTask()
296 …hccl_type, input_data_addr, output_data_addr, hccl_count_, data_type, op_type_, root_id_, stream_p… in GenDynamicKernel()
|
| D | hcom_all_to_all.cc | 84 … "AlltoAll Task : stream_id=" << stream_id << ", count=" << hccl_count_ << ", root_id=" << root_id_ in GenTask()
107 …hccl::HcclAdapter::GetInstance().GetHcclOpsKernelInfoStore(), hccl_count_, root_id_, op_type_, dat… in GenTask()
|
| D | hcom_all_broadcast.cc | 33 … hccl_data_type_list_[0], root_id_, stream_ptr); in Launch()
|
| D | hccl_kernel.h | 53 uint32_t root_id_; variable
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 245 root_id_(root_id), in HcclTaskInfo()
260 int64_t root_id() const { return root_id_; } in root_id()
275 int64_t root_id_; variable
|