Arm backend: Handle extension for int16 in tests

The sigmoid tests uses the EXT-INT16 of the TOSA 1.0 specification,
so the extension must be used in the tests.

Signed-off-by: Per Åstrand <[email protected]>
Change-Id: Ie0d1a03d73e46139c6b6a0822736e2270e716650

Author: per

backends/arm/test/ops/test_sigmoid_16bit.py

@@ -41,7 +41,7 @@ def get_16bit_sigmoid_quantizer(u55_config=False):
     tosa_version = conftest.get_option("tosa_version")
     tosa_profiles = {
         "1.0": TosaSpecification.create_from_string(
-            "TOSA-1.0+INT" + ("+u55" if u55_config else "")
+            "TOSA-1.0+INT+int16" + ("+u55" if u55_config else "")
         ),
     }
 
@@ -94,6 +94,7 @@ def test_sigmoid_tosa_INT(test_data):
         Sigmoid.aten_op,
         Sigmoid.exir_op,
         qtol=1,
+        tosa_extensions=["int16"],
     )
     pipeline.change_args("quantize", get_16bit_sigmoid_quantizer())
     pipeline.run()
@@ -114,7 +115,9 @@ def test_sigmoid_tosa_INT_add_sigmoid(test_data):
         Sigmoid.aten_op,
         Sigmoid.exir_op,
         qtol=1,
+        tosa_extensions=["int16"],
     )
+    pipeline.change_args("quantize", get_16bit_sigmoid_quantizer())
     pipeline.run()
 
 
@@ -154,6 +157,7 @@ def test_sigmoid_u55_INT_add_sigmoid(test_data):
         n_expected_delegates=1,
         quantize=True,
         u55_subset=True,
+        tosa_extensions=["int16"],
     )
     pipeline.change_args("quantize", get_16bit_sigmoid_quantizer(True))
     pipeline.run()

backends/arm/test/ops/test_sigmoid_32bit.py

@@ -57,7 +57,7 @@ def get_32bit_sigmoid_quantizer(u55_config=False):
     tosa_version = conftest.get_option("tosa_version")
     tosa_profiles = {
         "1.0": TosaSpecification.create_from_string(
-            "TOSA-1.0+INT" + ("+u55" if u55_config else "")
+            "TOSA-1.0+INT+int16" + ("+u55" if u55_config else "")
         ),
     }
 
@@ -110,6 +110,7 @@ def test_sigmoid_tosa_INT(test_data):
         Sigmoid.aten_op,
         Sigmoid.exir_op,
         qtol=1,
+        tosa_extensions=["int16"],
     )
     pipeline.change_args("quantize", get_32bit_sigmoid_quantizer())
     pipeline.run()
@@ -123,6 +124,7 @@ def test_sigmoid_tosa_INT_add_sigmoid(test_data):
         Sigmoid.aten_op,
         Sigmoid.exir_op,
         qtol=1,
+        tosa_extensions=["int16"],
     )
     pipeline.change_args("quantize", get_32bit_sigmoid_quantizer())
     pipeline.run()
@@ -136,6 +138,7 @@ def test_sigmoid_u55_INT(test_data):
         {Sigmoid.exir_op: 1},
         quantize=True,
         u55_subset=True,
+        tosa_extensions=["int16"],
     )
     pipeline.change_args("quantize", get_32bit_sigmoid_quantizer(True))
     pipeline.run()
@@ -150,6 +153,7 @@ def test_sigmoid_u55_INT_add_sigmoid(test_data):
         n_expected_delegates=1,
         quantize=True,
         u55_subset=True,
+        tosa_extensions=["int16"],
     )
     pipeline.change_args("quantize", get_32bit_sigmoid_quantizer(True))
     pipeline.run()