xla
[benchmarks] Add support for AMP execution.
#6447

Merged

[benchmarks] Add support for AMP execution. #6447

ysiraichi merged 3 commits into master from ysiraichi/add-support-amp

Maybe add `amp` context.

268ef848

Import `torch_xla.amp`.

7aa05f30

ysiraichi added xla:gpu

ysiraichi requested a review from

vanbasten23 1 year ago

ysiraichi requested a review from

frgossen 1 year ago

ysiraichi requested a review from

golechwierowicz 1 year ago

ysiraichi requested a review from

zpcore 1 year ago

golechwierowicz approved these changes on 2024-02-01

Fix lint issues.

d7bcadc3

ysiraichi merged ad822bf0 into master 1 year ago

Reviewers

golechwierowicz

vanbasten23

frgossen

zpcore

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone