Skip to content

Commit d185abe

Browse files
ci: update runson repository configuration (#2036)
Add custom gpu runner
1 parent 38f1057 commit d185abe

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

.github/runs-on.yml

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,3 +5,16 @@ images:
55
owner: "405894839083"
66
name: "arm64-rust-riscv-dev-32gb"
77
ami: "ami-040a96edd406c9478"
8+
9+
runners:
10+
# runners for running tests
11+
test-gpu-nvidia:
12+
family: ["g6", "g6e", "g5", "g4dn"]
13+
image: ubuntu24-gpu-x64
14+
cpu: [4, 8, 16]
15+
disk: default
16+
spot: capacity-optimized
17+
preinstall: |
18+
nvidia-smi
19+
nvidia-smi -L
20+
nvcc --version

0 commit comments

Comments
 (0)