Home
last modified time | relevance | path

Searched refs:InitNCCLComm (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/
Dcollective_wrapper.cc25 void InitNCCLComm() { NCCLWrapper::instance().InitNCCLComm(); } in InitNCCLComm() function
Dcollective_init.h28 using InitNCCLComm = void (*)(); variable
Dnccl_wrapper.h37 void InitNCCLComm();
Dcollective_wrapper.h34 extern "C" EXPORT_WRAPPER void InitNCCLComm();
Dnccl_wrapper.cc34 void NCCLWrapper::InitNCCLComm() { in InitNCCLComm() function in mindspore::device::gpu::NCCLWrapper
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc99 using InitNCCLComm = device::gpu::InitNCCLComm; typedef
120 … reinterpret_cast<InitNCCLComm>(dlsym(const_cast<void *>(collective_handle_), "InitNCCLComm")); in Init()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc95 reinterpret_cast<InitNCCLComm>(dlsym(const_cast<void *>(collective_handle_), "InitNCCLComm")); in Initialize()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc102 reinterpret_cast<InitNCCLComm>(dlsym(const_cast<void *>(collective_handle_), "InitNCCLComm")); in Init()