xla
Force to fp16 models to fp32 if `XLA_USE_FP16` is already set.
#6389

Merged

Force to fp16 models to fp32 if `XLA_USE_FP16` is already set. #6389

ysiraichi merged 4 commits into master from ysiraichi/fix-precision-fp16

Refactor `default_precision_flat` function.

f342fa27

Force model precision to be FP32.

964e91db

Force precision only for XLA:CUDA.

87924c85

ysiraichi added xla:gpu

ysiraichi requested a review from

golechwierowicz 1 year ago

ysiraichi requested a review from

frgossen 1 year ago

ysiraichi requested a review from

zpcore 1 year ago

ysiraichi requested a review from

vanbasten23 1 year ago

ysiraichi changed the title ~~Fallback to fp32 if `XLA_USE_FP16` is already set.~~ Force to fp16 models to fp32 if `XLA_USE_FP16` is already set. 1 year ago

Add comment.

33ac439b

golechwierowicz approved these changes on 2024-01-29

ysiraichi merged 2a17ac20 into master 1 year ago

Reviewers

golechwierowicz

frgossen

zpcore

vanbasten23

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone

xla Force to fp16 models to fp32 if `XLA_USE_FP16` is already set. #6389 Merged

Force to fp16 models to fp32 if `XLA_USE_FP16` is already set. #6389

xla
Force to fp16 models to fp32 if `XLA_USE_FP16` is already set.
#6389

Merged