Skip to content

Commit a6e3160

Browse files
Test g5g runner
1 parent 3176277 commit a6e3160

1 file changed

Lines changed: 22 additions & 0 deletions

File tree

.github/workflows/tests.yml

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,28 @@ jobs:
141141
- name: Run tests
142142
run: pytest --durations=100
143143

144+
cuda-aarch64-tests:
145+
if: github.repository == 'bitsandbytes-foundation/bitsandbytes'
146+
needs: build-cuda
147+
strategy:
148+
fail-fast: false
149+
matrix:
150+
os: [ubuntu-22.04-arm]
151+
arch: [aarch64]
152+
torch_version: ["2.7.0"]
153+
cuda_version: ["11.8.0", "12.8.1"]
154+
155+
runs-on: bandb-aws-g5g-4xlarge-plus-use1-public-80
156+
env:
157+
BNB_TEST_DEVICE: cuda
158+
steps:
159+
- name: Show GPU Information
160+
run: nvidia-smi
161+
162+
- name: Show pip packages
163+
run: pip list
164+
165+
144166
cuda-tests:
145167
if: github.repository == 'bitsandbytes-foundation/bitsandbytes'
146168
needs: build-cuda

0 commit comments

Comments
 (0)