fix

Azure · Sep 12, 2024 · 8006716 · 8006716
1 parent a06dbb4
commit 8006716
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
@@ -232,7 +232,7 @@ def _remove_redundant_signal_wait(self):
                                 break
                     elif op.inst == Instruction.reduce_packet or op.inst == Instruction.copy_packet:
                         for prev_op in op.prev:
-                            fused = optimizer.try_remove_op(prev_op, next_op.inst == Instruction.wait)
+                            fused = optimizer.try_remove_op(prev_op, prev_op.inst == Instruction.wait)
                             if fused:
                                 break
                     if fused: