xla
Fix DDP with gradient_as_bucket_view enabled
#8521
Merged

Fix DDP with gradient_as_bucket_view enabled #8521

yaochengji
Fix DDP with gradient_as_bucket_view enabled
3c2bc091
yaochengji yaochengji requested a review from mikegre-google mikegre-google 1 year ago
yaochengji yaochengji requested a review from tengyifei tengyifei 1 year ago
tengyifei tengyifei added tpuci
tengyifei
tengyifei approved these changes on 2025-01-03
tengyifei tengyifei merged bc3bf1f7 into master 1 year ago
tengyifei

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone