PR #3588 add device-agnostic GradScaler

add device-agnostic GradScaler #3588

SunMarc merged 6 commits into huggingface:main from yao-matrix:amp-xpu

add device-agnostic GradScaler

8ab6f232

fix bug

23d64ff0

IlyasMoutawwakil approved these changes on 2025-05-26

SunMarc commented on 2025-05-26

fix review comments

1d4bda36

fix

17b7a98e

format

de5de8a5

yao-matrix commented on 2025-05-26

SunMarc approved these changes on 2025-05-27

Apply style fixes

8eddaa6d

SunMarc merged 43526c5c into main 1 year ago

yao-matrix deleted the amp-xpu branch 1 year ago

Reviewers

SunMarc

IlyasMoutawwakil

Assignees

No one assigned

Labels

None yet

Milestone

No milestone