File tree Expand file tree Collapse file tree 1 file changed +14
-0
lines changed Expand file tree Collapse file tree 1 file changed +14
-0
lines changed Original file line number Diff line number Diff line change @@ -212,6 +212,17 @@ jobs:
212
212
METAL_DEBUG_ERROR_MODE=0 \
213
213
python -m xmlrunner discover -v python/tests -o test-results/gpu_jit
214
214
215
+ cuda_build_and_test :
216
+ machine :
217
+ image : linux-cuda-12:default
218
+ resource_class : gpu.nvidia.small.gen2
219
+ steps :
220
+ - checkout
221
+ - run :
222
+ name : Install Python package
223
+ command : |
224
+ sudo apt-get update
225
+
215
226
build_release :
216
227
parameters :
217
228
python_version :
@@ -348,6 +359,7 @@ workflows:
348
359
parameters :
349
360
macosx_deployment_target : ["13.5", "14.0"]
350
361
- linux_build_and_test
362
+ - cuda_build_and_test
351
363
- build_documentation
352
364
353
365
build_pypi_release :
@@ -455,6 +467,8 @@ workflows:
455
467
macosx_deployment_target : ["13.5", "14.0"]
456
468
- linux_build_and_test :
457
469
requires : [ hold ]
470
+ - cuda_build_and_test :
471
+ requires : [ hold ]
458
472
nightly_build :
459
473
when :
460
474
and :
You can’t perform that action at this time.
0 commit comments