File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments