minor

Azure · Sep 12, 2024 · 7818988 · 7818988
1 parent 4049fd0
commit 7818988
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -15,7 +15,7 @@
 
 class InstructionOptimizer:
 
-    def try_merge_same_instruction(
+    def try_merge_same_instructions(
         self,
         op: Op,
         next_op: Op,