xla
Reduce unnecessary tensor allocation in Adam and AdamW
#5700
Merged
Overview
Commits
2
Changes
View On GitHub

Reduce unnecessary tensor allocation in Adam and AdamW #5700

JackCaoG merged 2 commits into pytorch:master from baoleai:features/adamw
baoleai
baoleai Reduce unnecessary tensor allocation in adam and adamw
b6ce6eb7
baoleai fix lint
984950ca
baoleai baoleai changed the title Reduce unnecessary tensor allocation in adam and adamw Reduce unnecessary tensor allocation in Adam and AdamW 2 years ago
JackCaoG
JackCaoG JackCaoG added xla:gpu
ymwangg
ymwangg approved these changes on 2023-10-13
JackCaoG
JackCaoG
JackCaoG approved these changes on 2023-10-13
JackCaoG JackCaoG merged d9523716 into master 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
xla:gpu
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum