[feat] fix testcase;

2025-09-05 02:51:59 +00:00 · 2024-11-11 11:34:29 +00:00
parent 12919de424
commit 337debcf2a
3 changed files with 7 additions and 10 deletions
--- a/colossalai/pipeline/schedule/zero_bubble_pp.py
+++ b/colossalai/pipeline/schedule/zero_bubble_pp.py
@@ -101,9 +101,6 @@ class ZeroBubbleVPipeScheduler(PipelineSchedule):
        # init buffer
        self._free_buffers()

-    def _set_send_metadata_buffers(self, model_chunk_id):
-        pass
-
    def _free_buffers(self):
        # free local buffer
        # two dim array, first dim is the model chunk, second dim is the microbatch queue