perf: Prefer GPU if input ops on different devices

bd262fba · Bin Li · b742f824 · bd262fba
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

mace/core/net_optimizer.cc mace/core/net_optimizer.cc +3 -3

未找到文件。
--- a/mace/core/net_optimizer.cc
+++ b/mace/core/net_optimizer.cc
@@ -41,10 +41,10 @@ DeviceType NetOptimizer::SelectBestDevice(
  }
  // Greedy strategy: Use input op's device type as current op's device
  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