|
2 | 2 | #
|
3 | 3 | # This source code is licensed under the BSD-style license found in the
|
4 | 4 | # 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 | +""" |
5 | 11 |
|
6 | 12 | import operator
|
7 | 13 | from typing import Final, Set
|
|
24 | 30 | exir_ops.edge.aten.bitwise_and.Scalar,
|
25 | 31 | exir_ops.edge.aten.bitwise_or.Scalar,
|
26 | 32 | exir_ops.edge.aten.bitwise_xor.Scalar,
|
| 33 | + exir_ops.edge.aten.cos.default, |
27 | 34 | exir_ops.edge.aten.logical_and.default,
|
28 | 35 | exir_ops.edge.aten.logical_or.default,
|
29 | 36 | exir_ops.edge.aten.logical_xor.default,
|
|
113 | 120 | torch.ops.aten.scalar_tensor.default,
|
114 | 121 | exir_ops.edge.aten.gelu.default,
|
115 | 122 | exir_ops.edge.aten.alias_copy.default,
|
| 123 | + exir_ops.edge.aten.sin.default, |
116 | 124 | exir_ops.edge.aten.sinh.default,
|
117 | 125 | exir_ops.edge.aten.atan.default,
|
118 | 126 | exir_ops.edge.aten.acosh.default,
|
|
147 | 155 | exir_ops.edge.aten.cat.default,
|
148 | 156 | exir_ops.edge.aten.ceil.default,
|
149 | 157 | exir_ops.edge.aten.clamp.default,
|
| 158 | + exir_ops.edge.aten.cos.default, |
150 | 159 | exir_ops.edge.aten.cumsum.default,
|
151 | 160 | exir_ops.edge.aten.bmm.default,
|
152 | 161 | exir_ops.edge.aten.permute_copy.default,
|
|
223 | 232 | torch.ops.aten.scalar_tensor.default,
|
224 | 233 | exir_ops.edge.aten.gelu.default,
|
225 | 234 | exir_ops.edge.aten.alias_copy.default,
|
| 235 | + exir_ops.edge.aten.sin.default, |
226 | 236 | exir_ops.edge.aten.sinh.default,
|
227 | 237 | exir_ops.edge.aten.atan.default,
|
228 | 238 | exir_ops.edge.aten.acosh.default,
|
|
0 commit comments