We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a962e69 commit 9e4cbdaCopy full SHA for 9e4cbda
1 file changed
backends/arm/test/ops/test_max_pool.py
@@ -62,7 +62,9 @@ def _test_maxpool2d_tosa_MI_pipeline(
62
ArmTester(
63
module,
64
example_inputs=test_data,
65
- compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
+ compile_spec=common.get_tosa_compile_spec(
66
+ "TOSA-0.80.0+MI", permute_memory_to_nhwc=True
67
+ ),
68
)
69
.export()
70
.check(["torch.ops.aten.max_pool2d.default"])
@@ -87,7 +89,9 @@ def _test_maxpool2d_tosa_BI_pipeline(
87
89
88
90
91
92
93
+ "TOSA-0.80.0+BI", permute_memory_to_nhwc=True
94
95
96
.quantize(Quantize(quantizer, get_symmetric_quantization_config()))
97
0 commit comments