Support head dim 256#329

Closed

woct0rdho wants to merge 2 commits into

thu-ml:mainfrom

woct0rdho:head_dim_256

Commits on Jan 9, 2026

Support head dim 256
woct0rdho
committed
Disable head dim 256 on sm90

FurkanGozukara
authored and
woct0rdho
committed