From 45b938852ff9a2aa07cc9255278630b050f8e922 Mon Sep 17 00:00:00 2001 From: Connor1996 Date: Tue, 3 Mar 2026 00:22:29 -0800 Subject: [PATCH] fix(ref): remove flash attention debug sentinel write Signed-off-by: Connor1996 --- src/extensions_ref/src/flash_attention.metal | 8 -------- 1 file changed, 8 deletions(-) diff --git a/src/extensions_ref/src/flash_attention.metal b/src/extensions_ref/src/flash_attention.metal index b78af0e..6bd5fb4 100644 --- a/src/extensions_ref/src/flash_attention.metal +++ b/src/extensions_ref/src/flash_attention.metal @@ -62,14 +62,6 @@ using namespace metal; } } - if (simd_lid == 0) { - for (int c = 0; c < E; c++) { - if (is_i_in_range && n < N) { - out[n * L * E + (i * Br + a) * E + c] = -233.0; - } - } - } - for (int j = 0; j < Tc; j++) { int row_max = min((i + 1) * Br - 1, L - 1); int col_min = j * Bc;