Home
last modified time | relevance | path

Searched refs:GetLocalRankId (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dsync_batch_norm_gpu_kernel.h149 using GetLocalRankId = device::gpu::GetLocalRankId; in Init() local
151 … reinterpret_cast<GetLocalRankId>(dlsym(const_cast<void *>(collective_handle_), "local_rank_id")); in Init()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/
Dcollective_init.h29 using GetLocalRankId = int (*)(); variable
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc98 using GetLocalRankId = device::gpu::GetLocalRankId; typedef
106 … reinterpret_cast<GetLocalRankId>(dlsym(const_cast<void *>(collective_handle_), "local_rank_id")); in Init()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc71 … reinterpret_cast<GetLocalRankId>(dlsym(const_cast<void *>(collective_handle_), "local_rank_id")); in Initialize()