Skip to content

Commit a69a8bf

Browse files
authored
Tiny allow compiling with line info and release moe (#1659)
<!-- .github/pull_request_template.md --> ## 📌 Description useful for ncu <!-- 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 ad53393 commit a69a8bf

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

flashinfer/jit/core.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,10 @@ def gen_jit_spec(
197197
# non debug mode
198198
cuda_cflags += ["-DNDEBUG"]
199199

200+
# useful for ncu
201+
if bool(os.environ.get("FLASHINFER_JIT_LINEINFO", "0")):
202+
cuda_cflags += ["-lineinfo"]
203+
200204
if extra_cflags is not None:
201205
cflags += extra_cflags
202206
if extra_cuda_cflags is not None:

0 commit comments

Comments
 (0)