Skip to content

Commit 2cd065b

Browse files
authored
fix: export MAX_JOBS for AOT build (#1626)
<!-- .github/pull_request_template.md --> ## 📌 Description Previously, MAX_JOBS was calculated but not exported, causing the build system to use all available CPU cores for parallel compilation. ## 🔍 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 - [ ] I have installed `pre-commit` by running `pip install pre-commit` (or used your preferred method). - [ ] I have installed the hooks with `pre-commit install`. - [ ] 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 463cd22 commit 2cd065b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

scripts/task_test_aot_build_import.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,9 @@ else
2828
MAX_JOBS=$BASE_MAX_JOBS
2929
fi
3030

31+
# Export MAX_JOBS for PyTorch's cpp_extension to use
32+
export MAX_JOBS
33+
3134
: ${CUDA_VISIBLE_DEVICES:=""}
3235
export TORCH_CUDA_ARCH_LIST=$(python3 -c '
3336
import torch

0 commit comments

Comments
 (0)