Skip to content

Commit 84b28cc

Browse files
clean up for pr
1 parent 1e027c1 commit 84b28cc

File tree

5 files changed

+4
-7
lines changed

5 files changed

+4
-7
lines changed

cpu/include/group_points.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22
#include <torch/extension.h>
33

44
at::Tensor group_points(at::Tensor points, at::Tensor idx);
5-
at::Tensor group_points_grad(at::Tensor grad_out, at::Tensor idx, const int n);
5+
at::Tensor group_points_grad(at::Tensor grad_out, at::Tensor idx, const int n);

cpu/include/utils.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33

44
#define CHECK_CPU(x) AT_ASSERTM(!x.type().is_cuda(), #x " must be a CPU tensor")
55

6-
#define CHECK_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " must be a contiguous tensor")
6+
#define CHECK_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " must be a contiguous tensor")

cpu/src/bindings.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22

33
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
44
m.def("group_points", &group_points);
5-
}
5+
}

cpu/src/group_points.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,5 +26,4 @@ at::Tensor group_points(at::Tensor points, at::Tensor idx) {
2626
}
2727

2828
return output;
29-
30-
}
29+
}

test/test_grouping.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,5 @@ def test_simple(self):
5050
).detach().cpu().numpy(), expected)
5151

5252

53-
54-
5553
if __name__ == '__main__':
5654
unittest.main()

0 commit comments

Comments
 (0)