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(test): No more use the JitCompilationEngine

Browse Source
This commit is contained in:
Quentin Bourgerie
2022-03-17 14:40:27 +01:00
parent 5b83b700d2
commit fc996fc698
1 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
compiler/tests/unittest/end_to_end_jit_test.h
View File

@@ -100,7 +100,7 @@ static bool assert_expected_value(llvm::Expected<T> &&val, const V &exp) {
}; \
ASSERT_EQ(val.value(), exp);
static inline llvm::Optional<mlir::concretelang::KeySetCache>
static inline llvm::Optional<concretelang::clientlib::KeySetCache>
getTestKeySetCache() {
llvm::SmallString<0> cachePath;
@@ -108,36 +108,38 @@ getTestKeySetCache() {
llvm::sys::path::append(cachePath, "KeySetCache");
auto cachePathStr = std::string(cachePath);
return llvm::Optional<mlir::concretelang::KeySetCache>(
mlir::concretelang::KeySetCache(cachePathStr));
return llvm::Optional<concretelang::clientlib::KeySetCache>(
concretelang::clientlib::KeySetCache(cachePathStr));
}
static inline std::shared_ptr<mlir::concretelang::KeySetCache>
static inline std::shared_ptr<concretelang::clientlib::KeySetCache>
getTestKeySetCachePtr() {
return std::make_shared<mlir::concretelang::KeySetCache>(
return std::make_shared<concretelang::clientlib::KeySetCache>(
getTestKeySetCache().getValue());
}
// Jit-compiles the function specified by `func` from `src` and
// returns the corresponding lambda. Any compilation errors are caught
// and reult in abnormal termination.
inline llvm::Expected<mlir::concretelang::JitCompilerEngine::Lambda>
inline llvm::Expected<
mlir::concretelang::ClientServer<mlir::concretelang::JitLambdaSupport>>
internalCheckedJit(llvm::StringRef src, llvm::StringRef func = "main",
bool useDefaultFHEConstraints = false,
bool autoParallelize = false) {
mlir::concretelang::JitCompilerEngine engine;
auto options =
mlir::concretelang::CompilationOptions(std::string(func.data()));
if (useDefaultFHEConstraints)
engine.setFHEConstraints(defaultV0Constraints);
options.v0FHEConstraints = defaultV0Constraints;
#ifdef CONCRETELANG_PARALLEL_TESTING_ENABLED
engine.setAutoParallelize(true);
options.autoParallelize = true;
#else
engine.setAutoParallelize(autoParallelize);
options.autoParallelize = autoParallelize;
#endif
llvm::Expected<mlir::concretelang::JitCompilerEngine::Lambda> lambdaOrErr =
engine.buildLambda(src, func, getTestKeySetCache());
auto lambdaOrErr =
mlir::concretelang::ClientServer<mlir::concretelang::JitLambdaSupport>::
create(src, options, getTestKeySetCache(),
mlir::concretelang::JitLambdaSupport());
return lambdaOrErr;
}