sh in __init__

nerfstudio-project · Sep 29, 2023 · 3f8e568 · 3f8e568
1 parent eed6ee7
commit 3f8e568
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/diff_rast/cuda/__init__.py b/diff_rast/cuda/__init__.py
@@ -16,3 +16,5 @@ def call_cuda(*args, **kwargs):
 compute_cov2d_bounds_forward = _make_lazy_cuda_func("compute_cov2d_bounds_forward")
 project_gaussians_forward = _make_lazy_cuda_func("project_gaussians_forward")
 project_gaussians_backward = _make_lazy_cuda_func("project_gaussians_backward")
+compute_sh_forward = _make_lazy_cuda_func("compute_sh_forward")
+compute_sh_backward = _make_lazy_cuda_func("compute_sh_backward")