We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3176277 commit a6e3160Copy full SHA for a6e3160
1 file changed
.github/workflows/tests.yml
@@ -141,6 +141,28 @@ jobs:
141
- name: Run tests
142
run: pytest --durations=100
143
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
166
cuda-tests:
167
if: github.repository == 'bitsandbytes-foundation/bitsandbytes'
168
needs: build-cuda
0 commit comments