diff --git a/caffe2/contrib/fakelowp/test/test_int8_quant.py b/caffe2/contrib/fakelowp/test/test_int8_quant.py index 614f0643e942b..83d0cc176def7 100644 --- 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 index 1500390ca6e53..6bbd35cd434cf 100644 --- 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,