mindspore2022/mindspore/ccsrc/kernel/gpu
VectorSL cc7b05e3ce fix cudnn type error 2020-04-23 11:14:41 +08:00
..
arrays gpu dynamic memory pool supports multi-allReduce 2020-04-22 15:43:45 +08:00
control Add GPU send and recv controlling kernels. 2020-04-16 11:28:54 +08:00
cuda_impl gpu dynamic memory pool supports multi-allReduce 2020-04-22 15:43:45 +08:00
data initial version 2020-03-27 22:54:54 +08:00
math fix cudnn type error 2020-04-23 11:14:41 +08:00
nccl Add GPU send and recv controlling kernels. 2020-04-16 11:28:54 +08:00
nn fix cudnn type error 2020-04-23 11:14:41 +08:00
other gpu add assigin 2020-04-14 20:11:44 +08:00
quant clear the warmming scan by package 2020-04-15 21:45:06 +08:00
gpu_kernel.h initial version 2020-03-27 22:54:54 +08:00
gpu_kernel_factory.cc gpu change compute capacity strategy 2020-04-22 14:40:00 +08:00
gpu_kernel_factory.h initial version 2020-03-27 22:54:54 +08:00
kernel_constants.h initial version 2020-03-27 22:54:54 +08:00