xla
[benchmarks] Set autocast kwargs only if AMP.
#6612

Merged

[benchmarks] Set autocast kwargs only if AMP. #6612

ysiraichi merged 1 commit into master from ysiraichi/fix-benchmark-amp

[benchmarks] Set autocast kwargs only if AMP should be used.

3e5ca3c5

ysiraichi requested a review from

frgossen 2 years ago

ysiraichi added xla:gpu

ysiraichi requested a review from

vanbasten23 2 years ago

ysiraichi requested a review from

golechwierowicz 2 years ago

ysiraichi requested a review from

zpcore 2 years ago

ysiraichi requested a review from

cota 2 years ago

frgossen approved these changes on 2024-02-26

ysiraichi merged f65bae4d into master 2 years ago

Reviewers

frgossen

vanbasten23

golechwierowicz

zpcore

cota

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone