avoid converting a TensorOptions from float to integer

Summary: pytorch is adding checks that mean integer tensors with requires_grad=True need to be avoided. Fix accidentally creating them.

Reviewed By: jcjohnson, gkioxari

Differential Revision: D21576712

fbshipit-source-id: 008218997986800a36d93caa1a032ee91f2bffcd
This commit is contained in:
Jeremy Reizenstein
2020-05-14 13:13:36 -07:00
committed by Facebook GitHub Bot
parent 6a365d203f
commit 728179e848
9 changed files with 15 additions and 12 deletions

View File

@@ -140,6 +140,7 @@ class _RasterizePoints(torch.autograd.Function):
)
idx, zbuf, dists = _C.rasterize_points(*args)
ctx.save_for_backward(points, idx)
ctx.mark_non_differentiable(idx)
return idx, zbuf, dists
@staticmethod