From d0a1ba16ce1033afe0bdbfaf65055948043a37d6 Mon Sep 17 00:00:00 2001 From: Felix Zimmermann Date: Mon, 11 Nov 2024 08:19:53 -0800 Subject: [PATCH] Tighten type hints for tensor arithmetic Summary: Fixes https://github.com/pytorch/pytorch/issues/124015 cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy yf225 chenyang78 kadeng muchulee8 ColinPeppler amjames desertfire chauhang aakhundov X-link: https://github.com/pytorch/pytorch/pull/135392 Differential Revision: D65753120 Pulled By: ezyang --- devtools/inspector/_inspector_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/devtools/inspector/_inspector_utils.py b/devtools/inspector/_inspector_utils.py index c2e92f09148..83492f9963f 100644 --- a/devtools/inspector/_inspector_utils.py +++ b/devtools/inspector/_inspector_utils.py @@ -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: