mindspore2022/mindspore/ccsrc/kernel/gpu
VectorSL 1a6f62bd25 gpu update type check 2020-04-27 16:09:00 +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 gpu update conv kernel for auto-mixed-precision 2020-04-24 11:29:27 +08:00
nccl Add GPU send and recv controlling kernels. 2020-04-16 11:28:54 +08:00
nn !677 GPU fix codex for conv2d 2020-04-26 14:40:26 +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 update type check 2020-04-27 16:09: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