Skip to content

Commit e2c9a48

Browse files
authored
Arm backend: Make sure Softmax doesn't use EXT-INT16 (#21286)
cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell @rascani Signed-off-by: Elena Zhelezina <elena.zhelezina@arm.com>
1 parent 1e42581 commit e2c9a48

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

backends/arm/test/ops/test_softmax.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,8 +117,9 @@ def test_softmax_vgf_quant(test_data):
117117
data,
118118
[],
119119
quantize=True,
120+
# Make sure that we don't use any EXT-INT16
121+
tosa_spec="TOSA-1.0+INT",
120122
)
121-
pipeline.add_stage_after("quantize", pipeline.tester.check_not, [aten_op])
122123
# TODO: MLETORCH-1136 Change args of run_method_and_compare_outputs of the vgf tests
123124
# pipeline.change_args("run_method_and_compare_outputs", qtol=1)
124125
pipeline.run()

0 commit comments

Comments
 (0)