github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-04-17 03:00:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat(compiler): Add new action dump-sdfg

Browse Source 
Add a new action `dump-sdfg` to `concretecompiler` that causes the IR
to be dumpred right after the extraction of SDFG operations.
This commit is contained in:
Andi Drebes
2022-12-01 14:00:46 +01:00
committed by Andi Drebes
parent 3da32560b7
commit ee2f743a78
3 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
compiler/src/main.cpp
View File

@@ -50,6 +50,7 @@ enum Action {
DUMP_CONCRETE,
DUMP_CONCRETEWITHLOOPS,
DUMP_BCONCRETE,
DUMP_SDFG,
DUMP_STD,
DUMP_LLVM_DIALECT,
DUMP_LLVM_IR,
@@ -128,6 +129,8 @@ static llvm::cl::opt<enum Action> action(
llvm::cl::values(
clEnumValN(Action::DUMP_BCONCRETE, "dump-bconcrete",
"Lower to Bufferized Concrete and dump result")),
llvm::cl::values(clEnumValN(Action::DUMP_SDFG, "dump-sdfg",
"Lower to SDFG operations annd dump result")),
llvm::cl::values(clEnumValN(Action::DUMP_STD, "dump-std",
"Lower to std and dump result")),
llvm::cl::values(clEnumValN(Action::DUMP_LLVM_DIALECT, "dump-llvm-dialect",
@@ -489,6 +492,9 @@ mlir::LogicalResult processInputBuffer(
case Action::DUMP_BCONCRETE:
target = mlir::concretelang::CompilerEngine::Target::BCONCRETE;
break;
case Action::DUMP_SDFG:
target = mlir::concretelang::CompilerEngine::Target::SDFG;
break;
case Action::DUMP_STD:
target = mlir::concretelang::CompilerEngine::Target::STD;
break;