提交 62d8ba37 编写于 作者: 卢旭辉

Merge branch 'perf' into 'master'

perf: Prefer GPU if input ops on different devices

See merge request applied-machine-learning/sysml/mace!1297
...@@ -41,10 +41,10 @@ DeviceType NetOptimizer::SelectBestDevice( ...@@ -41,10 +41,10 @@ DeviceType NetOptimizer::SelectBestDevice(
} }
// Greedy strategy: Use input op's device type as current op's device // Greedy strategy: Use input op's device type as current op's device
for (auto device_type : inputs_op_devices) { for (auto device_type : inputs_op_devices) {
if (device_type != best_device) { if (device_type == best_device) {
best_device = device_type; return best_device;
} }
} }
return best_device; return DeviceType::CPU;
} }
} // namespace mace } // namespace mace
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册