Skip to content

Commit 7ce4c1c

Browse files
committed
added at:: namespace
1 parent 6667f3d commit 7ce4c1c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cpu/grid.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@ at::Tensor grid(at::Tensor pos, at::Tensor size, at::Tensor start,
88
num_voxels = num_voxels.cumprod(0);
99

1010
num_voxels = at::cat({at::ones(1, num_voxels.options()), num_voxels}, 0);
11-
auto index = empty(size.size(0), num_voxels.options());
12-
arange_out(index, size.size(0));
11+
auto index = at::empty(size.size(0), num_voxels.options());
12+
at::arange_out(index, size.size(0));
1313
num_voxels = num_voxels.index_select(0, index);
1414

1515
auto cluster = (pos / size.view({1, -1})).toType(at::kLong);

0 commit comments

Comments
 (0)