[feat] fix testcase;

colossalai/pipeline/p2p.py

@@ -432,8 +432,6 @@ def _communicate(
         overlap_p2p=overlap_p2p,
         send_first=send_first if send_first != None else True,
     )
-    # print(f"rank {dist.get_rank()}; recv_src {recv_src}; send_dst {send_dst}; metadata_send {metadata_send}; metadata_recv {metadata_recv};")
-
     if metadata_recv is not None:
         assert isinstance(metadata_recv, P2PMetadata)
         tree_spec = metadata_recv.tree_spec

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