chore(compiler/backend-gpu): Enable gpu benchmarks

zama-ai · Dec 10, 2024 · f3d46be · f3d46be
1 parent 2d45a57
commit f3d46be
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/compilers/concrete-compiler/compiler/Makefile b/compilers/concrete-compiler/compiler/Makefile
@@ -391,7 +391,7 @@ $(BENCHMARK_GPU_DIR):
 
 $(BENCHMARK_GPU_DIR)/end_to_end_linalg_apply_lookup_table.yaml: tests/end_to_end_fixture/end_to_end_linalg_apply_lookup_table_gen.py
 	$(Python3_EXECUTABLE) $< \
-	--bitwidth 1 2 3 4 5 6 7 --n-ct 1 128 1024 2048 8192
+	--bitwidth 1 2 3 4 5 6 7 --n-ct 1 128 1024 2048 8192  > $@
 
 
 generate-gpu-benchmarks: $(BENCHMARK_GPU_DIR) $(BENCHMARK_GPU_DIR)/end_to_end_linalg_apply_lookup_table.yaml

diff --git a/compilers/concrete-compiler/compiler/tests/end_to_end_fixture/benchmarks_gpu/cifar-16.yaml b/compilers/concrete-compiler/compiler/tests/end_to_end_fixture/benchmarks_gpu/cifar-16.yaml
@@ -0,0 +1 @@
+../benchmarks_cpu/cifar-16.yaml
diff --git a/...lers/concrete-compiler/compiler/tests/end_to_end_fixture/benchmarks_gpu/levelled_llm.yaml b/...lers/concrete-compiler/compiler/tests/end_to_end_fixture/benchmarks_gpu/levelled_llm.yaml
@@ -0,0 +1 @@
+../benchmarks_cpu/levelled_llm.yaml
diff --git a/...te-compiler/compiler/tests/end_to_end_fixture/end_to_end_linalg_apply_lookup_table_gen.py b/...te-compiler/compiler/tests/end_to_end_fixture/end_to_end_linalg_apply_lookup_table_gen.py
@@ -20,8 +20,8 @@ def get_lut_integer_type(p):
         raise Exception("Unexpected precision") 
 
 def generate(args):
-    print("# /!\ DO NOT EDIT MANUALLY THIS FILE MANUALLY")
-    print("# /!\ THIS FILE HAS BEEN GENERATED")
+    print("# /!\\ DO NOT EDIT MANUALLY THIS FILE MANUALLY")
+    print("# /!\\ THIS FILE HAS BEEN GENERATED")
     np.random.seed(0)
     for n_ct in args.n_ct:
         for p in args.bitwidth: