ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

[engin/schedule] use p2p_v2 to recontruct pipeline_schedule

Open LSTM-Kirigaya opened this issue 2 years ago • 0 comments

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

LSTM-Kirigaya avatar Aug 05 '22 10:08 LSTM-Kirigaya