Skip to content

GitLab

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "2f6af1a3defe256179d8b386c835ff2f2c942832"
Commit 1bc23a18 authored by Ruilong Li's avatar Ruilong Li
Browse files

minor fix if CUB not supported

parent 1714ba0b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
nerfacc/ray_marching.py
View file @ 1bc23a18
...@@ -208,6 +208,7 @@ def ray_marching( ...@@ -208,6 +208,7 @@ def ray_marching(
masks = render_visibility( masks = render_visibility(
alphas, alphas,
ray_indices=ray_indices, ray_indices=ray_indices,
packed_info=packed_info,
early_stop_eps=early_stop_eps, early_stop_eps=early_stop_eps,
alpha_thre=alpha_thre, alpha_thre=alpha_thre,
n_rays=rays_o.shape[0], n_rays=rays_o.shape[0],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment