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

feat: implement maxpool2d operation

Browse Source
This commit is contained in:
Umut
2023-02-17 16:46:46 +01:00
parent 56bdb05be3
commit bc69c87d62
24 changed files with 1873 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
compiler/lib/Support/CompilerEngine.cpp
View File

@@ -293,13 +293,6 @@ CompilerEngine::compile(llvm::SourceMgr &sm, Target target, OptionalLib lib) {
return errorDiag("Transforming FHE boolean ops failed");
}
// Encrypted mul rewriting
if (mlir::concretelang::pipeline::transformHighLevelFHEOps(mlirContext,
module, enablePass)
.failed()) {
return StreamStringError("Rewriting of encrypted mul failed");
}
if (options.chunkIntegers) {
if (mlir::concretelang::pipeline::transformFHEBigInt(
mlirContext, module, enablePass, options.chunkSize,
@@ -344,6 +337,13 @@ CompilerEngine::compile(llvm::SourceMgr &sm, Target target, OptionalLib lib) {
.failed()) {
return errorDiag("Lowering from FHELinalg to FHE failed");
}
if (mlir::concretelang::pipeline::transformHighLevelFHEOps(mlirContext,
module, enablePass)
.failed()) {
return StreamStringError("Rewriting of high level fhe ops failed");
}
if (target == Target::FHE_NO_LINALG)
return std::move(res);