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(
4040 output : TosaArg ,
4141 is_quant_node : bool ,
4242 ) -> None :
43- # Specification (0.80.0 ) states that input and output types
43+ # Specification (0.80) states that input and output types
4444 # should all be the same
4545 assert inputs [0 ].dtype == inputs [1 ].dtype == output .dtype
4646 # Handle int8 (quantized) and int32
@@ -98,7 +98,7 @@ def define_node(
9898 output : TosaArg ,
9999 is_quant_node : bool ,
100100 ) -> None :
101- # Specification (0.80.0 ) states that input and output types
101+ # Specification (0.80) states that input and output types
102102 # should all be the same
103103 assert inputs [0 ].dtype == inputs [1 ].dtype == output .dtype
104104
Original file line number Diff line number Diff line change @@ -49,7 +49,7 @@ def _test_maximum_tosa_MI_pipeline(
4949 ArmTester (
5050 module ,
5151 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" ),
5353 )
5454 .export ()
5555 .check_count ({"torch.ops.aten.maximum.default" : 1 })
@@ -68,7 +68,7 @@ def _test_maximum_tosa_BI_pipeline(
6868 ArmTester (
6969 module ,
7070 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" ),
7272 )
7373 .quantize ()
7474 .export ()
Original file line number Diff line number Diff line change @@ -49,7 +49,7 @@ def _test_minimum_tosa_MI_pipeline(
4949 ArmTester (
5050 module ,
5151 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" ),
5353 )
5454 .export ()
5555 .check_count ({"torch.ops.aten.minimum.default" : 1 })
@@ -68,7 +68,7 @@ def _test_minimum_tosa_BI_pipeline(
6868 ArmTester (
6969 module ,
7070 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" ),
7272 )
7373 .quantize ()
7474 .export ()
Original file line number Diff line number Diff line change @@ -54,7 +54,7 @@ def test_fold_qdq_pass(self):
5454 ArmTester (
5555 module ,
5656 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" ),
5858 )
5959 .quantize ()
6060 .export ()
You can’t perform that action at this time.
0 commit comments