Searched refs:TORCH_DEVICE_CUDA (Results 1 – 3 of 3) sorted by relevance
12 TORCH_DEVICE_CUDA, enumerator
22 case TORCH_DEVICE_CUDA: in EnumToDeviceType()
260 if (strstr(ceed_resource, "/gpu/cuda")) model_device_type = TORCH_DEVICE_CUDA; in SgsDDSetupNodalEvaluation_Sequential_Torch()