Skip to content

Commit d2bc644

Browse files
author
Viviane Potocnik
committed
lint: FA-2 FP16 and FP8
1 parent 8b52aa5 commit d2bc644

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

sw/dnn/flashattention_2/src/flashattention_2_fp16.h

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,6 @@ static inline void flashattention_2_fp16(flashattention_2_layer_t layer) {
119119

120120
// Iterate column blocks of K (corresponding to row blocks of V)
121121
for (int t_c = 0; t_c < T_c; t_c++) {
122-
123122
// DMA copy K column block (B_c, d) and V row block (B_c, d) to
124123
// TCDM. Both K and V are stored in (S, d) form in memory
125124
if (!snrt_is_compute_core()) {
@@ -246,7 +245,6 @@ static inline void flashattention_2_fp16(flashattention_2_layer_t layer) {
246245
snrt_mcycle();
247246
} // end of T_c loop
248247

249-
250248
// Rescaling for last t_c iteration
251249
// O_i = diag(l_i_Tc)^-1 * O_i
252250
if (snrt_is_compute_core()) {

sw/dnn/flashattention_2/src/flashattention_2_fp8.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ static inline void flashattention_2_fp8(flashattention_2_layer_t layer) {
8686

8787
// Allocate space for V^t
8888
char *V_t;
89-
if(!baseline) {
89+
if (!baseline) {
9090
V_t = tcdm_ptr;
9191
tcdm_ptr += B_c * d * sizeof(char);
9292
}

0 commit comments

Comments
 (0)