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

feat(compiler): Add support for tensor.{from_elements,dim} operations in TFHE passes

Browse Source
This commit is contained in:
Andi Drebes
2024-04-03 11:02:41 +02:00
parent 0c7e3a3518
commit 48d919bd25
4 changed files with 34 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
compilers/concrete-compiler/compiler/lib/Conversion/TFHEKeyNormalization/TFHEKeyNormalization.cpp
View File

@@ -371,6 +371,12 @@ void TFHEKeyNormalizationPass::runOnOperation() {
mlir::concretelang::addDynamicallyLegalTypeOp<mlir::tensor::EmptyOp>(
target, typeConverter);
patterns.add<
mlir::concretelang::GenericTypeConverterPattern<mlir::tensor::DimOp>>(
&getContext(), typeConverter);
mlir::concretelang::addDynamicallyLegalTypeOp<mlir::tensor::DimOp>(
target, typeConverter);
patterns.add<RegionOpTypeConverterPattern<mlir::linalg::GenericOp,
conversion::TypeConverter>>(
&getContext(), typeConverter);