Skip to content

Commit c4c6f09

Browse files
committed
start cuda circle config
1 parent 5866b38 commit c4c6f09

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

.circleci/config.yml

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,17 @@ jobs:
212212
METAL_DEBUG_ERROR_MODE=0 \
213213
python -m xmlrunner discover -v python/tests -o test-results/gpu_jit
214214
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+
215226
build_release:
216227
parameters:
217228
python_version:
@@ -348,6 +359,7 @@ workflows:
348359
parameters:
349360
macosx_deployment_target: ["13.5", "14.0"]
350361
- linux_build_and_test
362+
- cuda_build_and_test
351363
- build_documentation
352364

353365
build_pypi_release:
@@ -455,6 +467,8 @@ workflows:
455467
macosx_deployment_target: ["13.5", "14.0"]
456468
- linux_build_and_test:
457469
requires: [ hold ]
470+
- cuda_build_and_test:
471+
requires: [ hold ]
458472
nightly_build:
459473
when:
460474
and:

0 commit comments

Comments
 (0)