| #include <torch/extension.h> |
|
|
| #include "raymarching.h" |
|
|
| PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { |
| |
| m.def("packbits", &packbits, "packbits (CUDA)"); |
| m.def("near_far_from_aabb", &near_far_from_aabb, "near_far_from_aabb (CUDA)"); |
| m.def("sph_from_ray", &sph_from_ray, "sph_from_ray (CUDA)"); |
| m.def("morton3D", &morton3D, "morton3D (CUDA)"); |
| m.def("morton3D_invert", &morton3D_invert, "morton3D_invert (CUDA)"); |
| |
| m.def("march_rays_train", &march_rays_train, "march_rays_train (CUDA)"); |
| m.def("composite_rays_train_forward", &composite_rays_train_forward, "composite_rays_train_forward (CUDA)"); |
| m.def("composite_rays_train_backward", &composite_rays_train_backward, "composite_rays_train_backward (CUDA)"); |
| |
| m.def("march_rays", &march_rays, "march rays (CUDA)"); |
| m.def("composite_rays", &composite_rays, "composite rays (CUDA)"); |
| } |
