ColossalAI
ColossalAI copied to clipboard
[engin/schedule] use p2p_v2 to recontruct pipeline_schedule
use p2p_v2 to recontruct pipeline_schedule and test it with tests/test_data_pipeline_tensor_parallel/test_cifar_with_data_pipeline_tensor_v2.py