Skip to content

Commit a6a1e49

Browse files
authored
Fix flag order (#1392)
On AArch64, the order matters for the linker <!-- .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 - [ ] 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 7c79b41 commit a6a1e49

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

flashinfer/jit/cpp_ext.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,12 +86,12 @@ def generate_ninja_build_for_op(
8686
ldflags = [
8787
"-shared",
8888
"-L$torch_home/lib",
89+
"-L$cuda_home/lib64",
8990
"-lc10",
9091
"-lc10_cuda",
9192
"-ltorch_cpu",
9293
"-ltorch_cuda",
9394
"-ltorch",
94-
"-L$cuda_home/lib64",
9595
"-lcudart",
9696
]
9797

0 commit comments

Comments
 (0)