From 735e78c141fec9d2e96a953db87cde2747b4761d Mon Sep 17 00:00:00 2001 From: Quentin Bourgerie Date: Tue, 29 Nov 2022 14:37:32 +0100 Subject: [PATCH] chore(bench): Use batched concrete operators for gpu benchmarks --- compiler/tests/end_to_end_benchmarks/end_to_end_benchmark.cpp | 1 + 1 file changed, 1 insertion(+) 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 d3196e3cb..abb6bf896 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 @@ -145,6 +145,7 @@ static int registerEndToEndTestFromFile(std::string prefix, std::string path, cpu.loopParallelize = true; #else mlir::concretelang::CompilationOptions gpu; + gpu.batchConcreteOps = true; gpu.emitGPUOps = true; gpu.loopParallelize = true; registe("gpu", gpu);