Repository navigation

a16w16 gemm tuned dsv4 pro shapes - #4222

Open

ahmed-bsod wants to merge 9 commits into

mainfrom

ahmed/dsv4-tune-bf16gemm

Commits on Jul 15, 2026
- Use triton for DSv4 Pro bf16 GEMM
  
  azaidy
  authored and
  ahmed-bsod
  committed
  ba74208
- Add bf16 GEMM tuning for DSv4 Pro
  
  azaidy
  authored and
  ahmed-bsod
  committed
  d451661
- Tune triton BF16 GEMMs more for DSv4
  
  azaidy
  authored and
  ahmed-bsod
  committed
  9093b7f
- tuned m=64,128
  ahmed-bsod
  committed
  deb8fd8
- tuned m=1,32
  ahmed-bsod
  committed
  5d1b604
- tuned m=256
  ahmed-bsod
  committed
  cdc472f
- tuned m=16k
  ahmed-bsod
  committed
  29a0a30
- black formatting fix
  ahmed-bsod
  committed
  11d7fa1
Commits on Jul 16, 2026
- fixes WAR hazard in kernel rn due to compiler issuing tdm load to same buffer as the ds_load without waiting for all the ds_loads to drain. we leave one buffer gap between tdm load and lds read now
  ahmed-bsod
  committed
  31c4023