diff --git a/.github/workflows/gpu_4090_full_benchmark.yml b/.github/workflows/gpu_4090_benchmark.yml similarity index 93% rename from .github/workflows/gpu_4090_full_benchmark.yml rename to .github/workflows/gpu_4090_benchmark.yml index eafd63c912..fe349bcb94 100644 --- a/.github/workflows/gpu_4090_full_benchmark.yml +++ b/.github/workflows/gpu_4090_benchmark.yml @@ -1,5 +1,5 @@ -# Run all benchmarks on an RTX 4090 machine and return parsed results to Slab CI bot. -name: TFHE Cuda Backend - 4090 full benchmarks +# Run benchmarks on an RTX 4090 machine and return parsed results to Slab CI bot. +name: TFHE Cuda Backend - 4090 benchmarks env: CARGO_TERM_COLOR: always @@ -23,7 +23,7 @@ on: jobs: cuda-integer-benchmarks: - name: Cuda integer benchmarks for all operations flavor (RTX 4090) + name: Cuda integer benchmarks (RTX 4090) if: ${{ github.event_name == 'workflow_dispatch' || github.event_name == 'schedule' && github.repository == 'zama-ai/tfhe-rs' || contains(github.event.label.name, '4090_bench') }} @@ -35,9 +35,6 @@ jobs: strategy: fail-fast: false max-parallel: 1 - matrix: - command: [integer, integer_multi_bit] - op_flavor: [default, unchecked] steps: - name: Checkout tfhe-rs @@ -52,6 +49,7 @@ jobs: echo "COMMIT_DATE=$(git --no-pager show -s --format=%cd --date=iso8601-strict ${{ github.sha }})"; echo "COMMIT_HASH=$(git describe --tags --dirty)"; } >> "${GITHUB_ENV}" + echo "FAST_BENCH=TRUE" >> "${GITHUB_ENV}" - name: Install rust uses: dtolnay/rust-toolchain@21dc36fb71dd22e3317045c0c31a3f4249868b17 @@ -67,7 +65,7 @@ jobs: - name: Run integer benchmarks run: | - make BENCH_OP_FLAVOR=${{ matrix.op_flavor }} bench_${{ matrix.command }}_gpu + make BENCH_OP_FLAVOR=default bench_integer_multi_bit_gpu - name: Parse results run: | @@ -85,7 +83,7 @@ jobs: - name: Upload parsed results artifact uses: actions/upload-artifact@0b2256b8c012f0828dc542b3febcab082c67f72b with: - name: ${{ github.sha }}_${{ matrix.command }}_${{ matrix.op_flavor }} + name: ${{ github.sha }}_integer_multi_bit_gpu_default path: ${{ env.RESULTS_FILENAME }} - name: Send data to Slab @@ -146,7 +144,7 @@ jobs: path: slab token: ${{ secrets.FHE_ACTIONS_TOKEN }} - - name: Run integer benchmarks + - name: Run core crypto benchmarks run: | make bench_pbs_gpu make bench_ks_gpu