issues
search
pytorch
/
torchtitan
A native PyTorch Library for large model training
BSD 3-Clause "New" or "Revised" License
1.28k
stars
115
forks
source link
Add test for PP tracer frontend
#357
Closed
wconstab
closed
1 month ago
wconstab
commented
1 month ago
Stack from
ghstack
(oldest at bottom):
->
#357
switch to using public PipelineStage API
clean up some asserts in tracer codepath
Stack from ghstack (oldest at bottom):