Skip to content

Commit 1db729d

Browse files
authored
Create CI workflows for GPU benchmarks (#298)
1 parent ce9e61c commit 1db729d

File tree

3 files changed

+72
-1
lines changed

3 files changed

+72
-1
lines changed
Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
name: Run benchmarks on AMD GPU backend
2+
3+
on: pull_request
4+
5+
jobs:
6+
run-tests:
7+
strategy:
8+
matrix:
9+
runner:
10+
- cms-patatrack-mi300x
11+
- cms-patatrack-w7900
12+
runs-on: ${{ matrix.runner }}
13+
container: registry.cern.ch/ngt/lxplus-like:9
14+
15+
steps:
16+
- uses: actions/checkout@v3
17+
with:
18+
submodules: true
19+
20+
- name: Install dependencies
21+
run: |
22+
dnf install epel-release -y
23+
dnf makecache -y
24+
dnf install -y google-benchmark google-benchmark-devel
25+
26+
- name: Compile and run benchmark
27+
working-directory: ${{ github.workspace }}/benchmark/dataset_size
28+
run: |
29+
cmake -B build -DCMAKE_BUILD_TYPE=Release
30+
cmake --build build --parallel
31+
echo "Running serial backend"
32+
./build/serial.out
33+
echo "Running OpenMP backend"
34+
./build/openmp.out
35+
echo "Running CUDA backend"
36+
./build/hip.out
Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
name: Run benchmarks on NVIDIA GPU backend
2+
3+
on: pull_request
4+
5+
jobs:
6+
benchmark-nvidia-gpu:
7+
strategy:
8+
matrix:
9+
runner:
10+
- cms-patatrack-h100
11+
- cms-patatrack-l40s
12+
runs-on: ${{ matrix.runner }}
13+
container: registry.cern.ch/ngt/lxplus-like:9
14+
15+
steps:
16+
- uses: actions/checkout@v3
17+
with:
18+
submodules: true
19+
20+
- name: Install dependencies
21+
run: |
22+
dnf install epel-release -y
23+
dnf makecache -y
24+
dnf install -y google-benchmark google-benchmark-devel
25+
26+
- name: Compile and run benchmark
27+
working-directory: ${{ github.workspace }}/benchmark/dataset_size
28+
run: |
29+
cmake -B build -DCMAKE_BUILD_TYPE=Release
30+
cmake --build build --parallel
31+
echo "Running serial backend"
32+
./build/serial.out
33+
echo "Running OpenMP backend"
34+
./build/openmp.out
35+
echo "Running CUDA backend"
36+
./build/cuda.out

.github/workflows/benchmark-serial.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ jobs:
1515
fail-fast: false
1616

1717
steps:
18-
# checks out the code in the repository
1918
- uses: actions/checkout@v3
2019
with:
2120
submodules: true

0 commit comments

Comments
 (0)