Add TOSA version string to max_pool tests.

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

Author: per

backends/arm/test/misc/test_debug_feats.py

@@ -108,7 +108,8 @@ def test_numerical_diff_prints(self):
                 model,
                 example_inputs=model.get_inputs(),
                 compile_spec=common.get_tosa_compile_spec(
-                    "TOSA-0.80.0+MI", permute_memory_to_nhwc=True,
+                    "TOSA-0.80.0+MI",
+                    permute_memory_to_nhwc=True,
                     custom_path=tempfile.mkdtemp("diff_print_test"),
                 ),
             )

backends/arm/test/ops/test_max_pool.py

@@ -62,7 +62,9 @@ def _test_maxpool2d_tosa_MI_pipeline(
             ArmTester(
                 module,
                 example_inputs=test_data,
-                compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
+                compile_spec=common.get_tosa_compile_spec(
+                    "TOSA-0.80.0+MI", permute_memory_to_nhwc=True
+                ),
             )
             .export()
             .check(["torch.ops.aten.max_pool2d.default"])
@@ -87,7 +89,9 @@ def _test_maxpool2d_tosa_BI_pipeline(
             ArmTester(
                 module,
                 example_inputs=test_data,
-                compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
+                compile_spec=common.get_tosa_compile_spec(
+                    "TOSA-0.80.0+BI", permute_memory_to_nhwc=True
+                ),
             )
             .quantize(Quantize(quantizer, get_symmetric_quantization_config()))
             .export()