From 71b24b1255befd117f3fcee6d7e7fee4bf6c8912 Mon Sep 17 00:00:00 2001 From: Quentin Bourgerie Date: Tue, 29 Nov 2022 14:31:08 +0100 Subject: [PATCH] chore(ci/bench): Generate gpu benchmarks --- .github/workflows/benchmark.yml | 6 +++--- compiler/Makefile | 5 +++++ 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index 52074fbf9..cd30f29b5 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -141,12 +141,12 @@ jobs: cd compiler make BINDINGS_PYTHON_ENABLED=OFF build-benchmarks generate-cpu-benchmarks - - name: Build compiler and run end-to-end benchmarks (GPU) + - name: Build compiler and end-to-end benchmarks (GPU) if: ${{ inputs.backend == 'gpu' }} run: | set -e cd compiler - make BINDINGS_PYTHON_ENABLED=OFF CUDA_SUPPORT=ON build-benchmarks generate-benchmarks + make BINDINGS_PYTHON_ENABLED=OFF CUDA_SUPPORT=ON build-benchmarks generate-gpu-benchmarks - name: Download KeySetCache if: ${{ !contains(github.head_ref, 'newkeysetcache') }} @@ -159,7 +159,7 @@ jobs: run: | touch keysetcache.timestamp - - name: Build compiler and run end-to-end benchmarks + - name: Run end-to-end benchmarks run: | set -e cd compiler diff --git a/compiler/Makefile b/compiler/Makefile index 1cf28fbfc..9b80181a4 100644 --- a/compiler/Makefile +++ b/compiler/Makefile @@ -293,6 +293,11 @@ $(BENCHMARK_CPU_DIR): generate-cpu-benchmarks: $(BENCHMARK_CPU_DIR) $(BENCHMARK_CPU_DIR)/end_to_end_linalg_apply_lookup_table.yaml $(BENCHMARK_CPU_DIR)/end_to_end_apply_lookup_table.yaml $(BENCHMARK_CPU_DIR)/end_to_end_leveled.yaml +generate-gpu-benchmarks: + $(Python3_EXECUTABLE) ./tests/end_to_end_fixture/end_to_end_linalg_apply_lookup_table_gen.py \ + --min_bitwidth 1 --max_bitwidth 7 --n_ct 1 128 1024 2048 8192 \ + > tests/end_to_end_fixture/end_to_end_linalg_apply_lookup_table.yaml + build-benchmarks: build-initialized cmake --build $(BUILD_DIR) --target end_to_end_benchmark