Replace black_list with block_list (pytorch#42088)

Summary: Fixes pytorch#41735 Pull Request resolved: pytorch#42088 Reviewed By: pbelevich Differential Revision: D22794582 Pulled By: SplitInfinity fbshipit-source-id: e256353befefa2630b99f9bcf0b79df3a7a8dcbd
luxe · Aug 20, 2020 · c89d2c6 · c89d2c6
1 parent a12fe1a
commit c89d2c6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/caffe2/contrib/fakelowp/test/test_int8_quant.py b/caffe2/contrib/fakelowp/test/test_int8_quant.py
@@ -46,7 +46,7 @@ def test_dequantize(self):
             {"X": [5, 2]},
             debug=True,
             adjust_batch=True,
-            black_list=[0],
+            block_list=[0],
             use_onnx=False,
         )
         num_onnxified_ops = sum(

diff --git a/caffe2/python/onnx/onnxifi.py b/caffe2/python/onnx/onnxifi.py
@@ -25,7 +25,7 @@ def onnxifi_caffe2_net(
         use_onnx=True,
         merge_fp32_inputs_into_fp16=False,
         adjust_batch=True,
-        black_list=None,
+        block_list=None,
         weight_names=None,
         net_ssa_rewritten=False,
         timeout=0):
@@ -47,7 +47,7 @@ def onnxifi_caffe2_net(
         shape_hints.max_feature_len = max_seq_size
     pred_net_str = C.onnxifi(pred_net.SerializeToString(),
                              shape_hints.SerializeToString(),
-                             black_list if black_list else [],
+                             block_list if block_list else [],
                              weight_names if weight_names is not None else [],
                              max_batch_size,
                              max_seq_size,