Searched refs:HcclAllReduce (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_all_reduce.cc | 36 …auto hccl_result = hccl::HcclAdapter::GetInstance().HcclAllReduce(inputs[0]->addr, outputs[0]->add… in Launch()
|
| /third_party/mindspore/tests/ut/cpp/stub/ge/ |
| D | ge_task_launch_stub.cc | 41 HcclResult HcclAdapter::HcclAllReduce(void *, void *, uint64_t, HcclDataType, HcclReduceOp, aclrtSt… in HcclAllReduce() function in mindspore::hccl::HcclAdapter
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/plugin/ |
| D | hccl_plugin.h | 49 ORIGIN_METHOD(HcclAllReduce, HcclResult, void *, void *, uint64_t, HcclDataType, HcclReduceOp, Hccl…
|
| /third_party/mindspore/tests/ut/cpp/stub/hccl/ |
| D | hccl_stub.cc | 152 HcclResult HcclAllReduce(void *sendBuf, void *recvBuf, uint64_t count, HcclDataType dataType, HcclR… in HcclAllReduce() function
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | hccl_adapter.h | 68 …HcclResult HcclAllReduce(void *send_buf, void *recv_buf, uint64_t count, HcclDataType dataType, Hc…
|
| D | hccl_adapter.cc | 94 launch_hccl_all_reduce_ = DlsymFuncObj(HcclAllReduce, plugin_handle_); in InitPlugin()
279 HcclResult HcclAdapter::HcclAllReduce(void *send_buf, void *recv_buf, uint64_t count, HcclDataType … in HcclAllReduce() function in mindspore::hccl::HcclAdapter
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_bucket.cc | 151 …auto hccl_result = hccl::HcclAdapter::GetInstance().HcclAllReduce(ar_input_addr_, ar_output_addr_,… in LaunchAllReduce()
|