diff --git a/lib/Conversion/TorchToTosa/TorchToTosa.cpp b/lib/Conversion/TorchToTosa/TorchToTosa.cpp index 196ff31c86368..adc73972a39cf 100644 --- a/lib/Conversion/TorchToTosa/TorchToTosa.cpp +++ b/lib/Conversion/TorchToTosa/TorchToTosa.cpp @@ -6283,11 +6283,12 @@ void torch::populateTorchToTosaConversionIllegalOps(ConversionTarget &target) { AtenAnyOp, AtenSumOp, AtenSqueezeOp, AtenSqueezeDimOp, AtenMatmulOp, AtenMmOp, AtenBmmOp, AtenLinearOp, AtenAdaptiveAvgPool2dOp, AtenMaxPool2dOp, AtenAvgPool2dOp, AtenOnesOp, AtenZerosOp, - AtenFill_ScalarOp, AtenMaskedFillScalarOp, AtenMaskedFillTensorOp, - AtenTanhOp, AtenSigmoidOp, AtenErfOp, AtenHardtanhBackwardOp, AtenReluOp, - AtenLeakyReluOp, AtenArgmaxOp, AtenPowTensorScalarOp, - AtenPowTensorTensorOp, AtenPowScalarOp, AtenRsubScalarOp, - AtenConvolutionOp, ValueTensorLiteralOp, AtenReshapeOp, AtenBatchNormOp, + AtenFill_ScalarOp, AtenFillScalarOp, AtenFillTensorOp, + AtenMaskedFillScalarOp, AtenMaskedFillTensorOp, AtenTanhOp, AtenSigmoidOp, + AtenErfOp, AtenHardtanhBackwardOp, AtenReluOp, AtenLeakyReluOp, + AtenArgmaxOp, AtenPowTensorScalarOp, AtenPowTensorTensorOp, + AtenPowScalarOp, AtenRsubScalarOp, AtenConvolutionOp, + ValueTensorLiteralOp, AtenReshapeOp, AtenBatchNormOp, AtenNativeLayerNormOp, AtenFlattenUsingIntsOp, AtenUnflattenIntOp, AtenPermuteOp, AtenLog2Op, AtenThresholdOp, AtenUnsqueezeOp, AtenContiguousOp, AtenDropoutOp, AtenViewOp, AtenGeluOp, @@ -6299,7 +6300,7 @@ void torch::populateTorchToTosaConversionIllegalOps(ConversionTarget &target) { AtenRemainderScalarOp, AtenRemainderTensorOp, AtenFmodScalarOp, AtenFmodTensorOp, AtenCatOp, AtenSqrtOp, AtenIscloseOp, Aten__InterpolateSizeListScaleListOp, AtenTrilOp, AtenDiagonalOp, - AtenCloneOp>(); + AtenIndexSelectOp, AtenFlipOp, AtenRoundOp, AtenCloneOp>(); } void torch::populateTorchToTosaConversionPatterns(TypeConverter &typeConverter, @@ -6468,8 +6469,9 @@ void torch::populateTorchToTosaConversionPatterns(TypeConverter &typeConverter, #define INSERT_FILL_PATTERN(AtenOp) \ patterns.add>(typeConverter, context); - INSERT_FILL_PATTERN(AtenFillScalarOp); - INSERT_FILL_PATTERN(AtenFillTensorOp); + INSERT_FILL_PATTERN(AtenFill_ScalarOp); + INSERT_FILL_PATTERN(AtenFillScalarOp); + INSERT_FILL_PATTERN(AtenFillTensorOp); #undef INSERT_FILL_PATTERN #define INSERT_MASKED_FILL_PATTERN(AtenOp) \ @@ -6537,9 +6539,9 @@ void torch::populateTorchToTosaConversionPatterns(TypeConverter &typeConverter, INSERT_ATENOP_PATTERN(Aten__InterpolateSizeListScaleListOp); INSERT_ATENOP_PATTERN(AtenTrilOp); INSERT_ATENOP_PATTERN(AtenDiagonalOp); - INSERT_ATENOP_PATTERN(AtenIndexSelectOp); - INSERT_ATENOP_PATTERN(AtenFlipOp); - INSERT_ATENOP_PATTERN(AtenRoundOp); + INSERT_ATENOP_PATTERN(AtenIndexSelectOp); + INSERT_ATENOP_PATTERN(AtenFlipOp); + INSERT_ATENOP_PATTERN(AtenRoundOp); #undef INSERT_ATENOP_PATTERN #define INSERT_CLONE_ATENOP_PATTERN(AtenOp) \