pytorch / PiPPy

Pipeline Parallelism for PyTorch
BSD 3-Clause "New" or "Revised" License
715 stars 86 forks source link

Add nightly model tests against pytorch #1121

Closed kwen2501 closed 3 months ago

kwen2501 commented 3 months ago

Stack from ghstack (oldest at bottom):

and migrate gpt2_cpu_init example

kwen2501 commented 3 months ago

Merged #1120 wrongly. Hence a re-pull.