Skip to content

Commit 7d37bbc

Browse files
authored
Arm backend: Add all ops not supported on Ethos-U55 to support-check (#9489)
Note that this includes ops that are not currently implemented. These ops are added for completeness. Signed-off-by: Erik Lundell <erik.lundell@arm.com>
1 parent 8cd1b93 commit 7d37bbc

File tree

1 file changed

+20
-5
lines changed

1 file changed

+20
-5
lines changed

backends/arm/operator_support/tosa_supported_operators.py

Lines changed: 20 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -208,26 +208,41 @@ def is_node_supported(
208208
class EthosU55NotSupported(OperatorSupportBase):
209209
"""
210210
Certain operators are not supported on U55. These are listed in `unsupported_ops`.
211+
The comment mentions the unsupported TOSA operator that the aten operator maps to where it is not obvious.
212+
For unimplemented operators, this is the anticipated mapping, and it might be incorrect.
211213
"""
212214

213215
unsupported_ops = [
214-
exir_ops.edge.aten.any.default,
215-
exir_ops.edge.aten.any.dim,
216-
exir_ops.edge.aten.any.dims,
216+
exir_ops.edge.aten.any.default, # REDUCE_ANY
217+
exir_ops.edge.aten.any.dim, # REDUCE_ANY
218+
exir_ops.edge.aten.any.dims, # REDUCE_ANY
217219
exir_ops.edge.aten.bitwise_and.Tensor,
218220
exir_ops.edge.aten.bitwise_or.Tensor,
219221
exir_ops.edge.aten.bitwise_xor.Tensor,
222+
exir_ops.edge.aten.bitwise_not,
220223
exir_ops.edge.aten.logical_and.default,
221224
exir_ops.edge.aten.logical_or.default,
222225
exir_ops.edge.aten.logical_xor.default,
223226
exir_ops.edge.aten.logical_not.default,
224-
exir_ops.edge.aten.amax.default,
225-
exir_ops.edge.aten.amin.default,
227+
exir_ops.edge.aten.amax.default, # REDUCE_MAX
228+
exir_ops.edge.aten.amin.default, # REDUCE_MIN
226229
exir_ops.edge.aten.eq.Tensor,
227230
exir_ops.edge.aten.ge.Tensor,
228231
exir_ops.edge.aten.gt.Tensor,
229232
exir_ops.edge.aten.le.Tensor,
230233
exir_ops.edge.aten.lt.Tensor,
234+
exir_ops.edge.aten.flip.default, # REVERSE
235+
exir_ops.edge.aten.grid_sampler_2d, # GATHER
236+
exir_ops.edge.aten.scatter.src,
237+
exir_ops.edge.aten.scatter.value,
238+
exir_ops.edge.aten.select_scatter.default,
239+
exir_ops.edge.aten.scatter_reduce.two,
240+
exir_ops.edge.aten.scatter_add.default,
241+
exir_ops.edge.aten.upsample_nearest2d.vec, # RESIZE
242+
exir_ops.edge.aten.upsample_bilinear2d.vec, # RESIZE
243+
exir_ops.edge.aten.reflection_pad1d.default, # REVERSE
244+
exir_ops.edge.aten.reflection_pad2d.default, # REVERSE
245+
exir_ops.edge.aten.reflection_pad3d.default, # REVERSE
231246
]
232247

233248
def __init__(self, reporter: WhyNoPartitionReporter):

0 commit comments

Comments
 (0)