@@ -208,26 +208,41 @@ def is_node_supported(
208208class 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