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

test: add gpu benchs

Browse Source
This commit is contained in:
youben11
2022-09-22 07:18:40 +01:00
committed by Ayoub Benaissa
parent d615ff47f2
commit 2383c5aec3
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
compiler/tests/end_to_end_benchmarks/end_to_end_benchmark.cpp
View File

@@ -135,6 +135,11 @@ static int registerEndToEndTestFromFile(std::string prefix, std::string path,
mlir::concretelang::CompilationOptions loop;
loop.loopParallelize = true;
registe("loop", loop);
#ifdef CONCRETELANG_CUDA_SUPPORT
mlir::concretelang::CompilationOptions gpu;
gpu.useGPU = true;
registe("gpu", gpu);
#endif
// mlir::concretelang::CompilationOptions dataflow;
// dataflow.dataflowParallelize = true;
// registe("dataflow", dataflow);

5
compiler/tests/end_to_end_benchmarks/end_to_end_mlbench.cpp
View File

@@ -136,6 +136,11 @@ static int registerEndToEndTestFromFile(std::string prefix, std::string path,
mlir::concretelang::CompilationOptions loop;
loop.loopParallelize = true;
registe("loop", loop);
#ifdef CONCRETELANG_CUDA_SUPPORT
mlir::concretelang::CompilationOptions gpu;
gpu.useGPU = true;
registe("gpu", gpu);
#endif
#ifdef CONCRETELANG_DATAFLOW_EXECUTION_ENABLED
mlir::concretelang::CompilationOptions dataflow;
dataflow.dataflowParallelize = true;