From db938a172dc2a4b442572414070004fe56925068 Mon Sep 17 00:00:00 2001 From: Elena Zhelezina Date: Thu, 23 Jul 2026 13:35:34 +0100 Subject: [PATCH] Arm backend: Make sure Softmax doesn't use EXT-INT16 Change-Id: I5cb1a4163ea712fac1c0b74a0a68b3f249e161bd Signed-off-by: Elena Zhelezina Change-Id: Icf9c703d10a5d58270b30c6805351abb6a2ea3ca --- backends/arm/test/ops/test_softmax.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/backends/arm/test/ops/test_softmax.py b/backends/arm/test/ops/test_softmax.py index 4f98e2851e9..239f1cb7943 100644 --- a/backends/arm/test/ops/test_softmax.py +++ b/backends/arm/test/ops/test_softmax.py @@ -117,8 +117,9 @@ def test_softmax_vgf_quant(test_data): data, [], quantize=True, + # Make sure that we don't use any EXT-INT16 + tosa_spec="TOSA-1.0+INT", ) - pipeline.add_stage_after("quantize", pipeline.tester.check_not, [aten_op]) # TODO: MLETORCH-1136 Change args of run_method_and_compare_outputs of the vgf tests # pipeline.change_args("run_method_and_compare_outputs", qtol=1) pipeline.run()