Looser gradient check in test_rasterize_meshes

Summary: This has been failing intermittently

Reviewed By: nikhilaravi

Differential Revision: D21403157

fbshipit-source-id: 51b74d6c813b52effe72d14b565e250fcabbb463
This commit is contained in:
Jeremy Reizenstein
2020-05-05 09:24:24 -07:00
committed by Facebook GitHub Bot
parent 06ad1fb6c4
commit 8fc28baa27
2 changed files with 1 additions and 2 deletions

View File

@@ -13,7 +13,6 @@ std::tuple<at::Tensor, at::Tensor> 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);