From a27359a85c98038ac8f52a3eecae5c2b4f1a6b6d Mon Sep 17 00:00:00 2001 From: "Pyre Bot Jr." Date: Wed, 11 Sep 2024 20:17:59 -0700 Subject: [PATCH] Add type error suppressions for upcoming upgrade Reviewed By: MaggieMoss Differential Revision: D62550330 --- fbgemm_gpu/test/quantize/fused_8bit_rowwise_test.py | 2 ++ fbgemm_gpu/test/sparse/permute_indices_test.py | 3 +++ 2 files changed, 5 insertions(+) diff --git a/fbgemm_gpu/test/quantize/fused_8bit_rowwise_test.py b/fbgemm_gpu/test/quantize/fused_8bit_rowwise_test.py index a522880148..a2183fcd94 100644 --- a/fbgemm_gpu/test/quantize/fused_8bit_rowwise_test.py +++ b/fbgemm_gpu/test/quantize/fused_8bit_rowwise_test.py @@ -257,6 +257,8 @@ def test_quantize_and_dequantize_op( # noqa: C901 ) ) + # pyre-fixme[61]: `dequantized_data_gpu` is undefined, or not always + # defined. dequantized_data_trimmed = dequantized_data_gpu[:, :ncols].cpu() quantize_data_numpy = quantized_data_gpu.cpu().numpy() if quant_padding_float_type: diff --git a/fbgemm_gpu/test/sparse/permute_indices_test.py b/fbgemm_gpu/test/sparse/permute_indices_test.py index 00fcadec44..fa4fa490c5 100644 --- a/fbgemm_gpu/test/sparse/permute_indices_test.py +++ b/fbgemm_gpu/test/sparse/permute_indices_test.py @@ -87,7 +87,10 @@ def test_permute_indices( ) for t in range(T): for w in range(W): + # pyre-fixme[61]: `batch_sizes` is undefined, or not always defined. for b in range(batch_sizes[w]): + # pyre-fixme[61]: `batch_sizes` is undefined, or not always + # defined. permute_list.append(offset_w[w] + t * batch_sizes[w] + b) else: permute_list = list(range(T))