Searched refs:TorchDeviceTypes (Results 1 – 2 of 2) sorted by relevance
16 static const char *const TorchDeviceTypes[] = {"CPU", "CUDA", "HIP", "XPU", "TorchDeviceType", "TOR… variable
265 …PetscCall(PetscOptionsGetEnum(NULL, NULL, "-sgs_model_dd_torch_model_device", TorchDeviceTypes, (P… in SgsDDSetupNodalEvaluation_Sequential_Torch()