xla
Fix preserve_rng_state for activation checkpointing
#4690
Merged
Overview
Commits
6
Changes
View On GitHub

Fix preserve_rng_state for activation checkpointing #4690

JackCaoG merged 6 commits into pytorch:master from YangFei1990:fix_activation_checkpoint
YangFei1990
YangFei1990 fix rng state for activation checkpoint
29becf3c
JackCaoG JackCaoG requested a review from JackCaoG JackCaoG 2 years ago
JackCaoG
JackCaoG JackCaoG added fsdp
JackCaoG JackCaoG requested a review from alanwaketan alanwaketan 1 year ago
jeffhataws jeffhataws added backport_2.3
YangFei1990 resolve conflicts
05ad6e46
YangFei1990 add test
392fe6c0
YangFei1990 format change
d3fb5a67
JackCaoG
JackCaoG commented on 2024-03-19
alanwaketan
alanwaketan approved these changes on 2024-03-19
YangFei1990 formatting
355efb29
YangFei1990 formatting
78c08ddf
JackCaoG
JackCaoG approved these changes on 2024-03-20
JackCaoG JackCaoG merged 66acfebf into master 1 year ago
JackCaoG

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
fsdp backport_2.3
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum