[opt]add precision type into opt test=develop (#3151)

3501f3c2 · huzhiqiang · GitHub · c7fb4138 · 3501f3c2
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

lite/api/opt.cc lite/api/opt.cc +4 -1

未找到文件。
--- a/lite/api/opt.cc
+++ b/lite/api/opt.cc
@@ -88,7 +88,10 @@ std::vector<Place> ParserValidPlaces() {
  auto target_reprs = lite::Split(FLAGS_valid_targets, ",");
  for (auto& target_repr : target_reprs) {
    if (target_repr == "arm") {
-      valid_places.emplace_back(TARGET(kARM));
+      valid_places.emplace_back(
+          Place{TARGET(kARM), PRECISION(kFloat), DATALAYOUT(kNCHW)});
+      valid_places.emplace_back(
+          Place{TARGET(kARM), PRECISION(kInt32), DATALAYOUT(kNCHW)});
    } else if (target_repr == "opencl") {
      valid_places.emplace_back(
          Place{TARGET(kOpenCL), PRECISION(kFP16), DATALAYOUT(kImageDefault)});