Searched refs:model_device_type (Results 1 – 1 of 1) sorted by relevance
256 TorchDeviceType model_device_type; in SgsDDSetupNodalEvaluation_Sequential_Torch() local260 if (strstr(ceed_resource, "/gpu/cuda")) model_device_type = TORCH_DEVICE_CUDA; in SgsDDSetupNodalEvaluation_Sequential_Torch()261 else if (strstr(ceed_resource, "/gpu/hip")) model_device_type = TORCH_DEVICE_HIP; in SgsDDSetupNodalEvaluation_Sequential_Torch()264 else model_device_type = TORCH_DEVICE_CPU; in SgsDDSetupNodalEvaluation_Sequential_Torch()265 …ULL, "-sgs_model_dd_torch_model_device", TorchDeviceTypes, (PetscEnum *)&model_device_type, NULL)); in SgsDDSetupNodalEvaluation_Sequential_Torch()268 PetscCall(LoadModel_Torch(model_path, model_device_type)); in SgsDDSetupNodalEvaluation_Sequential_Torch()