Home
last modified time | relevance | path

Searched refs:NcclGpuKernel (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_gpu_kernel.h46 class NcclGpuKernel : public GpuKernel {
48 NcclGpuKernel() : group_name_(""), nccl_data_type_(ncclHalf) {} in NcclGpuKernel() function
49 ~NcclGpuKernel() override = default;
Dnccl_send_gpu_kernel.h28 class NcclSendGpuKernel : public NcclGpuKernel {
Dnccl_recv_gpu_kernel.h28 class NcclRecvGpuKernel : public NcclGpuKernel {
Dsync_batch_norm_grad_gpu_kernel.h33 class SyncBatchNormGradGpuKernel : public NcclGpuKernel {
Dnccl_collective_gpu_kernel.h44 class NcclCollectiveGpuKernel : public NcclGpuKernel {
Dsync_batch_norm_gpu_kernel.h33 class SyncBatchNormGpuKernel : public NcclGpuKernel {