|
65 | 65 | run: bash scripts/bench-taskset.sh cargo codspeed run |
66 | 66 | token: ${{ secrets.CODSPEED_TOKEN }} |
67 | 67 | mode: "simulation" |
| 68 | + |
| 69 | + bench-codspeed-cuda: |
| 70 | + if: github.repository == 'vortex-data/vortex' |
| 71 | + strategy: |
| 72 | + matrix: |
| 73 | + include: |
| 74 | + - { shard: 1, name: "Bitpacked", benches: "bitpacked_cuda" } |
| 75 | + - { shard: 2, name: "Dynamic dispatch", benches: "dynamic_dispatch_cuda" } |
| 76 | + - { shard: 3, name: "Standalone kernels", benches: "alp_cuda date_time_parts_cuda dict_cuda for_cuda runend_cuda throughput_cuda" } |
| 77 | + - { shard: 4, name: "NVIDIA kernels", benches: "filter_cuda zstd_cuda" } |
| 78 | + name: "Benchmark with Codspeed (CUDA Shard #${{ matrix.shard }} - ${{ matrix.name }})" |
| 79 | + timeout-minutes: 30 |
| 80 | + runs-on: runs-on=${{ github.run_id }}/family=g5/image=ubuntu24-gpu-x64/tag=bench-codspeed-cuda-${{ matrix.shard }} |
| 81 | + steps: |
| 82 | + - uses: runs-on/action@v2 |
| 83 | + with: |
| 84 | + sccache: s3 |
| 85 | + - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6 |
| 86 | + - uses: ./.github/actions/setup-rust |
| 87 | + with: |
| 88 | + repo-token: ${{ secrets.GITHUB_TOKEN }} |
| 89 | + - name: Display NVIDIA SMI details |
| 90 | + run: | |
| 91 | + nvidia-smi |
| 92 | + nvidia-smi -L |
| 93 | + nvidia-smi -q -d Memory |
| 94 | + - name: Install Codspeed |
| 95 | + uses: taiki-e/cache-cargo-install-action@66c9585ef5ca780ee69399975a5e911f47905995 |
| 96 | + with: |
| 97 | + tool: cargo-codspeed |
| 98 | + - name: Build benchmarks |
| 99 | + run: cargo codspeed build -m walltime -p vortex-cuda --profile bench |
| 100 | + - name: Run benchmarks |
| 101 | + uses: CodSpeedHQ/action@d872884a306dd4853acf0f584f4b706cf0cc72a2 |
| 102 | + env: |
| 103 | + CARGO_MANIFEST_DIR: ${{ github.workspace }}/vortex-cuda |
| 104 | + with: |
| 105 | + run: cargo codspeed run $(printf -- '--bench %s ' ${{ matrix.benches }}) |
| 106 | + token: ${{ secrets.CODSPEED_TOKEN }} |
| 107 | + mode: "walltime" |
0 commit comments