lalala-sh bcf5bb41be enable do top k weights in moe stage1 gemm (#2094) ...

* add switch for mul topk weights

* fix bf16/f16 bugs

* complete

2025-04-18 10:45:49 +08:00

..

block

MX GEMM - New GEMM pipeline for MX data types (#2059)

2025-04-15 17:17:07 -06:00

device

enable do top k weights in moe stage1 gemm (#2094)

2025-04-18 10:45:49 +08:00

element

add passthrough for int32->float32 (#2062)

2025-04-08 15:16:30 -07:00

grid

enable do top k weights in moe stage1 gemm (#2094)

2025-04-18 10:45:49 +08:00

thread

MX GEMM - New GEMM pipeline for MX data types (#2059)

2025-04-15 17:17:07 -06:00

warp

MX GEMM - Add MX BF8 example (#2071)

2025-04-16 15:25:02 -06:00