Skip to content

Commit c89d2c6

Browse files
priyanshuwustlfacebook-github-bot
authored andcommitted
Replace black_list with block_list (#42088)
Summary: Fixes #41735 Pull Request resolved: #42088 Reviewed By: pbelevich Differential Revision: D22794582 Pulled By: SplitInfinity fbshipit-source-id: e256353befefa2630b99f9bcf0b79df3a7a8dcbd
1 parent a12fe1a commit c89d2c6

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

caffe2/contrib/fakelowp/test/test_int8_quant.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def test_dequantize(self):
4646
{"X": [5, 2]},
4747
debug=True,
4848
adjust_batch=True,
49-
black_list=[0],
49+
block_list=[0],
5050
use_onnx=False,
5151
)
5252
num_onnxified_ops = sum(

caffe2/python/onnx/onnxifi.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def onnxifi_caffe2_net(
2525
use_onnx=True,
2626
merge_fp32_inputs_into_fp16=False,
2727
adjust_batch=True,
28-
black_list=None,
28+
block_list=None,
2929
weight_names=None,
3030
net_ssa_rewritten=False,
3131
timeout=0):
@@ -47,7 +47,7 @@ def onnxifi_caffe2_net(
4747
shape_hints.max_feature_len = max_seq_size
4848
pred_net_str = C.onnxifi(pred_net.SerializeToString(),
4949
shape_hints.SerializeToString(),
50-
black_list if black_list else [],
50+
block_list if block_list else [],
5151
weight_names if weight_names is not None else [],
5252
max_batch_size,
5353
max_seq_size,

0 commit comments

Comments
 (0)