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(
62
62
ArmTester (
63
63
module ,
64
64
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
+ ),
66
68
)
67
69
.export ()
68
70
.check (["torch.ops.aten.max_pool2d.default" ])
@@ -87,7 +89,9 @@ def _test_maxpool2d_tosa_BI_pipeline(
87
89
ArmTester (
88
90
module ,
89
91
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
+ ),
91
95
)
92
96
.quantize (Quantize (quantizer , get_symmetric_quantization_config ()))
93
97
.export ()
You can’t perform that action at this time.
0 commit comments