diff --git a/mace/test/BUILD b/mace/test/BUILD index 7c453ed869fd52720bab18521e649e1dd3417ea6..e23a170097bfe46dc46e42721d04681e17d72005 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", ], )