Skip to content

Commit a92ca0b

Browse files
Haixin Liucsarofeen
authored andcommitted
Remove unused Logger in get_matching_activations (pytorch#41023)
Summary: Pull Request resolved: pytorch#41023 Remove Logger in get_matching_activations since it's not used. ghstack-source-id: 107237046 Test Plan: buck test mode/dev caffe2/test:quantization -- 'test_compare_weights_lstm_dynamic' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_stub_lstm_dynamic' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_outputs_lstm_dynamic' buck test mode/dev caffe2/test:quantization -- 'test_compare_weights_conv_static' buck test mode/dev caffe2/test:quantization -- 'test_compare_weights_linear_static' buck test mode/dev caffe2/test:quantization -- 'test_compare_weights_linear_dynamic' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_stub_conv_static' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_stub_linear_static' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_stub_submodule_static' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_stub_functional_static' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_stub_linear_dynamic' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_outputs_conv_static' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_outputs_linear_static' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_outputs_functional_static' buck test mode/dev caffe2/test:quantization -- 'test_compare_model_outputs_linear_dynamic' Differential Revision: D22394957 fbshipit-source-id: 7d59e0f35e9f4c304b8487460d48236ee6e5a872
1 parent e9cdf46 commit a92ca0b

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

torch/quantization/_numeric_suite.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -381,13 +381,12 @@ def compare_model_stub(
381381
return ob_dict
382382

383383

384-
def get_matching_activations(float_module, q_module, Logger):
384+
def get_matching_activations(float_module, q_module):
385385
r"""Find the matching activation between float and quantized modules.
386386
387387
Args:
388388
float_module: float module used to generate the q_module
389389
q_module: module quantized from float_module
390-
Logger: type of logger used to prepare float_module and q_module
391390
392391
Return:
393392
act_dict: dict with key corresponding to quantized module names and each
@@ -467,5 +466,5 @@ def compare_model_outputs(
467466
prepare_model_outputs(float_model, q_model, Logger, white_list)
468467
float_model(*data)
469468
q_model(*data)
470-
act_compare_dict = get_matching_activations(float_model, q_model, Logger)
469+
act_compare_dict = get_matching_activations(float_model, q_model)
471470
return act_compare_dict

0 commit comments

Comments
 (0)