986e3c0a00677355c1edbd89a959264fa0aea1c4,test/quantization/test_numeric_suite_fx.py,TestFXNumericSuiteCoreAPIsModels,test_sparsenn_shadow,#TestFXNumericSuiteCoreAPIsModels#,1025

Before Change

            // check activation result correctness
            act_compare_dict = extract_shadow_logger_info(
                sparse_nn_q, OutputLogger)
            self.assertTrue(len(act_compare_dict) == 4)
            self.assert_ns_compare_dict_valid(act_compare_dict)

After Change

            idx = torch.LongTensor([1, 2, 4, 5, 4, 3, 2, 9])
            offsets = torch.LongTensor([0, 4])
            x = torch.randn(2, 4)
            self._test_match_activations(
                sparse_nn, (idx, offsets, x),
                results_len=4,
                should_log_inputs=should_log_inputs)

Instances