Skip to content

Commit 2989b3f

Browse files
authored
Merge pull request #25 from concerttttt/bufix/submodule-update-dvgo-cuda
Update .gitmodules for dvgo_cuda
2 parents 0bc3ee2 + 30412cc commit 2989b3f

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

.gitmodules

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,3 +7,6 @@
77
[submodule "externals/waymo-open-dataset"]
88
path = externals/waymo-open-dataset
99
url = https://github.com/waymo-research/waymo-open-dataset.git
10+
[submodule "externals/dvgo_cuda"]
11+
path = externals/dvgo_cuda
12+
url = https://github.com/sunset1995/DirectVoxGO.git

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)