xla
Enable default buffer donation for gradient accumulation
#8758

Merged

Enable default buffer donation for gradient accumulation #8758

bhavya01 merged 3 commits into pytorch:master from rpsilva-aws:rpsilva_grad_acc_aliasing

Separate the buffer donor context manager

9d914bf2

Introduce buffer donations to gradient accumulation API

71a1bd94

Minor gradient accumulation refactoring

4b74b45e

rpsilva-aws marked this pull request as ready for review 356 days ago

rpsilva-aws requested a review from

tengyifei 355 days ago

bhavya01 requested a review from

bhavya01 355 days ago

bhavya01 commented on 2025-02-27

rpsilva-aws requested a review from

bhavya01 355 days ago

bhavya01 approved these changes on 2025-02-27

bhavya01 merged 6712eb91 into master 355 days ago

rpsilva-aws deleted the rpsilva_grad_acc_aliasing branch 355 days ago

Reviewers

bhavya01

tengyifei

Assignees

No one assigned

Labels

None yet

Milestone

No milestone