| 8 | |
| 9 | |
| 10 | int gather_points_wrapper_fast(int b, int c, int n, int npoints, |
| 11 | at::Tensor points_tensor, at::Tensor idx_tensor, at::Tensor out_tensor){ |
| 12 | const float *points = points_tensor.data<float>(); |
| 13 | const int *idx = idx_tensor.data<int>(); |
| 14 | float *out = out_tensor.data<float>(); |
| 15 | |
| 16 | cudaStream_t stream = at::cuda::getCurrentCUDAStream().stream(); |
| 17 | gather_points_kernel_launcher_fast(b, c, n, npoints, points, idx, out, stream); |
| 18 | return 1; |
| 19 | } |
| 20 | |
| 21 | |
| 22 | int gather_points_grad_wrapper_fast(int b, int c, int n, int npoints, |
nothing calls this directly
no outgoing calls
no test coverage detected