Method test_implementation

backends/cortex_m/test/tester.py:128–148

Test the optimized op implementation in simulation

(self, qtol=0, atol=1e-03, calibration_samples=None)

        )

    def test_implementation(self, qtol=0, atol=1e-03, calibration_samples=None):
        """
        Test the optimized op implementation in simulation
        """

        if calibration_samples is not None:
            quantization_stage = CortexMQuantize(
                calibration_samples=calibration_samples
            )
        else:
            quantization_stage = None

        self.quantize(quantization_stage)
        self.export()
        self.to_edge()
        self.run_passes()
        self.to_executorch()
        self.serialize()
        self.run_method_and_compare_outputs(
            inputs=self.example_inputs, qtol=qtol, atol=atol
        )


@dataclass