Searched refs:private_def (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_all_to_all.cc | 75 std::vector<uint8_t> private_def; in GenTask() local
87 << ", private_def_size=" << task.private_def.size(); in GenTask()
89 private_def.resize(task.private_def.size()); in GenTask()
90 …auto sec_ret = memcpy_s(private_def.data(), private_def.size(), task.private_def.data(), task.priv… in GenTask()
106 workspace_addr, task.workspace_size, task.stream_num, private_def, in GenTask()
|
| D | hccl_kernel.cc | 232 std::vector<uint8_t> private_def; in GenTask() local
248 << ", private_def_size=" << task.private_def.size(); in GenTask()
250 private_def.resize(task.private_def.size()); in GenTask()
251 …auto sec_ret = memcpy_s(private_def.data(), private_def.size(), task.private_def.data(), task.priv… in GenTask()
268 … private_def, hccl::HcclAdapter::GetInstance().GetHcclOpsKernelInfoStore(), in GenTask()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | hccl_task.cc | 60 …char *private_def = reinterpret_cast<char *>(const_cast<char unsigned *>(task_info_->private_def()… in Distribute() local
61 auto private_def_len = static_cast<uint32_t>(task_info_->private_def().size()); in Distribute()
62 MS_LOG(INFO) << "The first address of the custom info, privateDef= " << private_def; in Distribute()
91 ge_task.privateDef = private_def; in Distribute()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 233 … const std::vector<uint8_t> &private_def, void *ops_kernel_store, int32_t count, int64_t root_id, in HcclTaskInfo() argument
242 private_def_(private_def), in HcclTaskInfo()
257 const std::vector<uint8_t> &private_def() const { return private_def_; } in private_def() function
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | hccl_adapter.h | 37 std::string private_def; member
|
| D | converter.cc | 231 return {task_def.private_def(), workspace_size, stream_num}; in ParseDomiTask()
|