From db1ab2cd2197668babac524ccc9599af989256df Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Thu, 28 Mar 2019 18:00:54 +0800
Subject: [PATCH] Fix data format bug in MaceTensor.

---
 .travis.yml                          | 2 +-
 mace/core/runtime/cpu/cpu_runtime.cc | 1 +
 mace/libmace/mace.cc                 | 7 ++++---
 3 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index 629ee5bd..b0ba55e0 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -125,7 +125,7 @@ jobs:
         - python tools/bazel_adb_run.py --target="//mace/test:mace_api_test" --run_target=False --target_abis=arm64-v8a || exit 1
         - python tools/bazel_adb_run.py --target="//mace/test:mace_api_mt_test" --run_target=False --target_abis=arm64-v8a || exit 1
         - echo 'Extra Test on ARM64'
-        - python tools/bazel_adb_run.py --target="//mace/utils:tuner_test" --run_target=False --target_abis=arm64-v8a || exit 1
+        - python tools/bazel_adb_run.py --target="//mace/utils:utils_test" --run_target=False --target_abis=arm64-v8a || exit 1
       env: TYPE=Extra-Test-ARM64-v8a
       os: linux
       dist: xenial
diff --git a/mace/core/runtime/cpu/cpu_runtime.cc b/mace/core/runtime/cpu/cpu_runtime.cc
index 5db5b36b..e5742ec1 100644
--- a/mace/core/runtime/cpu/cpu_runtime.cc
+++ b/mace/core/runtime/cpu/cpu_runtime.cc
@@ -20,6 +20,7 @@
 
 #include <algorithm>
 #include <cerrno>
+#include <cmath>
 #include <cstring>
 #include <fstream>
 #include <string>
diff --git a/mace/libmace/mace.cc b/mace/libmace/mace.cc
index 927930fe..022f8e35 100644
--- a/mace/libmace/mace.cc
+++ b/mace/libmace/mace.cc
@@ -293,9 +293,10 @@ MaceTensor::MaceTensor(const std::vector<int64_t> &shape,
                        std::shared_ptr<float> data,
                        const DataFormat format) {
   MACE_CHECK_NOTNULL(data.get());
-  MACE_CHECK(format == DataFormat::NHWC || format == DataFormat::NCHW
-                 || format == OIHW,
-             "MACE only support NHWC, NCHW and OIHW formats of input now.");
+  MACE_CHECK(format == DataFormat::DF_NONE || format == DataFormat::NHWC
+                 || format == DataFormat::NCHW || format == OIHW,
+             "MACE only support DF_NONE, NHWC, NCHW and OIHW "
+             "formats of input now.");
   impl_ = make_unique<MaceTensor::Impl>();
   impl_->shape = shape;
   impl_->data = data;
-- 
GitLab