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

refactor: replace some operands by attrs in bs/ks

Browse Source
This commit is contained in:
youben11
2022-10-11 14:31:03 +01:00
committed by Ayoub Benaissa
parent 0eedbf4dc4
commit ef778ac75b
25 changed files with 224 additions and 247 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -44,6 +44,11 @@
namespace mlir {
namespace concretelang {
// TODO: should be removed when bufferization is not related to CAPI lowering
// Control whether we should call a cpu of gpu function when lowering
// to CAPI
static bool EMIT_GPU_OPS;
bool getEmitGPUOption() { return EMIT_GPU_OPS; }
/// Creates a new compilation context that can be shared across
/// compilation engines and results
@@ -223,9 +228,13 @@ CompilerEngine::compile(llvm::SourceMgr &sm, Target target, OptionalLib lib) {
CompilationResult res(this->compilationContext);
mlir::MLIRContext &mlirContext = *this->compilationContext->getMLIRContext();
CompilationOptions &options = this->compilerOptions;
// enable/disable usage of gpu functions during bufferization
EMIT_GPU_OPS = options.emitGPUOps;
mlir::MLIRContext &mlirContext = *this->compilationContext->getMLIRContext();
if (options.verifyDiagnostics) {
// Only build diagnostics verifier handler if diagnostics should
// be verified in order to avoid diagnostic messages to be
@@ -355,8 +364,7 @@ CompilerEngine::compile(llvm::SourceMgr &sm, Target target, OptionalLib lib) {
// Concrete -> BConcrete
if (mlir::concretelang::pipeline::lowerConcreteToBConcrete(
mlirContext, module, this->enablePass, loopParallelize,
options.emitGPUOps)
mlirContext, module, this->enablePass, loopParallelize)
.failed()) {
return StreamStringError(
"Lowering from Concrete to Bufferized Concrete failed");