Searched defs:HAS_CUDA (Results 1 – 8 of 8) sorted by relevance
42 HAS_CUDA = torch.cuda.is_available() and has_triton() variable
28 HAS_CUDA = torch.cuda.is_available() variable
18 HAS_CUDA = torch.cuda.is_available() variable
29 HAS_CUDA = torch.cuda.is_available() variable
57 HAS_CUDA = torch.cuda.is_available() variable
37 HAS_CUDA = torch.cuda.is_available() variable
35 HAS_CUDA = HAS_CUDA and not torch.version.hip variable
54 from torch._C import _has_cuda as HAS_CUDA, _has_xpu as HAS_XPU unknown