Skip to content

Commit

Permalink
Add pyre fixme for downstream type errors from D65753120
Browse files Browse the repository at this point in the history
Summary:
X-link: pytorch/captum#1439

X-link: facebook/Ax#3055

X-link: ctrl-labs/src2#38515

X-link: ctrl-labs/src2#38514

bypass-github-export-checks "The check is bugged, I exported all the required exports"

Reviewed By: jermenkoo

Differential Revision: D65826205

fbshipit-source-id: b08f00caaac8f3cc235bc280c4fe3f99089a0753
  • Loading branch information
ezyang authored and facebook-github-bot committed Nov 12, 2024
1 parent 1ef8eb9 commit d9d4066
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions fbgemm_gpu/experimental/gemm/test/fp8_gemm_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -156,6 +156,8 @@ def _quantize_matmul_fp8(

expected_result = a @ b.T
if use_bias:
# pyre-fixme[6]: For 1st argument expected `Union[bool, complex,
# float, int, Tensor]` but got `Optional[Tensor]`.
expected_result += bias
self.assertTrue(
torch.allclose(result, expected_result, atol=2e-1, rtol=5e-2)
Expand Down

0 comments on commit d9d4066

Please sign in to comment.