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

fix(compiler): predicate OpenMP loop parallelization of Linalg on the --parallelize flag.

Browse Source
This commit is contained in:
Antoniu Pop
2022-01-31 09:07:03 +00:00
committed by Antoniu Pop
parent f52965e22f
commit ec633d57f6
3 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -296,8 +296,9 @@ CompilerEngine::compile(llvm::SourceMgr &sm, Target target, OptionalLib lib) {
}
// MLIR canonical dialects -> LLVM Dialect
if (mlir::concretelang::pipeline::lowerStdToLLVMDialect(mlirContext, module,
enablePass)
if (mlir::concretelang::pipeline::lowerStdToLLVMDialect(
mlirContext, module, enablePass,
/*parallelizeLoops =*/this->autoParallelize)
.failed()) {
return errorDiag("Failed to lower to LLVM dialect");
}