From b0bfe7f6b5d62bb023a220a08c0cda4e6406a986 Mon Sep 17 00:00:00 2001 From: liuqi Date: Thu, 19 Apr 2018 17:28:26 +0800 Subject: [PATCH] Fix api test BUILD file. --- mace/test/BUILD | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/mace/test/BUILD b/mace/test/BUILD index 7c453ed8..e23a1700 100644 --- a/mace/test/BUILD +++ b/mace/test/BUILD @@ -7,19 +7,23 @@ package( licenses(["notice"]) # Apache 2.0 +load("//mace:mace.bzl", "if_android", "if_neon_enabled", "if_openmp_enabled", "if_android_armv7", "if_hexagon_enabled") + cc_test( name = "mace_api_test", testonly = 1, - srcs = glob( - ["mace_api_test.cc"], - ), + srcs = ["mace_api_test.cc"], + copts = if_openmp_enabled(["-fopenmp"]) + + if_neon_enabled(["-DMACE_ENABLE_NEON"]) + + if_android_armv7(["-mfpu=neon -mfloat-abi=softfp"]) + + if_android(["-DMACE_ENABLE_OPENCL",]) + + if_hexagon_enabled(["-DMACE_ENABLE_HEXAGON"]), linkopts = ["-fopenmp"], linkstatic = 1, deps = [ - "//mace/core:core", + "//mace/ops:test", "//mace/kernels:kernels", "//mace/ops:ops", - "//mace/ops:test", "@gtest//:gtest_main", ], ) -- GitLab