add assert in traverse_grids

Linyou · Apr 25, 2023 · c93eaad · c93eaad
1 parent 4581d95
commit c93eaad
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/benchmarks/kplanes b/benchmarks/kplanes
diff --git a/benchmarks/tensorf b/benchmarks/tensorf
diff --git a/benchmarks/tineuvox b/benchmarks/tineuvox
diff --git a/nerfacc/grid.py b/nerfacc/grid.py
@@ -145,9 +145,13 @@ def traverse_grids(
         far_planes = torch.full_like(rays_o[:, 0], float("inf"))
 
     if rays_mask is None:
-        rays_mask = torch.Tensor()
+        rays_mask = torch.Tensor()  # Dummy tensor.
     if traverse_steps_limit is None:
         traverse_steps_limit = -1
+    if over_allocate:
+        assert (
+            traverse_steps_limit > 0
+        ), "traverse_steps_limit must be set if over_allocate is True."
 
     if t_sorted is None or t_indices is None or hits is None:
         # Compute ray aabb intersection for all levels of grid. [n_rays, m]