Skip to content

Arm backend: Add all ops not supported on Ethos-U55 to support-check #9489

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Mar 24, 2025
Merged

Arm backend: Add all ops not supported on Ethos-U55 to support-check #9489

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
151e1e3
Arm backend: Add all ops not supported on Ethos-U55 to support-check
Erik-Lundell Mar 17, 2025
1b99762
Merge branch 'main' into unsupported-u55
Erik-Lundell Mar 21, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 20 additions & 5 deletions backends/arm/operator_support/tosa_supported_operators.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -208,26 +208,41 @@ def is_node_supported(
class EthosU55NotSupported(OperatorSupportBase):
"""
Certain operators are not supported on U55. These are listed in `unsupported_ops`.
The comment mentions the unsupported TOSA operator that the aten operator maps to where it is not obvious.
For unimplemented operators, this is the anticipated mapping, and it might be incorrect.
"""

unsupported_ops = [
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Curious why does this have to be a white-list? As opposed to all - supported?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hm. The current logic is that we have a list of (TOSA-)supported ops, a.k.a. ops that we have implemented. The support on Ethos-U55 is that list minus this list (and some additional constraints, these are just the completely unsupported ops) We don't have an explicit list of ops supported on Ethos-U55. Maybe I am missing your point 😬

exir_ops.edge.aten.any.default,
exir_ops.edge.aten.any.dim,
exir_ops.edge.aten.any.dims,
exir_ops.edge.aten.any.default, # REDUCE_ANY
exir_ops.edge.aten.any.dim, # REDUCE_ANY
exir_ops.edge.aten.any.dims, # REDUCE_ANY
exir_ops.edge.aten.bitwise_and.Tensor,
exir_ops.edge.aten.bitwise_or.Tensor,
exir_ops.edge.aten.bitwise_xor.Tensor,
exir_ops.edge.aten.bitwise_not,
exir_ops.edge.aten.logical_and.default,
exir_ops.edge.aten.logical_or.default,
exir_ops.edge.aten.logical_xor.default,
exir_ops.edge.aten.logical_not.default,
exir_ops.edge.aten.amax.default,
exir_ops.edge.aten.amin.default,
exir_ops.edge.aten.amax.default, # REDUCE_MAX
exir_ops.edge.aten.amin.default, # REDUCE_MIN
exir_ops.edge.aten.eq.Tensor,
exir_ops.edge.aten.ge.Tensor,
exir_ops.edge.aten.gt.Tensor,
exir_ops.edge.aten.le.Tensor,
exir_ops.edge.aten.lt.Tensor,
exir_ops.edge.aten.flip.default, # REVERSE
exir_ops.edge.aten.grid_sampler_2d, # GATHER
exir_ops.edge.aten.scatter.src,
exir_ops.edge.aten.scatter.value,
exir_ops.edge.aten.select_scatter.default,
exir_ops.edge.aten.scatter_reduce.two,
exir_ops.edge.aten.scatter_add.default,
exir_ops.edge.aten.upsample_nearest2d.vec, # RESIZE
exir_ops.edge.aten.upsample_bilinear2d.vec, # RESIZE
exir_ops.edge.aten.reflection_pad1d.default, # REVERSE
exir_ops.edge.aten.reflection_pad2d.default, # REVERSE
exir_ops.edge.aten.reflection_pad3d.default, # REVERSE
]

def __init__(self, reporter: WhyNoPartitionReporter):
Expand Down
Loading