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))