@@ -208,26 +208,41 @@ def is_node_supported(
208
208
class EthosU55NotSupported (OperatorSupportBase ):
209
209
"""
210
210
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.
211
213
"""
212
214
213
215
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
217
219
exir_ops .edge .aten .bitwise_and .Tensor ,
218
220
exir_ops .edge .aten .bitwise_or .Tensor ,
219
221
exir_ops .edge .aten .bitwise_xor .Tensor ,
222
+ exir_ops .edge .aten .bitwise_not ,
220
223
exir_ops .edge .aten .logical_and .default ,
221
224
exir_ops .edge .aten .logical_or .default ,
222
225
exir_ops .edge .aten .logical_xor .default ,
223
226
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
226
229
exir_ops .edge .aten .eq .Tensor ,
227
230
exir_ops .edge .aten .ge .Tensor ,
228
231
exir_ops .edge .aten .gt .Tensor ,
229
232
exir_ops .edge .aten .le .Tensor ,
230
233
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
231
246
]
232
247
233
248
def __init__ (self , reporter : WhyNoPartitionReporter ):
0 commit comments