biondizzle 44b40d41fe fix: compile CuTeDSL kernel with real tensors, not dummy shapes ...

The kernel's TMA descriptors are sized from compilation-time shapes.
Dummy 256x256 caused wrong memory access for real 3584x6144 data.
Now compiles with actual runtime tensors on first use, cached by
(num_experts, K, N). Compilation happens once during warmup.
Forward call remains cudagraph-safe.

2026-05-16 20:05:59 +00:00

..

kernel

refactor: copy CuTeDSL kernel into repo with local imports

2026-05-16 02:57:54 +00:00

__init__.py

refactor: copy CuTeDSL kernel into repo with local imports

2026-05-16 02:57:54 +00:00

bridge.py

fix: compile CuTeDSL kernel with real tensors, not dummy shapes

2026-05-16 20:05:59 +00:00

moe_pipeline.py

restore: new bridge/moe_pipeline/layertest

2026-05-16 19:55:19 +00:00