biondizzle 0a5cfe0433 add kernel compile caching — compile once, invoke on subsequent calls ...

First call: cute.compile() with real tensors (warmup).
Subsequent calls: just invoke compiled() with new CuTe views.
No cute.compile() in the forward path = cudagraph-safe.

2026-05-16 20:45:46 +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

add kernel compile caching — compile once, invoke on subsequent calls

2026-05-16 20:45:46 +00:00

moe_pipeline.py

restore: new bridge/moe_pipeline/layertest

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