diff --git a/msccl/language/mscclpp/instruction_dag.py b/msccl/language/mscclpp/instruction_dag.py index 56cd166..703b0e1 100644 --- a/msccl/language/mscclpp/instruction_dag.py +++ b/msccl/language/mscclpp/instruction_dag.py @@ -302,7 +302,7 @@ def _fuse_same_instructions(self): if inst_type in instruction_handlers: for next_op in op.next: same_buf_func = instruction_handlers[inst_type] - if optimizer.try_merge_same_instruction(op, next_op, tb, queue, inst_type, same_buf_func): + if optimizer.try_merge_same_instructions(op, next_op, tb, queue, inst_type, same_buf_func): fused = True break if fused: diff --git a/msccl/language/mscclpp/instruction_optimizer.py b/msccl/language/mscclpp/instruction_optimizer.py index 9474afd..6640851 100644 --- a/msccl/language/mscclpp/instruction_optimizer.py +++ b/msccl/language/mscclpp/instruction_optimizer.py @@ -15,7 +15,7 @@ class InstructionOptimizer: - def try_merge_same_instruction( + def try_merge_same_instructions( self, op: Op, next_op: Op,