Searched refs:dd_input_device (Results 1 – 1 of 1) sorted by relevance
77 torch::DeviceType dd_input_device; in ModelInference_Torch() local83 PetscCall(PetscMemTypeToDeviceType(input_mem_type, &dd_input_device)); in ModelInference_Torch()85 PetscCallCXX(options = torch::TensorOptions().dtype(torch::kFloat64).device(dd_input_device)); in ModelInference_Torch()86 if (dd_input_device == torch::kXPU) { // XPU requires device-to-host-to-device transfer in ModelInference_Torch()88 …_inputs_ptr, {num_nodes, num_input_comps}, {num_input_comps, 1}, nullptr, options, dd_input_device) in ModelInference_Torch()