chenchenghao/DeepGEMM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
062cb160cff9e9316a41ac9365650e0ed16b4b68
DeepGEMM/deep_gemm/include/deep_gemm/impls
History
Zhean Xu 891d57b4db Add various optimizations and Mega MoE benchmarks (#316) 
* Merge with private repo

* Add Mega MoE Benchmark

* Minor fix

* Update

---------

Co-authored-by: Chenggang Zhao <chenggangz@deepseek.com>
2026-04-24 18:41:37 +08:00
..
sm90_bf16_gemm.cuh
[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.cuh
[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.cuh
[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.cuh
[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304)
2026-04-17 09:45:14 +08:00
sm90_fp8_mqa_logits.cuh
[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304)
2026-04-17 09:45:14 +08:00
sm90_fp8_paged_mqa_logits.cuh
Add various optimizations and Mega MoE benchmarks (#316)
2026-04-24 18:41:37 +08:00
sm90_tf32_hc_prenorm_gemm.cuh
[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.cuh
[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.cuh
[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304)
2026-04-17 09:45:14 +08:00
sm100_fp4_mqa_logits.cuh
Add various optimizations and Mega MoE benchmarks (#316)
2026-04-24 18:41:37 +08:00
sm100_fp4_paged_mqa_logits.cuh
Add various optimizations and Mega MoE benchmarks (#316)
2026-04-24 18:41:37 +08:00
sm100_fp8_fp4_gemm_1d1d.cuh
[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.cuh
Add various optimizations and Mega MoE benchmarks (#316)
2026-04-24 18:41:37 +08:00
sm100_fp8_gemm_1d1d.cuh
Fix sync issue of TMEM alloc/dealloc (#292)
2026-03-22 16:41:28 +08:00
sm100_fp8_mqa_logits.cuh
[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304)
2026-04-17 09:45:14 +08:00
sm100_fp8_paged_mqa_logits.cuh
Add various optimizations and Mega MoE benchmarks (#316)
2026-04-24 18:41:37 +08:00
sm100_tf32_hc_prenorm_gemm.cuh
[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.cuh
[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304)
2026-04-17 09:45:14 +08:00
smxx_layout.cuh
[Public release 26/04] Introducing Mega MoE, FP4 Indexer and other features/fixes (#304)
2026-04-17 09:45:14 +08:00