biondizzle 9fec7d609e Fix gsa_buffer shape mismatch for MoE (M>1 rows) ...

compute_amax_gsa returns a scalar, but quantize_from_buffer expects (M,).
Broadcast the scalar gsa to (M,) — all rows use the same gsa (global max).

2026-06-01 21:33:59 +00:00

..

__init__.py

Restructure: cutedsl/ -> dsv4/ with proper layering

2026-05-21 17:30:44 +00:00

custom_ops.py

Stage E: head-packed MQA/GQA, batch dim, custom_op, integration API

2026-05-27 15:15:03 +00:00

gemm_runner.py

fix: import SF_VEC_SIZE from quantize in gemm_runner (was NameError)

2026-06-01 00:04:48 +00:00

layouts.py

Restructure: cutedsl/ -> dsv4/ with proper layering

2026-05-21 17:30:44 +00:00

quantize.py

Fix gsa_buffer shape mismatch for MoE (M>1 rows)

2026-06-01 21:33:59 +00:00

rope.py

Fix production rope.py: FP32 arithmetic for forward_rope_partial + inverse_rope_bf16

2026-05-31 09:17:36 +00:00

router.py

router: catch CuTeDSL warmup failures fast, don't let MLIR errors slow down init

2026-06-01 00:00:07 +00:00

topk_select.py

Router: full kernel stack — hash, topk, activation+topk, dense decode/prefill

2026-05-21 21:54:05 +00:00

topk.py

Restructure: cutedsl/ -> dsv4/ with proper layering

2026-05-21 17:30:44 +00:00