biondizzle c0850a6859 Fix weight TMA descriptors: packed E2M1 needs K/2, block_k/2, swizzle/2 ...

Weights are packed E2M1 (2 per byte) but TMA descriptors were using
unpacked dimensions — K-dim in elements instead of bytes, 128B swizzle
instead of 64B, full block_k instead of block_k/2. This caused OOB
reads and swizzle mismatch with the UMMA descriptor, producing
illegal instruction traps.

2026-05-12 06:51:39 +00:00

..

epilogue.hpp

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

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

runtime_utils.hpp

fix: remove duplicate kInt8 case — kPackedFP4 is already kInt8

2026-05-11 22:55:28 +00:00

sm90_bf16_gemm.hpp

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

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

sm90_bmk_bnk_mn.hpp

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

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

sm90_fp8_gemm_1d1d.hpp

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

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

sm90_fp8_gemm_1d2d.hpp

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

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

sm90_tf32_hc_prenorm_gemm.hpp

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

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

sm100_bf16_gemm.hpp

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

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

sm100_bmk_bnk_mn.hpp

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

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

sm100_fp8_fp4_gemm_1d1d.hpp

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

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

sm100_fp8_fp4_mega_moe.hpp

Add various optimizations and Mega MoE benchmarks (#316)

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

sm100_fp8_gemm_1d1d.hpp

Multiple updates and refactorings (#280)

2026-01-16 17:06:52 +08:00

sm100_fp8_nvfp4_mega_moe.hpp

Fix weight TMA descriptors: packed E2M1 needs K/2, block_k/2, swizzle/2

2026-05-12 06:51:39 +00:00

sm100_tf32_hc_prenorm_gemm.hpp

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

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

smxx_clean_logits.hpp

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

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

smxx_cublaslt.hpp

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

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

smxx_fp8_fp4_mqa_logits.hpp

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

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

smxx_fp8_fp4_paged_mqa_logits.hpp

Add various optimizations and Mega MoE benchmarks (#316)

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

smxx_fp8_mqa_logits.hpp

Multiple updates and refactorings (#231)

2025-11-21 17:49:47 +08:00

smxx_fp8_paged_mqa_logits.hpp

Multiple updates and refactorings (#280)

2026-01-16 17:06:52 +08:00

smxx_layout.hpp

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

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