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 support for full unrolling of loops with SDFG-convertible ops

Browse Source 
This adds a new option `--unroll-loops-with-sdfg-convertible-ops`,
which causes loops containing SDFG-convertible operations to be fully
unrolled upon the extraction of SDFG-operations using the
`--emit-sdfg-ops` switch. This avoids constant roundtrips between an
SDFG-capable accelerator and the host during execution of a loop.

The option is limited to `scf.for` loops with static bounds and a
static step size. Since full unrolling of loops with large bounds
results in a large number of operations, the option is disabled by
default.
This commit is contained in:
Andi Drebes
2022-12-07 10:26:29 +01:00
committed by Andi Drebes
parent 2dce654406
commit e2e6df322e
9 changed files with 160 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
compiler/include/concretelang/Conversion/ExtractSDFGOps/Pass.h
View File

@@ -11,7 +11,8 @@
namespace mlir {
namespace concretelang {
std::unique_ptr<OperationPass<mlir::func::FuncOp>> createExtractSDFGOpsPass();
std::unique_ptr<OperationPass<mlir::func::FuncOp>>
createExtractSDFGOpsPass(bool unroll);
} // namespace concretelang
} // namespace mlir

6
compiler/include/concretelang/Support/CompilerEngine.h
View File

@@ -60,6 +60,7 @@ struct CompilationOptions {
bool loopParallelize;
bool batchConcreteOps;
bool emitSDFGOps;
bool unrollLoopsWithSDFGConvertibleOps;
bool dataflowParallelize;
bool optimizeConcrete;
/// use GPU during execution by generating GPU operations if possible
@@ -73,8 +74,9 @@ struct CompilationOptions {
CompilationOptions()
: v0FHEConstraints(llvm::None), verifyDiagnostics(false),
autoParallelize(false), loopParallelize(false), batchConcreteOps(false),
emitSDFGOps(false), dataflowParallelize(false), optimizeConcrete(true),
emitGPUOps(false), clientParametersFuncName(llvm::None),
emitSDFGOps(false), unrollLoopsWithSDFGConvertibleOps(false),
dataflowParallelize(false), optimizeConcrete(true), emitGPUOps(false),
clientParametersFuncName(llvm::None),
optimizerConfig(optimizer::DEFAULT_CONFIG){};
CompilationOptions(std::string funcname) : CompilationOptions() {

7
compiler/include/concretelang/Support/Pipeline.h
View File

@@ -58,9 +58,10 @@ mlir::LogicalResult
optimizeConcrete(mlir::MLIRContext &context, mlir::ModuleOp &module,
std::function<bool(mlir::Pass *)> enablePass);
mlir::LogicalResult
extractSDFGOps(mlir::MLIRContext &context, mlir::ModuleOp &module,
std::function<bool(mlir::Pass *)> enablePass);
mlir::LogicalResult extractSDFGOps(mlir::MLIRContext &context,
mlir::ModuleOp &module,
std::function<bool(mlir::Pass *)> enablePass,
bool unrollLoops);
mlir::LogicalResult
lowerBConcreteToStd(mlir::MLIRContext &context, mlir::ModuleOp &module,