From 8f7fa90cc74c91b2ebf4778b9972883495547190 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Thu, 17 Mar 2022 13:52:53 +0800
Subject: [PATCH] fix(lite): fix lar cuda option crash for mdl model

GitOrigin-RevId: f886b42eae7efb7d55b198d8d3acf94ba1e9e1f4
---
 lite/load_and_run/src/options/device_options.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lite/load_and_run/src/options/device_options.cpp b/lite/load_and_run/src/options/device_options.cpp
index 49771bafc..fcdb03af4 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<ModelMdl>(
         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;
             };
         }
-- 
GitLab