PR #7630 Make CUDA OpenXLA fallback the default.

Make CUDA OpenXLA fallback the default. #7630

ysiraichi merged 7 commits into master from ysiraichi/make-cuda-fallback-default

ysiraichi added xla:gpu

ysiraichi requested a review from

miladm 1 year ago

ysiraichi requested a review from

vanbasten23 1 year ago

ysiraichi requested a review from

JackCaoG 1 year ago

miladm approved these changes on 2024-07-08

JackCaoG commented on 2024-07-08

JackCaoG approved these changes on 2024-07-08

vanbasten23 commented on 2024-07-10

vanbasten23 approved these changes on 2024-07-10

Add getter of `XlaDeviceType` enum, given `DeviceType`.

41c957b4

Introduce `device_count` CUDA function.

9740535c

Remove `XLA_FALLBACK_CUDA` environment variable.

4fcfed8a

Make CUDA OpenXLA fallback the default.

34a3ac1d

ysiraichi force pushed from 462a877a to 34a3ac1d 1 year ago

Check for CUDA kernel.

bc6e77af

Add comments.

0510afc8

Introduce a set of CPU fallback operations.

fa695086

ysiraichi commented on 2024-07-23

ysiraichi merged 806de832 into master 1 year ago

Reviewers

vanbasten23

JackCaoG

miladm

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone