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

feat: support GPU (bootstrapping)

Browse Source
This commit is contained in:
youben11
2022-07-21 14:45:28 +01:00
committed by Ayoub Benaissa
parent a487b03699
commit d169a27fc0
26 changed files with 715 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
compiler/src/main.cpp
View File

@@ -98,6 +98,12 @@ llvm::cl::opt<bool>
"dialects. (Enabled by default)"),
llvm::cl::init<bool>(true));
llvm::cl::opt<bool>
useGPU("use-gpu",
llvm::cl::desc("enable/disable generating concrete GPU "
"operations (Disabled by default)"),
llvm::cl::init<bool>(false));
llvm::cl::list<std::string> passes(
"passes",
llvm::cl::desc("Specify the passes to run (use only for compiler tests)"),
@@ -283,6 +289,7 @@ cmdlineCompilationOptions() {
options.loopParallelize = cmdline::loopParallelize;
options.dataflowParallelize = cmdline::dataflowParallelize;
options.optimizeConcrete = cmdline::optimizeConcrete;
options.useGPU = cmdline::useGPU;
if (!cmdline::v0Constraint.empty()) {
if (cmdline::v0Constraint.size() != 2) {