github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-10 07:08:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(gpu): run only unsigned bench in multi-bit GPU workflows

Browse Source
This commit is contained in:
Agnes Leroy
2024-06-27 14:50:46 +02:00
committed by Agnès Leroy
parent 773adcc26f
commit b06beabfa2
3 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/integer_multi_bit_benchmark.yml vendored
View File

@@ -62,7 +62,7 @@ jobs:
- name: Run multi-bit benchmarks with AVX512
run: |
make FAST_BENCH=TRUE bench_integer_multi_bit
make FAST_BENCH=TRUE bench_unsigned_integer_multi_bit
- name: Parse benchmarks to csv
run: |

2
.github/workflows/integer_multi_bit_multi_gpu_benchmark.yml vendored
View File

@@ -119,7 +119,7 @@ jobs:
- name: Run multi-bit benchmarks with AVX512
run: |
make FAST_BENCH=${{ env.FAST_BENCH }} BENCH_OP_FLAVOR=default bench_integer_multi_bit_gpu
make FAST_BENCH=${{ env.FAST_BENCH }} BENCH_OP_FLAVOR=default bench_unsigned_integer_multi_bit_gpu
- name: Parse results
run: |

8
Makefile
View File

@@ -824,6 +824,14 @@ bench_integer_multi_bit_gpu: install_rs_check_toolchain
--bench integer-bench \
--features=$(TARGET_ARCH_FEATURE),integer,gpu,internal-keycache,nightly-avx512 -p $(TFHE_SPEC) --
.PHONY: bench_unsigned_integer_multi_bit_gpu # Run benchmarks for unsigned integer on GPU backend using multi-bit parameters
bench_unsigned_integer_multi_bit_gpu: install_rs_check_toolchain
RUSTFLAGS="$(RUSTFLAGS)" __TFHE_RS_BENCH_TYPE=MULTI_BIT \
__TFHE_RS_BENCH_OP_FLAVOR=$(BENCH_OP_FLAVOR) __TFHE_RS_FAST_BENCH=$(FAST_BENCH) \
cargo $(CARGO_RS_CHECK_TOOLCHAIN) bench \
--bench integer-bench \
--features=$(TARGET_ARCH_FEATURE),integer,gpu,internal-keycache,nightly-avx512 -p $(TFHE_SPEC) -- unsigned
.PHONY: bench_integer_zk # Run benchmarks for integer encryption with ZK proofs
bench_integer_zk: install_rs_check_toolchain
RUSTFLAGS="$(RUSTFLAGS)" \