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

test: add GPU end2end tests

Browse Source
This commit is contained in:
youben11
2022-09-22 07:49:21 +01:00
committed by Ayoub Benaissa
parent 2383c5aec3
commit 7cd45d1514
17 changed files with 169 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
compiler/include/concretelang/Conversion/ConcreteToBConcrete/Pass.h
View File

@@ -12,7 +12,7 @@ namespace mlir {
namespace concretelang {
/// Create a pass to convert `Concrete` dialect to `BConcrete` dialect.
std::unique_ptr<OperationPass<ModuleOp>>
createConvertConcreteToBConcretePass(bool loopParallelize, bool useGPU);
createConvertConcreteToBConcretePass(bool loopParallelize, bool emitGPUOps);
} // namespace concretelang
} // namespace mlir

4
compiler/include/concretelang/Dialect/BConcrete/IR/BConcreteOps.td
View File

@@ -155,7 +155,7 @@ def BConcrete_AwaitFutureOp :
// This is a different op in BConcrete just because of the way we are lowering to CAPI
// When the CAPI lowering is detached from bufferization, we can remove this op, and lower
// to the appropriate CAPI (gpu or cpu) depending on the useGPU compilation option
// to the appropriate CAPI (gpu or cpu) depending on the emitGPUOps compilation option
def BConcrete_BootstrapLweGPUBufferOp : BConcrete_Op<"bootstrap_lwe_gpu_buffer"> {
let arguments = (ins
1DTensorOf<[I64]>:$input_ciphertext,
@@ -172,7 +172,7 @@ def BConcrete_BootstrapLweGPUBufferOp : BConcrete_Op<"bootstrap_lwe_gpu_buffer">
// This is a different op in BConcrete just because of the way we are lowering to CAPI
// When the CAPI lowering is detached from bufferization, we can remove this op, and lower
// to the appropriate CAPI (gpu or cpu) depending on the useGPU compilation option
// to the appropriate CAPI (gpu or cpu) depending on the emitGPUOps compilation option
def BConcrete_KeySwitchLweGPUBufferOp : BConcrete_Op<"keyswitch_lwe_gpu_buffer"> {
let arguments = (ins
1DTensorOf<[I64]>:$ciphertext,

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

@@ -55,7 +55,7 @@ struct CompilationOptions {
bool asyncOffload;
bool optimizeConcrete;
/// use GPU during execution by generating GPU operations if possible
bool useGPU;
bool emitGPUOps;
llvm::Optional<std::vector<int64_t>> fhelinalgTileSizes;
llvm::Optional<std::string> clientParametersFuncName;
@@ -66,7 +66,7 @@ struct CompilationOptions {
: v0FHEConstraints(llvm::None), verifyDiagnostics(false),
autoParallelize(false), loopParallelize(false),
dataflowParallelize(false), asyncOffload(false), optimizeConcrete(true),
useGPU(false), clientParametersFuncName(llvm::None),
emitGPUOps(false), clientParametersFuncName(llvm::None),
optimizerConfig(optimizer::DEFAULT_CONFIG){};
CompilationOptions(std::string funcname) : CompilationOptions() {

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

@@ -47,7 +47,7 @@ lowerTFHEToConcrete(mlir::MLIRContext &context, mlir::ModuleOp &module,
mlir::LogicalResult
lowerConcreteToBConcrete(mlir::MLIRContext &context, mlir::ModuleOp &module,
std::function<bool(mlir::Pass *)> enablePass,
bool parallelizeLoops, bool useGPU);
bool parallelizeLoops, bool emitGPUOps);
mlir::LogicalResult
optimizeConcrete(mlir::MLIRContext &context, mlir::ModuleOp &module,