Skip to content

Commit b0633a4

Browse files
committed
removed else branch that causes failure inside VARLEN if branch
1 parent 0d07174 commit b0633a4

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

python/perf-kernels/flash-attention.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -482,10 +482,9 @@ def attn_fwd(Q, K, V, bias, SM_SCALE: tl.constexpr, L, Out, stride_qz, stride_qh
482482
if start_m * BLOCK_M > seqlen_q:
483483
continue_condition = False
484484
# return
485-
else:
486-
cu_seqlens_k_start = tl.load(cu_seqlens_k + off_z)
487-
cu_seqlens_k_end = tl.load(cu_seqlens_k + off_z + 1)
488-
seqlen_k = cu_seqlens_k_end - cu_seqlens_k_start
485+
cu_seqlens_k_start = tl.load(cu_seqlens_k + off_z)
486+
cu_seqlens_k_end = tl.load(cu_seqlens_k + off_z + 1)
487+
seqlen_k = cu_seqlens_k_end - cu_seqlens_k_start
489488
else:
490489
cu_seqlens_q_start = 0
491490
cu_seqlens_k_start = 0

0 commit comments

Comments
 (0)