Skip to content

Pull requests: pytorch/PiPPy

Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Assignee
Filter by who’s assigned
Sort

Pull requests list

add loss to looped_bfs cla signed
#1059 by H-Huang was merged Apr 16, 2024 Loading…
Pipeline stages share fwd/bwd send/recv ops cla signed
#1058 by H-Huang was merged Apr 16, 2024 Loading…
Fix #1053: wait for comm ops in Interleaved 1F1B cla signed
#1055 by kwen2501 was merged Apr 13, 2024 Loading…
Fix backward microbatch index in 1F1B cla signed
#1052 by kwen2501 was merged Apr 12, 2024 Loading…
Support loss computation in Interleaved 1F1B cla signed
#1051 by kwen2501 was merged Apr 12, 2024 Loading…
Add base class for multi-stage schedules cla signed
#1048 by kwen2501 was merged Apr 12, 2024 Loading…
Consolidate step() function cla signed
#1047 by kwen2501 was merged Apr 12, 2024 Loading…
Move GPipe and 1F1B to PipelineScheduleSingle cla signed
#1046 by kwen2501 was merged Apr 12, 2024 Loading…
Add FSDP support for GPipe schedule cla signed
#1045 by wconstab was merged Apr 17, 2024 Loading…
Fix loss calculation in bwd test cla signed
#1044 by kwen2501 was merged Apr 10, 2024 Loading…
Rename Schedules cla signed
#1041 by H-Huang was merged Apr 10, 2024 Loading…
fix some small TODOs in pipeline stage cla signed
#1040 by H-Huang was merged Apr 9, 2024 Loading…
fix some small TODOs in pipeline stage cla signed
#1039 by H-Huang was merged Apr 9, 2024 Loading…
Use PiPPy to distribute Mixture of Experts cla signed
#1038 by kwen2501 was merged Apr 9, 2024 Loading…
Add a print_readable method cla signed
#1037 by kwen2501 was merged Apr 9, 2024 Loading…
Remove no-backward warning cla signed
#1036 by kwen2501 was merged Apr 5, 2024 Loading…
Bring back stage_backward helper cla signed
#1032 by kwen2501 was merged Apr 4, 2024 Loading…
Add stage constructor for stage module cla signed
#1030 by kwen2501 was merged Apr 4, 2024 Loading…
Fix sending values that don't require_grad cla signed
#1029 by wconstab was merged Apr 3, 2024 Loading…
ProTip! Filter pull requests by the default branch with base:main.