From 7818988ea2733754410237f3dfd42f81da93c21b Mon Sep 17 00:00:00 2001 From: Binyang Li Date: Thu, 12 Sep 2024 09:08:51 +0000 Subject: [PATCH] minor --- msccl/language/mscclpp/instruction_dag.py | 2 +- msccl/language/mscclpp/instruction_optimizer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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,