verl/utils/megatron/pipeline_parallel.py

• compute_transformers_input_shapes · function · L22-L40 — def compute_transformers_input_shapes(batches, meta_info)
• make_batch_generator · function · L43-L51 — def make_batch_generator(batches, vpp_size)