From 68baaf56d7ddc64200ce7721131157015f4723ed Mon Sep 17 00:00:00 2001 From: wuchenghui Date: Mon, 7 May 2018 17:54:35 +0800 Subject: [PATCH] fix dsp --- mace/core/runtime/hexagon/hexagon_controller.h | 2 +- mace/core/runtime/hexagon/hexagon_nn_ops.h | 4 ++-- tools/mace_tools.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mace/core/runtime/hexagon/hexagon_controller.h b/mace/core/runtime/hexagon/hexagon_controller.h index 1bf57f9d..07b7a453 100644 --- a/mace/core/runtime/hexagon/hexagon_controller.h +++ b/mace/core/runtime/hexagon/hexagon_controller.h @@ -15,7 +15,7 @@ #ifndef MACE_CORE_RUNTIME_HEXAGON_HEXAGON_CONTROLLER_H_ #define MACE_CORE_RUNTIME_HEXAGON_HEXAGON_CONTROLLER_H_ -#include "mace/core/runtime/hexagon/hexagon_nn.h" +#include "mace/third_party/nnlib/hexagon_nn.h" #ifdef __cplusplus extern "C" { diff --git a/mace/core/runtime/hexagon/hexagon_nn_ops.h b/mace/core/runtime/hexagon/hexagon_nn_ops.h index dba6c56e..d7d97cb0 100644 --- a/mace/core/runtime/hexagon/hexagon_nn_ops.h +++ b/mace/core/runtime/hexagon/hexagon_nn_ops.h @@ -27,7 +27,7 @@ namespace mace { typedef enum op_type_enum { #define DEF_OP(NAME, ...) OP_##NAME, -#include "mace/core/runtime/hexagon/ops.h" // NOLINT(build/include) +#include "mace/third_party/nnlib/ops.h" // NOLINT(build/include) NN_OPS_MAX #undef DEF_OP @@ -38,7 +38,7 @@ class OpMap { void Init() { #define DEF_OP(NAME) op_map_[#NAME] = OP_##NAME; -#include "mace/core/runtime/hexagon/ops.h" // NOLINT(build/include) +#include "mace/third_party/nnlib/ops.h" // NOLINT(build/include) #undef DEF_OP } diff --git a/tools/mace_tools.py b/tools/mace_tools.py index 231e158a..a2874610 100644 --- a/tools/mace_tools.py +++ b/tools/mace_tools.py @@ -41,7 +41,7 @@ def get_target_socs(configs): else: available_socs = sh_commands.adb_get_all_socs() target_socs = available_socs - if hasattr(configs, "target_socs"): + if "target_socs" in configs: target_socs = set(configs["target_socs"]) target_socs = target_socs & available_socs -- GitLab