diff --git a/lite/load_and_run/src/options/device_options.cpp b/lite/load_and_run/src/options/device_options.cpp index 49771baf..fcdb03af 100644 --- a/lite/load_and_run/src/options/device_options.cpp +++ b/lite/load_and_run/src/options/device_options.cpp @@ -80,7 +80,9 @@ void XPUDeviceOption::config_model_internel( if (enable_cuda) { mgb_log_warn("using cuda device\n"); model->get_mdl_config().comp_node_mapper = [](mgb::CompNode::Locator& loc) { - loc.type = mgb::CompNode::DeviceType::CUDA; + if (loc.type == mgb::CompNode::DeviceType::UNSPEC) { + loc.type = mgb::CompNode::DeviceType::CUDA; + } loc.device = 0; }; }