Skip to content

Commit 30412cc

Browse files
authored
Update relative file path in dvgo_utils.py
Update relative path of dvgo cpp and cu files.
1 parent 6afde85 commit 30412cc

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

src/dwm/models/base_vq_models/dvgo_utils.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,13 @@
55
from torch.utils.cpp_extension import load
66

77
parent_dir = os.path.dirname(os.path.abspath(__file__))
8+
cpp_file = os.path.join(parent_dir, '../../../../externals/dvgo_cuda/lib/cuda/render_utils.cpp')
9+
cu_file = os.path.join(parent_dir, '../../../../externals/dvgo_cuda/lib/cuda/render_utils_kernel.cu')
810
render_utils_cuda = load(
911
name='render_utils_cuda',
10-
sources=[
11-
os.path.join(parent_dir, path)
12-
for path in ['../../../../externals/dvgo_cuda/render_utils.cpp', '../../../../externals/dvgo_cuda/render_utils_kernel.cu']],
12+
sources=[cpp_file, cu_file],
1313
verbose=True)
1414

15-
1615
def sample_ray(rays_o, rays_d, near, far, stepsize, xyz_min, xyz_max, voxel_size):
1716
'''Sample query points on rays.
1817
All the output points are sorted from near to far.

0 commit comments

Comments
 (0)