Skip to content

Commit 6e0accc

Browse files
committed
Uncomment code
Signed-off-by: nithinsubbiah <nithinsubbiah@gmail.com>
1 parent 2160a75 commit 6e0accc

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

wave_lang/kernel/wave/schedule_reordering.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -548,10 +548,10 @@ def select_reorder_strategy(
548548
hardware_constraint,
549549
):
550550
flat_wave_count = math.prod(hardware_constraint.waves_per_block)
551-
# if flat_wave_count == 4 and is_compatible_strategy(
552-
# mTile, nTile, kTile, mma_bitwidth, use_global_to_shared, mma_type, fourWaveConfig
553-
# ):
554-
# return SchedReorderStrategy.FOUR_WAVE_INTERWEAVE
551+
if flat_wave_count == 4 and is_compatible_strategy(
552+
mTile, nTile, kTile, mma_bitwidth, use_global_to_shared, mma_type, fourWaveConfig
553+
):
554+
return SchedReorderStrategy.FOUR_WAVE_INTERWEAVE
555555
if flat_wave_count != 8:
556556
return SchedReorderStrategy.NONE
557557
if is_compatible_strategy(

0 commit comments

Comments
 (0)