mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-12-22 23:30:35 +08:00
suppress errors in vision - batch 1
Summary: This diff is auto-generated to upgrade the Pyre version and suppress errors in vision. The upgrade will affect Pyre local configurations in the following directories: ``` vision/ale/search vision/fair/fvcore vision/fair/pytorch3d vision/ocr/rosetta_hash vision/vogue/personalization ``` Differential Revision: D21688454 fbshipit-source-id: 1f3c3fee42b6da2e162fd0932742ab8c5c96aa45
This commit is contained in:
committed by
Facebook GitHub Bot
parent
d689baac5e
commit
ae68a54f67
@@ -89,6 +89,7 @@ def rasterize_points(
|
||||
|
||||
if bin_size != 0:
|
||||
# There is a limit on the number of points per bin in the cuda kernel.
|
||||
# pyre-fixme[6]: Expected `int` for 1st param but got `Union[int, None, int]`.
|
||||
points_per_bin = 1 + (image_size - 1) // bin_size
|
||||
if points_per_bin >= kMaxPointsPerBin:
|
||||
raise ValueError(
|
||||
@@ -101,6 +102,7 @@ def rasterize_points(
|
||||
|
||||
# Function.apply cannot take keyword args, so we handle defaults in this
|
||||
# wrapper and call apply with positional args only
|
||||
# pyre-fixme[16]: `_RasterizePoints` has no attribute `apply`.
|
||||
return _RasterizePoints.apply(
|
||||
points_packed,
|
||||
cloud_to_packed_first_idx,
|
||||
@@ -138,6 +140,7 @@ class _RasterizePoints(torch.autograd.Function):
|
||||
bin_size,
|
||||
max_points_per_bin,
|
||||
)
|
||||
# pyre-fixme[16]: Module `pytorch3d` has no attribute `_C`.
|
||||
idx, zbuf, dists = _C.rasterize_points(*args)
|
||||
ctx.save_for_backward(points, idx)
|
||||
ctx.mark_non_differentiable(idx)
|
||||
|
||||
Reference in New Issue
Block a user