Skip to content

Commit 28faa92

Browse files
fix tile scheduler dispatch logic
1 parent 9d482ad commit 28faa92

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

flash_attn/cute/flash_fwd_sm100.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,7 @@ def __init__(
240240
if self.dynamic_persistent and not self.use_clc_scheduler:
241241
self.use_varlen_scheduler = True
242242
self.TileScheduler = DynamicPersistentVarlenScheduler
243-
elif self.is_static_persistent:
243+
elif self.is_static_persistent and not self.use_clc_scheduler:
244244
self.TileScheduler = StaticPersistentTileScheduler
245245
else:
246246
self.use_varlen_scheduler = True

0 commit comments

Comments
 (0)