xla
[benchmarks] Fix AMP setup for torchbench models.
#7067

Merged

[benchmarks] Fix AMP setup for torchbench models. #7067

JackCaoG merged 4 commits into master from ysiraichi/fix-benchmark-amp-setup

Fix AMP setup.

d7d228b2

Better comments.

9f2ce8ac

ysiraichi added xla:gpu

Fix typos.

bf0b41ac

Fix lint issues.

c07b2a1a

ysiraichi requested a review from

vanbasten23 1 year ago

JackCaoG approved these changes on 2024-05-16

JackCaoG merged aeed89eb into master 1 year ago

vanbasten23 commented on 2024-05-16

Reviewers

JackCaoG

vanbasten23

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone