Browse Source

fix(lite): fix lar cuda option crash for mdl model

GitOrigin-RevId: f886b42eae
tags/v1.9.0
Megvii Engine Team 3 years ago
parent
commit
8f7fa90cc7
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      lite/load_and_run/src/options/device_options.cpp

+ 3
- 1
lite/load_and_run/src/options/device_options.cpp View File

@@ -80,7 +80,9 @@ void XPUDeviceOption::config_model_internel<ModelMdl>(
if (enable_cuda) { if (enable_cuda) {
mgb_log_warn("using cuda device\n"); mgb_log_warn("using cuda device\n");
model->get_mdl_config().comp_node_mapper = [](mgb::CompNode::Locator& loc) { 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; loc.device = 0;
}; };
} }


Loading…
Cancel
Save