xla
Make clip_grad_norm_ follow the input's dtype
#7205

Merged

Make clip_grad_norm_ follow the input's dtype #7205

alanwaketan merged 2 commits into master from alanwaketan/clip_grad_norm

fix clip_grad_norm_

74e04280

fix linters

b67a6ebe

alanwaketan requested a review from

JackCaoG 1 year ago

alanwaketan assigned

alanwaketan 1 year ago

JackCaoG commented on 2024-06-05

JackCaoG approved these changes on 2024-06-05

alanwaketan merged f5c00624 into master 1 year ago

Reviewers

JackCaoG

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone