Skip to content

perf(moe): persist the b12x direct-micro CuTe-DSL kernel to the disk cache #4701

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Kushagra7777 wants to merge 22 commits into
base: main
Choose a base branch
from
+721 −74
Open

perf(moe): persist the b12x direct-micro CuTe-DSL kernel to the disk cache #4701

Show file tree
Hide file tree
Changes from 13 commits
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
4768c85
perf(moe): persist the b12x direct-micro CuTe-DSL kernel to the disk …
claude Aug 11, 2026
70a1d80
Merge branch 'flashinfer-ai:main' into main
Kushagra7777 Aug 11, 2026
8272511
test(moe): key max_active_ctas at m=1 where it reaches codegen
claude Aug 11, 2026
456bd33
fix(moe): reach the legacy probe chain when .library raises
claude Aug 12, 2026
7191098
fix(moe): load CUDA library for TVM-FFI builds
Kushagra7777 Aug 12, 2026
c61a0bb
Revert "fix(moe): load CUDA library for TVM-FFI builds"
Kushagra7777 Aug 12, 2026
b7ed007
fix(moe): probe the direct-micro block dim via _load_cuda_library
claude Aug 12, 2026
d77d2f9
fix(moe): probe the block dim via a non-TVM-FFI reference compile
claude Aug 12, 2026
c61816c
fix(moe): launch direct-micro barriers as length-1 views to match the…
gavinkvx Aug 21, 2026
a5f4161
Merge pull request #1 from gavinkvx/fix-direct-micro-barrier-abi
Kushagra7777 Aug 24, 2026
9e3a792
Merge branch 'flashinfer-ai:main' into main
Kushagra7777 Aug 24, 2026
ed9ac7e
docs: clarify sm121 validation and probe assumptions
Kushagra7777 Aug 24, 2026
33e51d8
Merge branch 'main' into main
Kushagra7777 Aug 31, 2026
081927d
Merge branch 'flashinfer-ai:main' into main
Kushagra7777 Sep 10, 2026
839c209
Merge branch 'flashinfer-ai:main' into main
Kushagra7777 Sep 14, 2026
2833b95
Merge branch 'flashinfer-ai:main' into fix/b12x-direct-micro-cache-re…
Kushagra7777 Sep 14, 2026
b29c199
fix(moe): correct direct-micro disk cache reuse and probe recovery
Kushagra7777 Sep 14, 2026
6252ba4
Merge branch 'fix/b12x-direct-micro-cache-review' of https://github.c…
Kushagra7777 Sep 14, 2026
59f94f5
test(moe): use equivalent shapes for direct-micro cache reuse
Kushagra7777 Sep 14, 2026
bbe56c1
Merge remote-tracking branch 'origin/main' into fix/b12x-direct-micro…
Kushagra7777 Sep 14, 2026
4a480e0
Merge pull request #2 from Kushagra7777/fix/b12x-direct-micro-cache-r…
Kushagra7777 Sep 14, 2026
1ad3de6
Merge branch 'main' into main
Kushagra7777 Sep 14, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading
Loading