提交 50636e91 编写于 作者: Y yejianwu

fix deps in tests

上级 209f1cb3
...@@ -52,6 +52,7 @@ api_test: ...@@ -52,6 +52,7 @@ api_test:
- if [ -z "$TARGET_SOCS" ]; then TARGET_SOCS=random; fi - if [ -z "$TARGET_SOCS" ]; then TARGET_SOCS=random; fi
- python tools/bazel_adb_run.py --target="//mace/test:mace_api_test" --run_target=True --stdout_processor=unittest_stdout_processor --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS - python tools/bazel_adb_run.py --target="//mace/test:mace_api_test" --run_target=True --stdout_processor=unittest_stdout_processor --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS
- python tools/bazel_adb_run.py --target="//mace/test:mace_api_mt_test" --run_target=True --stdout_processor=unittest_stdout_processor --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS - python tools/bazel_adb_run.py --target="//mace/test:mace_api_mt_test" --run_target=True --stdout_processor=unittest_stdout_processor --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS
- python tools/bazel_adb_run.py --target="//mace/test:mace_api_exception_test" --run_target=True --stdout_processor=unittest_stdout_processor --target_abis=armeabi-v7a,arm64-v8a --target_socs=$TARGET_SOCS
ops_benchmark: ops_benchmark:
stage: ops_benchmark stage: ops_benchmark
......
...@@ -51,7 +51,7 @@ TEST(CoreTest, INIT_MODE) { ...@@ -51,7 +51,7 @@ TEST(CoreTest, INIT_MODE) {
for (auto &op_def : op_defs) { for (auto &op_def : op_defs) {
net_def.add_op()->CopyFrom(op_def); net_def.add_op()->CopyFrom(op_def);
} }
std::shared_ptr<OperatorRegistryBase> op_registry(new OperatorRegistryBase()); std::shared_ptr<OperatorRegistryBase> op_registry(new OperatorRegistry());
auto net = auto net =
CreateNet(op_registry, net_def, &ws, DeviceType::GPU, NetMode::INIT); CreateNet(op_registry, net_def, &ws, DeviceType::GPU, NetMode::INIT);
net->Run(); net->Run();
......
...@@ -29,6 +29,7 @@ ...@@ -29,6 +29,7 @@
#include "mace/core/tensor.h" #include "mace/core/tensor.h"
#include "mace/core/workspace.h" #include "mace/core/workspace.h"
#include "mace/kernels/opencl/helper.h" #include "mace/kernels/opencl/helper.h"
#include "mace/ops/ops_register.h"
#include "mace/utils/utils.h" #include "mace/utils/utils.h"
namespace mace { namespace mace {
...@@ -110,7 +111,7 @@ class OpDefBuilder { ...@@ -110,7 +111,7 @@ class OpDefBuilder {
class OpsTestNet { class OpsTestNet {
public: public:
OpsTestNet() : op_registry_(new OperatorRegistryBase()) {} OpsTestNet() : op_registry_(new OperatorRegistry()) {}
template <DeviceType D, typename T> template <DeviceType D, typename T>
void AddInputFromArray(const std::string &name, void AddInputFromArray(const std::string &name,
......
...@@ -24,8 +24,7 @@ cc_test( ...@@ -24,8 +24,7 @@ cc_test(
linkstatic = 1, linkstatic = 1,
deps = [ deps = [
"//mace/ops:test", "//mace/ops:test",
"//mace/kernels:kernels", "//mace/libmace:libmace",
"//mace/ops:ops",
"@gtest//:gtest_main", "@gtest//:gtest_main",
], ],
) )
...@@ -45,8 +44,7 @@ cc_test( ...@@ -45,8 +44,7 @@ cc_test(
linkstatic = 1, linkstatic = 1,
deps = [ deps = [
"//mace/ops:test", "//mace/ops:test",
"//mace/kernels:kernels", "//mace/libmace:libmace",
"//mace/ops:ops",
"@gtest//:gtest_main", "@gtest//:gtest_main",
], ],
) )
...@@ -66,8 +64,7 @@ cc_test( ...@@ -66,8 +64,7 @@ cc_test(
linkstatic = 1, linkstatic = 1,
deps = [ deps = [
"//mace/ops:test", "//mace/ops:test",
"//mace/kernels:kernels", "//mace/libmace:libmace",
"//mace/ops:ops",
"@gtest//:gtest_main", "@gtest//:gtest_main",
], ],
) )
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册