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

fix(compiler): Remove custom llvm pipeline as it's not used

Browse Source
This commit is contained in:
Quentin Bourgerie
2022-03-22 12:36:20 +01:00
parent cd2280ccae
commit 1422b575d0
2 changed files with 5 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
compiler/include/concretelang/Support/JitLambdaSupport.h
View File

@@ -32,10 +32,7 @@ class JitLambdaSupport
: public LambdaSupport<concretelang::JITLambda *, JitCompilationResult> {
public:
JitLambdaSupport(
llvm::Optional<llvm::StringRef> runtimeLibPath = llvm::None,
llvm::function_ref<llvm::Error(llvm::Module *)> llvmOptPipeline =
mlir::makeOptimizingTransformer(3, 0, nullptr));
JitLambdaSupport(llvm::Optional<llvm::StringRef> runtimeLibPath = llvm::None);
llvm::Expected<std::unique_ptr<JitCompilationResult>>
compile(llvm::SourceMgr &program, CompilationOptions options) override;

13
compiler/lib/Support/JitLambdaSupport.cpp
View File

@@ -11,20 +11,17 @@ namespace mlir {
namespace concretelang {
JitLambdaSupport::JitLambdaSupport(
llvm::Optional<llvm::StringRef> runtimeLibPath,
llvm::function_ref<llvm::Error(llvm::Module *)> llvmOptPipeline)
: runtimeLibPath(runtimeLibPath), llvmOptPipeline(llvmOptPipeline) {}
llvm::Optional<llvm::StringRef> runtimeLibPath)
: runtimeLibPath(runtimeLibPath) {}
llvm::Expected<std::unique_ptr<JitCompilationResult>>
JitLambdaSupport::compile(llvm::SourceMgr &program,
CompilationOptions options) {
// Setup the compiler engine
auto context = std::make_shared<CompilationContext>();
concretelang::CompilerEngine engine(context);
engine.setCompilationOptions(options);
// Compile to LLVM Dialect
auto compilationResult =
engine.compile(program, CompilerEngine::Target::LLVM_IR);
@@ -36,16 +33,14 @@ JitLambdaSupport::compile(llvm::SourceMgr &program,
if (!options.clientParametersFuncName.hasValue()) {
return StreamStringError("Need to have a funcname to JIT compile");
}
// Compile from LLVM Dialect to JITLambda
auto mlirModule = compilationResult.get().mlirModuleRef->get();
auto lambda = concretelang::JITLambda::create(
*options.clientParametersFuncName, mlirModule, llvmOptPipeline,
runtimeLibPath);
*options.clientParametersFuncName, mlirModule,
mlir::makeOptimizingTransformer(3, 0, nullptr), runtimeLibPath);
if (auto err = lambda.takeError()) {
return std::move(err);
}
if (!compilationResult.get().clientParameters.hasValue()) {
// i.e. that should not occurs
return StreamStringError("No client parameters has been generated");