pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.44k stars 451 forks source link

OpenXLA CUDA fallback: Tracking Issue #7342

Open ysiraichi opened 2 months ago

ysiraichi commented 2 months ago

PR #7318 introduced OpenXLA CUDA fallback. However, that's an opt-in feature that relies on the environment variable: XLA_FALLBACK_CUDA. Even though that works fine, as this comment suggests, it doesn't provide a good UX. Therefore, we should improve it further by adding the following changes:

cc @miladm @JackCaoG @vanbasten23

ManfeiBai commented 2 months ago

Hi, @vanbasten23, is that ok to assign this ticket to you?

ysiraichi commented 2 months ago

I will take on this one. @vanbasten23 let me know if you are already working on any of them.