biondizzle e498a2c729 fix: single transpose back to MN-major, don't double-transpose ...

The .contiguous().transpose() dance was swapping dims back.
A single transpose from (g,k,mn) gives (g,mn,k) with stride(-2)=1,
which is exactly the MN-major layout TMA expects.

2026-05-12 14:23:02 +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: single transpose back to MN-major, don't double-transpose

2026-05-12 14:23:02 +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