Skip to content
This repository was archived by the owner on Jan 28, 2025. It is now read-only.

Commit ceaf52f

Browse files
authored
Fix condition that checks if put operation can be fused (#32)
1 parent 25744c3 commit ceaf52f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

msccl/language/mscclpp/instruction_optimizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,8 +122,8 @@ def try_fuse_with_put(self, op: Op, next_op: Op, tb: Threadblock, queue: list) -
122122
:return: True if operations are merged, False otherwise.
123123
"""
124124
if (
125-
next_op.inst == Instruction.put
126-
or next_op.inst == Instruction.put_packet
125+
(next_op.inst == Instruction.put
126+
or next_op.inst == Instruction.put_packet)
127127
and same_tb(op, next_op)
128128
and same_count(op, next_op)
129129
and buf_dst_src_match(op, next_op)

0 commit comments

Comments
 (0)