| 8 | #include "voxelization/vox.hpp" |
| 9 | |
| 10 | PYBIND11_MODULE(_pvcnn_backend, m) { |
| 11 | m.def("gather_features_forward", &gather_features_forward, |
| 12 | "Gather Centers' Features forward (CUDA)"); |
| 13 | m.def("gather_features_backward", &gather_features_backward, |
| 14 | "Gather Centers' Features backward (CUDA)"); |
| 15 | m.def("furthest_point_sampling", &furthest_point_sampling_forward, |
| 16 | "Furthest Point Sampling (CUDA)"); |
| 17 | m.def("ball_query", &ball_query_forward, "Ball Query (CUDA)"); |
| 18 | m.def("grouping_forward", &grouping_forward, |
| 19 | "Grouping Features forward (CUDA)"); |
| 20 | m.def("grouping_backward", &grouping_backward, |
| 21 | "Grouping Features backward (CUDA)"); |
| 22 | m.def("three_nearest_neighbors_interpolate_forward", |
| 23 | &three_nearest_neighbors_interpolate_forward, |
| 24 | "3 Nearest Neighbors Interpolate forward (CUDA)"); |
| 25 | m.def("three_nearest_neighbors_interpolate_backward", |
| 26 | &three_nearest_neighbors_interpolate_backward, |
| 27 | "3 Nearest Neighbors Interpolate backward (CUDA)"); |
| 28 | |
| 29 | m.def("trilinear_devoxelize_forward", &trilinear_devoxelize_forward, |
| 30 | "Trilinear Devoxelization forward (CUDA)"); |
| 31 | m.def("trilinear_devoxelize_backward", &trilinear_devoxelize_backward, |
| 32 | "Trilinear Devoxelization backward (CUDA)"); |
| 33 | m.def("avg_voxelize_forward", &avg_voxelize_forward, |
| 34 | "Voxelization forward with average pooling (CUDA)"); |
| 35 | m.def("avg_voxelize_backward", &avg_voxelize_backward, |
| 36 | "Voxelization backward (CUDA)"); |
| 37 | } |
nothing calls this directly
no outgoing calls
no test coverage detected