biondizzle 26a8ab75a1 NVFP4: fix SF pipeline — 2 K-cols per BLOCK_K for group=16 ...

- TMA: issue two tma::copy calls per K-block (K_box=1, 2 SF K-columns)
- UTCCP: double loop for 2 K-columns, correct SMEM offsets
- TMEM: double SFA/SFB column counts (SF_BLOCK_M/32 * 2)
- Heuristic: fix smem_size (2× SF, packed FP4 A/B, packed send buffers, no amax)
- Staging kernel: fix double-count bug in packed_k_mask

2026-05-12 08:08:17 +00:00

..

include/deep_gemm

NVFP4: fix SF pipeline — 2 K-cols per BLOCK_K for group=16

2026-05-12 08:08:17 +00:00

legacy

[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304)

2026-04-17 09:45:14 +08:00

mega

Fix fold_global_scale: UE4M3 scales use .to(float32), not shift-by-23

2026-05-12 05:52:33 +00:00

testing

[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304)

2026-04-17 09:45:14 +08:00

utils

[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304)

2026-04-17 09:45:14 +08:00

__init__.py

Add various optimizations and Mega MoE benchmarks (#316)

2026-04-24 18:41:37 +08:00