vadim0x60

def torch_device(priority=['xla', 'cuda', 'mps', 'xpu', 'cpu']):
    import torch
    t = torch.Tensor([0])
    for device in map(torch.device, priority):
        try:
            t.to(device)
            return device
        except (RuntimeError, AssertionError):
            pass