Skip to content

Commit 7bac369

Browse files
yongwwwyzh119
andauthored
ci: enable tests for sm75 (G4) (#1705)
<!-- .github/pull_request_template.md --> ## πŸ“Œ Description enable the test as mentioned in #1697 ## πŸ” 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 - [x] Tests have been added or updated as needed. - [ ] All tests are passing (`unittest`, etc.). ## Reviewer Notes cc: @yzh119 Co-authored-by: Zihao Ye <[email protected]>
1 parent a6e9c9f commit 7bac369

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

β€ŽJenkinsfileβ€Ž

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -303,5 +303,11 @@ stage('Unittest') {
303303
run_with_spot_retry('GPU-G5-SPOT', 'GPU-G5', 'JIT-Unittest-5-cu129',
304304
{ node_type -> shard_run_unittest_GPU(node_type, 5, 'cu129') })
305305
},
306+
// JIT unit test for CUDA 12.9 with SM75 (AWS G4)
307+
// For now, we only enable sampling test for SM75
308+
'JIT-Unittest-G4-cu129': {
309+
run_with_spot_retry('GPU-G4-SPOT', 'GPU-G4', 'JIT-Unittest-G4-cu129',
310+
{ node_type -> shard_run_unittest_GPU(node_type, 3, 'cu129') })
311+
},
306312
)
307313
}

0 commit comments

Comments
Β (0)