Searched refs:dd_output_device (Results 1 – 1 of 1) sorted by relevance
116 torch::DeviceType dd_output_device; in ModelInference_Torch() local124 PetscCall(PetscMemTypeToDeviceType(output_mem_type, &dd_output_device)); in ModelInference_Torch()128 if (dd_output_device == torch::kXPU) { // XPU requires device-to-host-to-device transfer in ModelInference_Torch()149 … PetscCallCXX(options = torch::TensorOptions().dtype(torch::kFloat64).device(dd_output_device)); in ModelInference_Torch()