Skip to content

Commit c230e94

Browse files
authored
Adjust 'test_quantize_op_fusion_pass' test to account for SHIFT_INT8 value (2^20)
Differential Revision: D85986326 Pull Request resolved: #15503
1 parent bbc0967 commit c230e94

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/cortex_m/test/test_quantize_op_fusion_pass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,7 @@ def forward(self, x, y):
298298

299299
# Output multiplier should be close to 2^30 (for 1.0 scale)
300300
self.assertAlmostEqual(output_mult, 2**30, delta=1000)
301-
self.assertEqual(output_shift, -1)
301+
self.assertEqual(output_shift, -18)
302302

303303
def test_executorch_program_generation(self):
304304
"""Verify ExecuTorch program generation with fused ops"""

0 commit comments

Comments
 (0)