xla
Do not set `PJRT_DEVICE=CUDA` automatically on import.
#9540
Merged

Do not set `PJRT_DEVICE=CUDA` automatically on import. #9540

ysiraichi
ysiraichi Remove CUDA detection.
a96fc314
ysiraichi ysiraichi requested a review from zhanyong-wan zhanyong-wan 136 days ago
ysiraichi ysiraichi requested a review from tengyifei tengyifei 136 days ago
ysiraichi ysiraichi requested a review from bhavya01 bhavya01 136 days ago
ysiraichi ysiraichi changed the title Remove CUDA detection from runtime initialization. Do not set `PJRT_DEVICE=CUDA` automatically on import. 136 days ago
ghpvnist
ghpvnist approved these changes on 2025-08-05
ysiraichi ysiraichi merged 15496cd2 into master 136 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone