[CK] Mxfp4 moe blockscale buf2lds version support (#2455)

Browse Source

* change cshuffle size

* added mxfp4 moe async buffer loading without B preshuffle

* added mx moe B shuffling + scale shuffling (async loads)

* minor fix

---------

Co-authored-by: mtgu0705 <mtgu@amd.com>

...

This commit is contained in:

Mingtao Gu

2025-07-06 15:42:00 +08:00

committed by GitHub

parent 3d70c638d1

commit 7998ae8969

19 changed files with 10677 additions and 3473 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

1532

include/ck/tensor_operation/gpu/grid/gridwise_moe_mx_gemm.hpp

File diff suppressed because it is too large Load Diff

2761

include/ck/tensor_operation/gpu/grid/gridwise_moe_mx_gemm_bpreshuffle.hpp Normal file

File diff suppressed because it is too large Load Diff