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(compiler): add a key cache

Browse Source
This commit is contained in:
Mayeul@Zama
2021-11-25 18:26:42 +01:00
committed by mayeul-zama
parent f193fd71a2
commit dad4390518
17 changed files with 533 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
compiler/lib/CAPI/Support/CompilerEngine.cpp
View File

@@ -1,20 +1,31 @@
#include "llvm/ADT/SmallString.h"
#include "zamalang-c/Support/CompilerEngine.h"
#include "zamalang/Support/CompilerEngine.h"
#include "zamalang/Support/Jit.h"
#include "zamalang/Support/JitCompilerEngine.h"
#include "zamalang/Support/KeySetCache.h"
using mlir::zamalang::JitCompilerEngine;
mlir::zamalang::JitCompilerEngine::Lambda
buildLambda(const char *module, const char *funcName,
const char *runtimeLibPath) {
const char *runtimeLibPath, const char *keySetCachePath) {
// Set the runtime library path if not nullptr
llvm::Optional<llvm::StringRef> runtimeLibPathOptional = {};
if (runtimeLibPath != nullptr)
runtimeLibPathOptional = runtimeLibPath;
mlir::zamalang::JitCompilerEngine engine;
using KeySetCache = mlir::zamalang::KeySetCache;
using optKeySetCache = llvm::Optional<mlir::zamalang::KeySetCache>;
auto cacheOpt = optKeySetCache();
if (keySetCachePath != nullptr) {
cacheOpt = KeySetCache(std::string(keySetCachePath));
}
llvm::Expected<mlir::zamalang::JitCompilerEngine::Lambda> lambdaOrErr =
engine.buildLambda(module, funcName, runtimeLibPathOptional);
engine.buildLambda(module, funcName, cacheOpt, runtimeLibPathOptional);
if (!lambdaOrErr) {
std::string backingString;
llvm::raw_string_ostream os(backingString);