You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_pipeline/test_schedule
Wenhao Chen ef4f0ee854
[hotfix]: add pp sanity check and fix mbs arg (#5268)
11 months ago
..
test_interleaved.py [pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134) 11 months ago
test_oneF_oneB.py [hotfix]: add pp sanity check and fix mbs arg (#5268) 11 months ago
test_pipeline_schedule_utils.py