Skip to content

Commit 9805fde

Browse files
Sebastian-LarssonStrycekSimon
authored andcommitted
Arm backend: Remove sin_cos_support.py (pytorch#14370)
The operator_support check always approved sin and cos operators, which means they can be moved to the supported operator lists in tosa_profile_supported_op_lists.py. Signed-off-by: Sebastian Larsson <[email protected]>
1 parent 9b598be commit 9805fde

File tree

3 files changed

+10
-32
lines changed

3 files changed

+10
-32
lines changed

backends/arm/operator_support/__init__.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
pool_2d_support,
1717
reduce_sum_support,
1818
right_shift_support,
19-
sin_cos_support,
2019
slice_copy_support,
2120
to_dim_order_copy_support,
2221
tosa_supported_operators,

backends/arm/operator_support/sin_cos_support.py

Lines changed: 0 additions & 31 deletions
This file was deleted.

backends/arm/operator_support/tosa_profile_supported_op_lists.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,12 @@
22
#
33
# This source code is licensed under the BSD-style license found in the
44
# LICENSE file in the root directory of this source tree.
5+
"""Define TOSA profile support lists for INT and FP.
6+
7+
Expose static sets of EXIR operator overloads used by the TOSA partitioner to
8+
seed positive support checks for different profiles.
9+
10+
"""
511

612
import operator
713
from typing import Final, Set
@@ -24,6 +30,7 @@
2430
exir_ops.edge.aten.bitwise_and.Scalar,
2531
exir_ops.edge.aten.bitwise_or.Scalar,
2632
exir_ops.edge.aten.bitwise_xor.Scalar,
33+
exir_ops.edge.aten.cos.default,
2734
exir_ops.edge.aten.logical_and.default,
2835
exir_ops.edge.aten.logical_or.default,
2936
exir_ops.edge.aten.logical_xor.default,
@@ -113,6 +120,7 @@
113120
torch.ops.aten.scalar_tensor.default,
114121
exir_ops.edge.aten.gelu.default,
115122
exir_ops.edge.aten.alias_copy.default,
123+
exir_ops.edge.aten.sin.default,
116124
exir_ops.edge.aten.sinh.default,
117125
exir_ops.edge.aten.atan.default,
118126
exir_ops.edge.aten.acosh.default,
@@ -147,6 +155,7 @@
147155
exir_ops.edge.aten.cat.default,
148156
exir_ops.edge.aten.ceil.default,
149157
exir_ops.edge.aten.clamp.default,
158+
exir_ops.edge.aten.cos.default,
150159
exir_ops.edge.aten.cumsum.default,
151160
exir_ops.edge.aten.bmm.default,
152161
exir_ops.edge.aten.permute_copy.default,
@@ -223,6 +232,7 @@
223232
torch.ops.aten.scalar_tensor.default,
224233
exir_ops.edge.aten.gelu.default,
225234
exir_ops.edge.aten.alias_copy.default,
235+
exir_ops.edge.aten.sin.default,
226236
exir_ops.edge.aten.sinh.default,
227237
exir_ops.edge.aten.atan.default,
228238
exir_ops.edge.aten.acosh.default,

0 commit comments

Comments
 (0)