Skip to content

Commit 0c1fd7a

Browse files
authored
ci: add cuda-13 unittests to CI (#1603)
<!-- .github/pull_request_template.md --> ## 📌 Description <!-- What does this PR do? Briefly describe the changes and why they’re needed. --> ## 🔍 Related Issues <!-- Link any related issues here --> ## 🚀 Pull Request Checklist Thank you for contributing to FlashInfer! Before we review your pull request, please make sure the following items are complete. ### ✅ Pre-commit Checks - [x] I have installed `pre-commit` by running `pip install pre-commit` (or used your preferred method). - [x] I have installed the hooks with `pre-commit install`. - [x] I have run the hooks manually with `pre-commit run --all-files` and fixed any reported issues. > If you are unsure about how to set up `pre-commit`, see [the pre-commit documentation](https://pre-commit.com/). ## 🧪 Tests - [ ] Tests have been added or updated as needed. - [ ] All tests are passing (`unittest`, etc.). ## Reviewer Notes <!-- Optional: anything you'd like reviewers to focus on, concerns, etc. -->
1 parent 689b34f commit 0c1fd7a

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

Jenkinsfile

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ import org.jenkinsci.plugins.pipeline.modeldefinition.Utils
4343
docker_run_cu126 = "bash ci/bash.sh flashinfer/flashinfer-ci-cu126:latest"
4444
docker_run_cu128 = "bash ci/bash.sh flashinfer/flashinfer-ci-cu128:latest"
4545
docker_run_cu129 = "bash ci/bash.sh flashinfer/flashinfer-ci-cu129:latest"
46+
docker_run_cu130 = "bash ci/bash.sh flashinfer/flashinfer-ci-cu130:latest"
4647

4748
def per_exec_ws(folder) {
4849
return "workspace/exec_${env.EXECUTOR_NUMBER}/" + folder
@@ -137,6 +138,8 @@ def run_unittest_CPU_AOT_COMPILE(node_type, cuda_version) {
137138
docker_run = docker_run_cu128
138139
} else if (cuda_version == "cu129") {
139140
docker_run = docker_run_cu129
141+
} else if (cuda_version == "cu130") {
142+
docker_run = docker_run_cu130
140143
} else {
141144
error("Unknown CUDA version: ${cuda_version}")
142145
}
@@ -243,7 +246,7 @@ stage('Unittest') {
243246
cancel_previous_build()
244247
parallel(
245248
failFast: true,
246-
// CUDA 12.6 Tests
249+
// CUDA 12.6 AOT Tests
247250
'AOT-Build-Import-x86-64-cu126': {
248251
run_with_spot_retry('CPU-LARGE-SPOT', 'CPU-LARGE', 'AOT-Build-Import-x86-64-cu126',
249252
{ node_type -> run_unittest_CPU_AOT_COMPILE(node_type, 'cu126') })
@@ -252,7 +255,7 @@ stage('Unittest') {
252255
run_with_spot_retry('ARM-LARGE-SPOT', 'ARM-LARGE', 'AOT-Build-Import-aarch64-cu126',
253256
{ node_type -> run_unittest_CPU_AOT_COMPILE(node_type, 'cu126') })
254257
},
255-
// CUDA 12.8 Tests
258+
// CUDA 12.8 AOT Tests
256259
'AOT-Build-Import-x86-64-cu128': {
257260
run_with_spot_retry('CPU-LARGE-SPOT', 'CPU-LARGE', 'AOT-Build-Import-x86-64-cu128',
258261
{ node_type -> run_unittest_CPU_AOT_COMPILE(node_type, 'cu128') })
@@ -261,7 +264,7 @@ stage('Unittest') {
261264
run_with_spot_retry('ARM-LARGE-SPOT', 'ARM-LARGE', 'AOT-Build-Import-aarch64-cu128',
262265
{ node_type -> run_unittest_CPU_AOT_COMPILE(node_type, 'cu128') })
263266
},
264-
// CUDA 12.9 Tests
267+
// CUDA 12.9 AOT Tests
265268
'AOT-Build-Import-x86-64-cu129': {
266269
run_with_spot_retry('CPU-LARGE-SPOT', 'CPU-LARGE', 'AOT-Build-Import-x86-64-cu129',
267270
{ node_type -> run_unittest_CPU_AOT_COMPILE(node_type, 'cu129') })
@@ -270,6 +273,15 @@ stage('Unittest') {
270273
run_with_spot_retry('ARM-LARGE-SPOT', 'ARM-LARGE', 'AOT-Build-Import-aarch64-cu129',
271274
{ node_type -> run_unittest_CPU_AOT_COMPILE(node_type, 'cu129') })
272275
},
276+
// CUDA 13.0 AOT Tests
277+
'AOT-Build-Import-x86-64-cu130': {
278+
run_with_spot_retry('CPU-LARGE-SPOT', 'CPU-LARGE', 'AOT-Build-Import-x86-64-cu130',
279+
{ node_type -> run_unittest_CPU_AOT_COMPILE(node_type, 'cu130') })
280+
},
281+
'AOT-Build-Import-aarch64-cu130': {
282+
run_with_spot_retry('ARM-LARGE-SPOT', 'ARM-LARGE', 'AOT-Build-Import-aarch64-cu130',
283+
{ node_type -> run_unittest_CPU_AOT_COMPILE(node_type, 'cu130') })
284+
},
273285
// JIT unittest only for cu129
274286
'JIT-Unittest-1-cu129': {
275287
run_with_spot_retry('GPU-G5-SPOT', 'GPU-G5', 'JIT-Unittest-1-cu129',

0 commit comments

Comments
 (0)