Skip to content

Commit 51dc602

Browse files
committed
Add TOSA version string to max_pool tests.
Signed-off-by: Per Åstrand <[email protected]> Change-Id: I14ce1d9f53396d3ff49536875f24dabd24a71c10
1 parent 371bcfd commit 51dc602

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

backends/arm/test/ops/test_max_pool.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,9 @@ def _test_maxpool2d_tosa_MI_pipeline(
6262
ArmTester(
6363
module,
6464
example_inputs=test_data,
65-
compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
65+
compile_spec=common.get_tosa_compile_spec(
66+
"TOSA-0.80.0+MI", permute_memory_to_nhwc=True
67+
),
6668
)
6769
.export()
6870
.check(["torch.ops.aten.max_pool2d.default"])
@@ -87,7 +89,9 @@ def _test_maxpool2d_tosa_BI_pipeline(
8789
ArmTester(
8890
module,
8991
example_inputs=test_data,
90-
compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
92+
compile_spec=common.get_tosa_compile_spec(
93+
"TOSA-0.80.0+BI", permute_memory_to_nhwc=True
94+
),
9195
)
9296
.quantize(Quantize(quantizer, get_symmetric_quantization_config()))
9397
.export()

0 commit comments

Comments
 (0)