diff --git a/compiler/include/zamalang/Conversion/Utils/TensorOpTypeConversion.h b/compiler/include/zamalang/Conversion/Utils/TensorOpTypeConversion.h index c66572636..5f939fce8 100644 --- a/compiler/include/zamalang/Conversion/Utils/TensorOpTypeConversion.h +++ b/compiler/include/zamalang/Conversion/Utils/TensorOpTypeConversion.h @@ -13,9 +13,21 @@ inline void populateWithTensorTypeConverterPatterns(mlir::RewritePatternSet &patterns, mlir::ConversionTarget &target, mlir::TypeConverter &typeConverter) { + // ExtractOp patterns.add>( patterns.getContext(), typeConverter); addDynamicallyLegalTypeOp(target, typeConverter); + // ExtractSliceOp + patterns.add>( + patterns.getContext(), typeConverter); + addDynamicallyLegalTypeOp(target, + typeConverter); + // InsertSliceOp + patterns.add>( + patterns.getContext(), typeConverter); + addDynamicallyLegalTypeOp(target, typeConverter); + + // FromElementsOp patterns.add>( patterns.getContext(), typeConverter); addDynamicallyLegalTypeOp(target,