mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-12-23 07:40:34 +08:00
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:
committed by
Facebook GitHub Bot
parent
6a365d203f
commit
728179e848
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user