Skip to content

Commit

Permalink
Tighten type hints for tensor arithmetic
Browse files Browse the repository at this point in the history
Differential Revision: D65753120

Pull Request resolved: #6752
  • Loading branch information
ezyang authored Nov 11, 2024
1 parent 576e96c commit f90cf2d
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions devtools/inspector/_inspector_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,6 +218,7 @@ def verify_debug_data_equivalence(

if isinstance(output_a, torch.Tensor):
assert bool(
# pyre-fixme[6]: For 1st argument expected `Tensor` but got `bool`.
torch.all(output_a == output_b)
), "Tensors Debug Data is different. Expected to be equal."
else:
Expand Down

0 comments on commit f90cf2d

Please sign in to comment.