Skip to content

Commit a34cbea

Browse files
perper
committed
Add TOSA version string to all TOSA tests
Change-Id: Ia2b21c14f56793d6a9125ae7371779df93ecf9d5
1 parent d9e4403 commit a34cbea

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

46 files changed

+139
-101
lines changed

‎backends/arm/test/misc/test_debug_feats.py‎

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def _tosa_MI_pipeline(self, module: torch.nn.Module, dump_file=None):
4949
ArmTester(
5050
module,
5151
example_inputs=module.get_inputs(),
52-
compile_spec=common.get_tosa_compile_spec(),
52+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
5353
)
5454
.export()
5555
.to_edge()
@@ -63,7 +63,7 @@ def _tosa_BI_pipeline(self, module: torch.nn.Module, dump_file=None):
6363
ArmTester(
6464
module,
6565
example_inputs=module.get_inputs(),
66-
compile_spec=common.get_tosa_compile_spec(),
66+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
6767
)
6868
.quantize()
6969
.export()
@@ -108,7 +108,7 @@ def test_numerical_diff_prints(self):
108108
model,
109109
example_inputs=model.get_inputs(),
110110
compile_spec=common.get_tosa_compile_spec(
111-
permute_memory_to_nhwc=True,
111+
"TOSA-0.80.0+MI", permute_memory_to_nhwc=True,
112112
custom_path=tempfile.mkdtemp("diff_print_test"),
113113
),
114114
)
@@ -135,7 +135,7 @@ def test_dump_ops_and_dtypes():
135135
ArmTester(
136136
model,
137137
example_inputs=model.get_inputs(),
138-
compile_spec=common.get_tosa_compile_spec(),
138+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
139139
)
140140
.quantize()
141141
.dump_dtype_distribution()
@@ -159,7 +159,7 @@ def test_dump_ops_and_dtypes_parseable():
159159
ArmTester(
160160
model,
161161
example_inputs=model.get_inputs(),
162-
compile_spec=common.get_tosa_compile_spec(),
162+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
163163
)
164164
.quantize()
165165
.dump_dtype_distribution(print_table=False)
@@ -190,7 +190,7 @@ def test_collate_tosa_BI_tests(self):
190190
ArmTester(
191191
model,
192192
example_inputs=model.get_inputs(),
193-
compile_spec=common.get_tosa_compile_spec(),
193+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
194194
)
195195
.quantize()
196196
.export()
@@ -220,7 +220,7 @@ def test_dump_tosa_ops(caplog):
220220
ArmTester(
221221
model,
222222
example_inputs=model.get_inputs(),
223-
compile_spec=common.get_tosa_compile_spec(),
223+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
224224
)
225225
.quantize()
226226
.export()

‎backends/arm/test/misc/test_dim_order_guards.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def test_tosa_MI_pipeline(self):
3434
ArmTester(
3535
module,
3636
example_inputs=module.get_inputs(),
37-
compile_spec=common.get_tosa_compile_spec(),
37+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
3838
)
3939
.export()
4040
.to_edge()
@@ -48,7 +48,7 @@ def test_tosa_BI_pipeline(self):
4848
ArmTester(
4949
module,
5050
example_inputs=module.get_inputs(),
51-
compile_spec=common.get_tosa_compile_spec(),
51+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
5252
)
5353
.quantize()
5454
.export()

‎backends/arm/test/misc/test_lifted_tensor.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def test_partition_lifted_tensor_tosa_MI(self, op, data):
6060
ArmTester(
6161
LiftedTensor(op),
6262
example_inputs=data,
63-
compile_spec=common.get_tosa_compile_spec(),
63+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
6464
)
6565
.export()
6666
.to_edge()
@@ -77,7 +77,7 @@ def test_partition_lifted_tensor_tosa_BI(self, op, data):
7777
ArmTester(
7878
LiftedTensor(op),
7979
example_inputs=data,
80-
compile_spec=common.get_tosa_compile_spec(),
80+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
8181
)
8282
.quantize()
8383
.export()
@@ -95,7 +95,7 @@ def test_partition_lifted_scalar_tensor_tosa_MI(self, op, data, arg1):
9595
ArmTester(
9696
LiftedScalarTensor(op, arg1),
9797
example_inputs=(data),
98-
compile_spec=common.get_tosa_compile_spec(),
98+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
9999
)
100100
.export()
101101
.to_edge()
@@ -110,7 +110,7 @@ def test_partition_lifted_scalar_tensor_tosa_BI(self, op, data, arg1):
110110
ArmTester(
111111
LiftedScalarTensor(op, arg1),
112112
example_inputs=(data),
113-
compile_spec=common.get_tosa_compile_spec(),
113+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
114114
)
115115
.quantize()
116116
.export()

‎backends/arm/test/models/test_mobilenet_v2_arm.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,9 @@ def test_mv2_tosa_MI(self):
5454
ArmTester(
5555
self.mv2,
5656
example_inputs=self.model_inputs,
57-
compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
57+
compile_spec=common.get_tosa_compile_spec(
58+
"TOSA-0.80.0+MI", permute_memory_to_nhwc=True
59+
),
5860
)
5961
.export()
6062
.to_edge(config=self._edge_compile_config)
@@ -69,7 +71,9 @@ def test_mv2_tosa_BI(self):
6971
ArmTester(
7072
self.mv2,
7173
example_inputs=self.model_inputs,
72-
compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
74+
compile_spec=common.get_tosa_compile_spec(
75+
"TOSA-0.80.0+BI", permute_memory_to_nhwc=True
76+
),
7377
)
7478
.quantize()
7579
.export()

‎backends/arm/test/ops/test_add.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def _test_add_tosa_MI_pipeline(
6161
ArmTester(
6262
module,
6363
example_inputs=test_data,
64-
compile_spec=common.get_tosa_compile_spec(),
64+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
6565
)
6666
.export()
6767
.check_count({"torch.ops.aten.add.Tensor": 1})
@@ -80,7 +80,7 @@ def _test_add_tosa_BI_pipeline(
8080
ArmTester(
8181
module,
8282
example_inputs=test_data,
83-
compile_spec=common.get_tosa_compile_spec(),
83+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
8484
)
8585
.quantize()
8686
.export()

‎backends/arm/test/ops/test_avg_pool.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,9 @@ def _test_avgpool2d_tosa_MI_pipeline(
5555
ArmTester(
5656
module,
5757
example_inputs=test_data,
58-
compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
58+
compile_spec=common.get_tosa_compile_spec(
59+
"TOSA-0.80.0+MI", permute_memory_to_nhwc=True
60+
),
5961
)
6062
.export()
6163
.check(["torch.ops.aten.avg_pool2d.default"])
@@ -76,7 +78,9 @@ def _test_avgpool2d_tosa_BI_pipeline(
7678
ArmTester(
7779
module,
7880
example_inputs=test_data,
79-
compile_spec=common.get_tosa_compile_spec(permute_memory_to_nhwc=True),
81+
compile_spec=common.get_tosa_compile_spec(
82+
"TOSA-0.80.0+BI", permute_memory_to_nhwc=True
83+
),
8084
)
8185
.quantize(Quantize(quantizer, get_symmetric_quantization_config()))
8286
.export()

‎backends/arm/test/ops/test_batch_norm.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -533,7 +533,7 @@ def _test_batchnorm2d_tosa_MI_pipeline(
533533
ArmTester(
534534
module,
535535
example_inputs=test_data,
536-
compile_spec=common.get_tosa_compile_spec(),
536+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
537537
)
538538
.export()
539539
.check_not(["torch.ops.quantized_decomposed"])
@@ -561,7 +561,7 @@ def _test_batchnorm2d_no_stats_tosa_MI_pipeline(
561561
ArmTester(
562562
module,
563563
example_example_inputs=test_data,
564-
compile_spec=common.get_tosa_compile_spec(),
564+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
565565
)
566566
.export()
567567
.check_count({"torch.ops.aten._native_batch_norm_legit.no_stats": 1})
@@ -590,7 +590,7 @@ def _test_batchnorm2d_tosa_BI_pipeline(
590590
ArmTester(
591591
module,
592592
example_inputs=test_data,
593-
compile_spec=common.get_tosa_compile_spec(),
593+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
594594
)
595595
.quantize()
596596
.export()

‎backends/arm/test/ops/test_bmm.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def _test_bmm_tosa_MI_pipeline(
5050
ArmTester(
5151
module,
5252
example_inputs=test_data,
53-
compile_spec=common.get_tosa_compile_spec(),
53+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
5454
)
5555
.export()
5656
.check_count({"torch.ops.aten.bmm.default": 1})
@@ -70,7 +70,7 @@ def _test_bmm_tosa_BI_pipeline(
7070
ArmTester(
7171
module,
7272
example_inputs=test_data,
73-
compile_spec=common.get_tosa_compile_spec(),
73+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
7474
)
7575
.quantize()
7676
.export()

‎backends/arm/test/ops/test_cat.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def _test_cat_tosa_MI_pipeline(
5656
ArmTester(
5757
module,
5858
example_inputs=test_data,
59-
compile_spec=common.get_tosa_compile_spec(),
59+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
6060
)
6161
.export()
6262
.check_count({"torch.ops.aten.cat.default": 1})
@@ -76,7 +76,7 @@ def _test_cat_tosa_BI_pipeline(
7676
ArmTester(
7777
module,
7878
example_inputs=test_data,
79-
compile_spec=common.get_tosa_compile_spec(),
79+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
8080
)
8181
.quantize()
8282
.export()

‎backends/arm/test/ops/test_clone.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def _test_clone_tosa_MI_pipeline(
4747
ArmTester(
4848
module,
4949
example_inputs=test_data,
50-
compile_spec=common.get_tosa_compile_spec(),
50+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+MI"),
5151
)
5252
.export()
5353
.check_count({"torch.ops.aten.clone.default": 1})
@@ -66,7 +66,7 @@ def _test_clone_tosa_BI_pipeline(
6666
ArmTester(
6767
module,
6868
example_inputs=test_data,
69-
compile_spec=common.get_tosa_compile_spec(),
69+
compile_spec=common.get_tosa_compile_spec("TOSA-0.80.0+BI"),
7070
)
7171
.quantize(Quantize(quantizer, get_symmetric_quantization_config()))
7272
.export()

0 commit comments

Comments
 (0)