Searched refs:compute_cuda_kernel (Results 1 – 1 of 1) sorted by relevance
14 __global__ static void compute_cuda_kernel( in compute_cuda_kernel() function48 compute_cuda_kernel<<<grid, block, 0, at::cuda::getCurrentCUDAStream()>>>( in compute_cuda()