Home
last modified time | relevance | path

Searched refs:model_device_type (Results 1 – 1 of 1) sorted by relevance

/honee/problems/
H A Dsgs_dd_model.c256 TorchDeviceType model_device_type; in SgsDDSetupNodalEvaluation_Sequential_Torch() local
260 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()