pytorch / PiPPy

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

Add tests for input check #1100

Closed H-Huang closed 6 months ago

H-Huang commented 6 months ago

Fixes https://github.com/pytorch/PiPPy/issues/1054. Already fixed, but just adding tests for validation

Stack from ghstack (oldest at bottom):