986e3c0a00677355c1edbd89a959264fa0aea1c4,test/quantization/test_numeric_suite_fx.py,TestFXNumericSuiteCoreAPIs,test_logging_inputs,#TestFXNumericSuiteCoreAPIs#,950

Before Change

        m = M().eval()
        mp = prepare_fx(m, {"": torch.quantization.default_qconfig})
        mp(torch.randn(1, 1, 4, 4))
        // TODO(future PR): prevent the need for copying here, we can copy the
        // modules but should reuse the underlying tensors
        mp_copy = copy.deepcopy(mp)
        mq = convert_fx(mp_copy)

        mp_shadows_mq = add_shadow_loggers(
            "fp32_prepared", mp, "int8", mq, OutputLogger, should_log_inputs=True)
        mp_shadows_mq(torch.randn(1, 1, 4, 4))

        act_compare_dict = extract_shadow_logger_info(
            mp_shadows_mq, OutputLogger)

After Change

                return x

        m = M().eval()
        self._test_match_shadow_activations(
            m, (torch.randn(1, 1, 4, 4),),
            results_len=2,
            should_log_inputs=True)


class TestFXNumericSuiteCoreAPIsModels(FXNumericSuiteQuantizationTestCase):
    

Instances