wu-arch/kernels
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
986d5072239cd565123794976fb25b8c8aa6311b
kernels/tests/regression/flash_attention/kernel.cpp
Hansung Kim 986d507223 flash: Fix single-tile GEMM for warp-specialized 
With 4 warps, we can only do 32x64 GEMM; serialize 64x64 into 2 32x64
GEMM calls by split by the row.
2024-08-30 17:12:46 -07:00

27 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink