Fix condition that checks if put operation can be fused (#32)

Azure · Jan 7, 2025 · ceaf52f · ceaf52f
1 parent 25744c3
commit ceaf52f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/msccl/language/mscclpp/instruction_optimizer.py b/msccl/language/mscclpp/instruction_optimizer.py
@@ -122,8 +122,8 @@ def try_fuse_with_put(self, op: Op, next_op: Op, tb: Threadblock, queue: list) -
         :return: True if operations are merged, False otherwise.
         """
         if (
-            next_op.inst == Instruction.put
-            or next_op.inst == Instruction.put_packet
+            (next_op.inst == Instruction.put
+            or next_op.inst == Instruction.put_packet)
             and same_tb(op, next_op)
             and same_count(op, next_op)
             and buf_dst_src_match(op, next_op)