xla

cb4983e9 - Automatically detect CUDA if it isn't set and default to upstream torch.cuda number of devices (#6605)