github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 03:55:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

enhance(compiler): Handle tensor.extract_slice and tensor.insert_slice in the FHE lowering #168

Browse Source
This commit is contained in:
Quentin Bourgerie
2021-10-15 16:38:12 +02:00
parent 3e27592cd1
commit d93da5bdc0
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
compiler/include/zamalang/Conversion/Utils/TensorOpTypeConversion.h
View File

@@ -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,