From d93da5bdc0a5de3ff124e0f82f37205be560bafe Mon Sep 17 00:00:00 2001
From: Quentin Bourgerie <bourgerie.quentin@gmail.com>
Date: Fri, 15 Oct 2021 16:38:12 +0200
Subject: [PATCH] enhance(compiler): Handle tensor.extract_slice and
 tensor.insert_slice in the FHE lowering #168

---
 .../Conversion/Utils/TensorOpTypeConversion.h        | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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<GenericTypeConverterPattern<mlir::tensor::ExtractOp>>(
       patterns.getContext(), typeConverter);
   addDynamicallyLegalTypeOp<mlir::tensor::ExtractOp>(target, typeConverter);
+  // ExtractSliceOp
+  patterns.add<GenericTypeConverterPattern<mlir::tensor::ExtractSliceOp>>(
+      patterns.getContext(), typeConverter);
+  addDynamicallyLegalTypeOp<mlir::tensor::ExtractSliceOp>(target,
+                                                          typeConverter);
+  // InsertSliceOp
+  patterns.add<GenericTypeConverterPattern<mlir::tensor::InsertSliceOp>>(
+      patterns.getContext(), typeConverter);
+  addDynamicallyLegalTypeOp<mlir::tensor::InsertSliceOp>(target, typeConverter);
+
+  // FromElementsOp
   patterns.add<GenericTypeConverterPattern<mlir::tensor::FromElementsOp>>(
       patterns.getContext(), typeConverter);
   addDynamicallyLegalTypeOp<mlir::tensor::FromElementsOp>(target,