Skip to content

hgemm_splitk: add mixed sllitk/slicek policies#530

Draft
xytpai wants to merge 4 commits into
mainfrom
xyt/hgemm_spk_slicek
Draft

hgemm_splitk: add mixed sllitk/slicek policies#530
xytpai wants to merge 4 commits into
mainfrom
xyt/hgemm_spk_slicek

Conversation

@xytpai
Copy link
Copy Markdown
Contributor

@xytpai xytpai commented May 15, 2026

Motivation,

Add mixed sllitk/slicek implementations for hgemm.

Results:

rm -rf ~/.flydsl/ ; python3 tests/kernels/test_hgemm_splitk.py -m=32 -n=384 -k=7168 --TILE_M=32 --TILE_N=64 --TILE_N=256 --SPLIT_K=14 --BLOCK_M_WARPS=1 --BLOCK_N_WARPS=2 --BLOCK_K_WARPS=2

rm -rf ~/.flydsl/ ; python3 tests/kernels/test_hgemm_splitk.py -m=32 -n=7168 -k=2048 --TILE_M=16 --TILE_N=64 --TILE_N=256 --SPLIT_K=2 --BLOCK_M_WARPS=1 --BLOCK_N_WARPS=2 --BLOCK_K_WARPS=1

rm -rf ~/.flydsl/ ; python3 tests/kernels/test_hgemm_splitk.py -m=32 -n=384 -k=16384 --TILE_M=32 --TILE_N=64 --TILE_N=256 --SPLIT_K=16 --BLOCK_M_WARPS=1 --BLOCK_N_WARPS=2 --BLOCK_K_WARPS=2

rm -rf ~/.flydsl/ ; python3 tests/kernels/test_hgemm_splitk.py -m=8 -n=5120 -k=2880 --TILE_M=32 --TILE_N=128 --TILE_N=64 --SPLIT_K=9 --BLOCK_M_WARPS=1 --BLOCK_N_WARPS=4 --BLOCK_K_WARPS=1

rm -rf ~/.flydsl/ ; python3 tests/kernels/test_hgemm_splitk.py -m=32 -n=2880 -k=2048 --TILE_M=32 --TILE_N=64 --TILE_N=256 --SPLIT_K=4 --BLOCK_M_WARPS=1 --BLOCK_N_WARPS=2 --BLOCK_K_WARPS=2

@xytpai xytpai changed the title hgemm_splitk: Add a mixed sllitk/slicek policy hgemm_splitk: add mixed sllitk/slicek policies May 15, 2026
@xytpai xytpai marked this pull request as draft May 15, 2026 08:06
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant