From 2383c5aec3984125a1d3b9977862e477b7620c01 Mon Sep 17 00:00:00 2001 From: youben11 Date: Thu, 22 Sep 2022 07:18:40 +0100 Subject: [PATCH] test: add gpu benchs --- .../tests/end_to_end_benchmarks/end_to_end_benchmark.cpp | 5 +++++ compiler/tests/end_to_end_benchmarks/end_to_end_mlbench.cpp | 5 +++++ 2 files changed, 10 insertions(+) diff --git a/compiler/tests/end_to_end_benchmarks/end_to_end_benchmark.cpp b/compiler/tests/end_to_end_benchmarks/end_to_end_benchmark.cpp index 34ea19afc..615660af4 100644 --- a/compiler/tests/end_to_end_benchmarks/end_to_end_benchmark.cpp +++ b/compiler/tests/end_to_end_benchmarks/end_to_end_benchmark.cpp @@ -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); diff --git a/compiler/tests/end_to_end_benchmarks/end_to_end_mlbench.cpp b/compiler/tests/end_to_end_benchmarks/end_to_end_mlbench.cpp index 981f0bec6..b9a01f91f 100644 --- a/compiler/tests/end_to_end_benchmarks/end_to_end_mlbench.cpp +++ b/compiler/tests/end_to_end_benchmarks/end_to_end_mlbench.cpp @@ -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;