Searched refs:HcclBroadcast (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_all_broadcast.cc | 32 auto hccl_result = hccl::HcclAdapter::GetInstance().HcclBroadcast(inputs[0]->addr, hccl_count_, in Launch()
|
| /third_party/mindspore/tests/ut/cpp/stub/ge/ |
| D | ge_task_launch_stub.cc | 38 HcclResult HcclAdapter::HcclBroadcast(void *, uint64_t, HcclDataType, uint32_t, aclrtStream) const { in HcclBroadcast() function in mindspore::hccl::HcclAdapter
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/plugin/ |
| D | hccl_plugin.h | 48 ORIGIN_METHOD(HcclBroadcast, HcclResult, void *, uint64_t, HcclDataType, uint32_t, HcclComm, aclrtS…
|
| /third_party/mindspore/tests/ut/cpp/stub/hccl/ |
| D | hccl_stub.cc | 157 HcclResult HcclBroadcast(void *buf, uint64_t count, HcclDataType dataType, uint32_t root, HcclComm … in HcclBroadcast() function
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | hccl_adapter.h | 67 …HcclResult HcclBroadcast(void *buf, uint64_t count, HcclDataType dataType, uint32_t root, aclrtStr…
|
| D | hccl_adapter.cc | 93 launch_hccl_broadcast_ = DlsymFuncObj(HcclBroadcast, plugin_handle_); in InitPlugin()
273 HcclResult HcclAdapter::HcclBroadcast(void *buf, uint64_t count, HcclDataType dataType, uint32_t ro… in HcclBroadcast() function in mindspore::hccl::HcclAdapter
|