fix

Azure · Sep 11, 2024 · 6a8e3ce · 6a8e3ce
1 parent 5b202d6
commit 6a8e3ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/msccl/language/mscclpp/instruction_dag.py b/msccl/language/mscclpp/instruction_dag.py
@@ -294,7 +294,7 @@ def _optimize_rrcs_rs(self):
                     fused = False
                     if op.inst == Instruction.read_reduce_copy or op.inst == Instruction.read_reduce_copy_send:
                         for next_op in op.next:
-                            fused = optimizer.try_merge_with_put(op, next_op, tb, queue, Instruction.read_reduce_copy_send)
+                            fused = optimizer.try_merge_with_put(op, next_op, tb, queue, op.inst)
                             if fused:
                                 break
                         if fused: