diff --git a/pytorch3d/csrc/knn/knn_cpu.cpp b/pytorch3d/csrc/knn/knn_cpu.cpp index 0150a11d..914b6fb5 100644 --- a/pytorch3d/csrc/knn/knn_cpu.cpp +++ b/pytorch3d/csrc/knn/knn_cpu.cpp @@ -13,7 +13,6 @@ std::tuple KNearestNeighborIdxCpu( const int N = p1.size(0); const int P1 = p1.size(1); const int D = p1.size(2); - const int P2 = p2.size(1); auto long_opts = p1.options().dtype(torch::kInt64); torch::Tensor idxs = torch::full({N, P1, K}, 0, long_opts); diff --git a/tests/test_rasterize_meshes.py b/tests/test_rasterize_meshes.py index 28e9ff7b..45ce04e9 100644 --- a/tests/test_rasterize_meshes.py +++ b/tests/test_rasterize_meshes.py @@ -507,7 +507,7 @@ class TestRasterizeMeshes(TestCaseMixin, unittest.TestCase): grad_var1.grad.data.zero_() loss2.backward() grad_verts2 = grad_var2.grad.data.clone().cpu() - self.assertClose(grad_verts1, grad_verts2, rtol=1e-3) + self.assertClose(grad_verts1, grad_verts2, rtol=2e-3) def _test_perspective_correct(self, rasterize_meshes_fn, device, bin_size=None): # fmt: off