github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-08 19:44:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(ci/bench): Generate gpu benchmarks

Browse Source
This commit is contained in:
Quentin Bourgerie
2022-11-29 14:31:08 +01:00
parent 0602fe64ce
commit 71b24b1255
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.github/workflows/benchmark.yml vendored
View File

@@ -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

5
compiler/Makefile
View File

@@ -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