Merge branch 'main' into mahdieh/broadcast

Azure · Jan 15, 2025 · 8f60592 · 8f60592
2 parents 80fb073 + ceaf52f
commit 8f60592
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)