update desc

deepseek-ai · Feb 28, 2025 · 6199b0b · 6199b0b
1 parent 9028983
commit 6199b0b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/csrc/fp8_transpose_v.h b/csrc/fp8_transpose_v.h
@@ -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
 
 template <int kBlockN, int kHeadDim, typename SmemLayoutK>