Skip to content

Commit ac7fe2b

Browse files
committed
Revert "Correct TOSA version to 0.80"
This reverts commit 0fb2ccf. Signed-off-by: Digant Desai <[email protected]>
1 parent 11beed1 commit ac7fe2b

File tree

3 files changed

+13
-0
lines changed

3 files changed

+13
-0
lines changed

backends/arm/operators/op_avg_pool2d.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,10 @@
2020
class AvgPool2dVisitor(NodeVisitor):
2121
target = "aten.avg_pool2d.default"
2222

23+
tosa_specs = [
24+
TosaSpecification.create_from_string("TOSA-0.80.0+BI"),
25+
]
26+
2327
def __init__(self, *args):
2428
super().__init__(*args)
2529

backends/arm/operators/op_batch_norm.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,10 @@
2121
class BatchNormVisitor(NodeVisitor):
2222
target = "aten._native_batch_norm_legit_no_training.default"
2323

24+
tosa_specs = [
25+
TosaSpecification.create_from_string("TOSA-0.80.0+MI"),
26+
]
27+
2428
def __init__(self, *args):
2529
super().__init__(*args)
2630

backends/arm/operators/op_div.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,11 @@
2121
class DivVisitor(NodeVisitor):
2222
target = "aten.div.Tensor"
2323

24+
# Only supported for MI
25+
tosa_specs = [
26+
TosaSpecification.create_from_string("TOSA-0.80.0+MI"),
27+
]
28+
2429
def __init__(self, *args):
2530
super().__init__(*args)
2631

0 commit comments

Comments
 (0)