Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DCUDALoops.cuh184 static void launch_legacy_kernel(int64_t N, const func_t& f) { in launch_legacy_kernel() function
275 launch_legacy_kernel<128, unroll_factor>(numel, [=] GPU_LAMBDA(int idx) { in gpu_kernel_impl_nocast()
313 launch_legacy_kernel<512, 1>(numel, [=]GPU_LAMBDA(int idx) { in gpu_kernel_impl()
338 launch_legacy_kernel<128, 4>(numel, [=] GPU_LAMBDA(int idx) { in gpu_kernel_impl()