DeepSeek/FlashMLA
1
0
Fork 0
mirror of https://github.com/deepseek-ai/FlashMLA synced 2025-06-26 18:15:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update desc

Browse Source
This commit is contained in:
chenhongmin.will 2025-03-01 07:53:04 +08:00
parent 90289837fc
commit 6199b0b4b5
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
csrc/fp8_transpose_v.h
View File

@ -1,3 +1,8 @@
/**
* ref to Fa3's SmemTranspose64x64:
* https://github.com/Dao-AILab/flash-attention/blob/0823cf7b5d96499c1c79a4f64b1e256a035ba4b4/hopper/mainloop_fwd_sm90_tma_gmma_ws.hpp#L26
*/
#pragma once #pragma once
template <int kBlockN, int kHeadDim, typename SmemLayoutK> template <int kBlockN, int kHeadDim, typename SmemLayoutK>