Home
last modified time | relevance | path

Searched refs:gpu_kernel_nocast (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DCopy.cu39 gpu_kernel_nocast(iter, [] GPU_LAMBDA(float value) { in float8_copy_kernel_cuda()
44 gpu_kernel_nocast(iter, [] GPU_LAMBDA(Half value) { in float8_copy_kernel_cuda()
49 gpu_kernel_nocast(iter, [] GPU_LAMBDA(BFloat16 value) { in float8_copy_kernel_cuda()
60 gpu_kernel_nocast(iter, [] GPU_LAMBDA(float value) { in float8_copy_kernel_cuda()
70 gpu_kernel_nocast(iter, [] GPU_LAMBDA(Half value) { in float8_copy_kernel_cuda()
80 gpu_kernel_nocast(iter, [] GPU_LAMBDA(BFloat16 value) { in float8_copy_kernel_cuda()
96 gpu_kernel_nocast(iter, [] GPU_LAMBDA(float value) { in float8_copy_kernel_cuda()
101 gpu_kernel_nocast(iter, [] GPU_LAMBDA(Half value) { in float8_copy_kernel_cuda()
106 gpu_kernel_nocast(iter, [] GPU_LAMBDA(BFloat16 value) { in float8_copy_kernel_cuda()
117 gpu_kernel_nocast(iter, [] GPU_LAMBDA(float value) { in float8_copy_kernel_cuda()
[all …]
DLoops.cuh76 void gpu_kernel_nocast(TensorIteratorBase& iter, const func_t& f) { in gpu_kernel_nocast() function
90 gpu_kernel_nocast(sub_iter, f); in gpu_kernel_nocast()