File tree Expand file tree Collapse file tree 4 files changed +7
-7
lines changed Expand file tree Collapse file tree 4 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -40,7 +40,7 @@ def define_node(
40
40
output : TosaArg ,
41
41
is_quant_node : bool ,
42
42
) -> None :
43
- # Specification (0.80.0 ) states that input and output types
43
+ # Specification (0.80) states that input and output types
44
44
# should all be the same
45
45
assert inputs [0 ].dtype == inputs [1 ].dtype == output .dtype
46
46
# Handle int8 (quantized) and int32
@@ -98,7 +98,7 @@ def define_node(
98
98
output : TosaArg ,
99
99
is_quant_node : bool ,
100
100
) -> None :
101
- # Specification (0.80.0 ) states that input and output types
101
+ # Specification (0.80) states that input and output types
102
102
# should all be the same
103
103
assert inputs [0 ].dtype == inputs [1 ].dtype == output .dtype
104
104
Original file line number Diff line number Diff line change @@ -49,7 +49,7 @@ def _test_maximum_tosa_MI_pipeline(
49
49
ArmTester (
50
50
module ,
51
51
example_inputs = test_data ,
52
- compile_spec = common .get_tosa_compile_spec ("TOSA-0.80.0 +MI" ),
52
+ compile_spec = common .get_tosa_compile_spec ("TOSA-0.80+MI" ),
53
53
)
54
54
.export ()
55
55
.check_count ({"torch.ops.aten.maximum.default" : 1 })
@@ -68,7 +68,7 @@ def _test_maximum_tosa_BI_pipeline(
68
68
ArmTester (
69
69
module ,
70
70
example_inputs = test_data ,
71
- compile_spec = common .get_tosa_compile_spec ("TOSA-0.80.0 +BI" ),
71
+ compile_spec = common .get_tosa_compile_spec ("TOSA-0.80+BI" ),
72
72
)
73
73
.quantize ()
74
74
.export ()
Original file line number Diff line number Diff line change @@ -49,7 +49,7 @@ def _test_minimum_tosa_MI_pipeline(
49
49
ArmTester (
50
50
module ,
51
51
example_inputs = test_data ,
52
- compile_spec = common .get_tosa_compile_spec ("TOSA-0.80.0 +MI" ),
52
+ compile_spec = common .get_tosa_compile_spec ("TOSA-0.80+MI" ),
53
53
)
54
54
.export ()
55
55
.check_count ({"torch.ops.aten.minimum.default" : 1 })
@@ -68,7 +68,7 @@ def _test_minimum_tosa_BI_pipeline(
68
68
ArmTester (
69
69
module ,
70
70
example_inputs = test_data ,
71
- compile_spec = common .get_tosa_compile_spec ("TOSA-0.80.0 +BI" ),
71
+ compile_spec = common .get_tosa_compile_spec ("TOSA-0.80+BI" ),
72
72
)
73
73
.quantize ()
74
74
.export ()
Original file line number Diff line number Diff line change @@ -54,7 +54,7 @@ def test_fold_qdq_pass(self):
54
54
ArmTester (
55
55
module ,
56
56
example_inputs = module .get_inputs (),
57
- compile_spec = common .get_tosa_compile_spec ("TOSA-0.80.0 +BI" ),
57
+ compile_spec = common .get_tosa_compile_spec ("TOSA-0.80+BI" ),
58
58
)
59
59
.quantize ()
60
60
.export ()
You can’t perform that action at this time.
0 commit comments