From cfb8cb1bb1c198dbd1010d0adfa942c94548c3a6 Mon Sep 17 00:00:00 2001 From: Koan-Sin Tan Date: Fri, 14 Aug 2020 20:47:30 +0800 Subject: [PATCH] make `//tensorflow/lite/kernels:all` build make ``` bazel build --config android_arm64 //tensorflow/lite/kernels:all ``` works --- tensorflow/lite/kernels/BUILD | 19 +++++++++++++++++++ tensorflow/lite/kernels/kernel_util_test.cc | 2 +- 2 files changed, 20 insertions(+), 1 deletion(-) diff --git a/tensorflow/lite/kernels/BUILD b/tensorflow/lite/kernels/BUILD index a56d370afeb..f1aa5fa04fd 100644 --- a/tensorflow/lite/kernels/BUILD +++ b/tensorflow/lite/kernels/BUILD @@ -1,4 +1,5 @@ load("//tensorflow/lite:build_def.bzl", "tflite_copts") +load("//tensorflow/lite:build_def.bzl", "tflite_linkopts") load("//tensorflow/lite/micro:build_def.bzl", "micro_copts") load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite_combined") load("//tensorflow:tensorflow.bzl", "tf_opts_nortti_if_android") @@ -150,6 +151,12 @@ cc_library( "@com_google_absl//absl/types:optional", "@com_googlesource_code_re2//:re2", ], + linkopts = tflite_linkopts() + select({ + "//tensorflow:android": [ + "-lm", + ], + "//conditions:default": [], + }), ) cc_test( @@ -277,6 +284,12 @@ cc_test( "//third_party/eigen3", "@com_google_googletest//:gtest", ], + linkopts = tflite_linkopts() + select({ + "//tensorflow:android": [ + "-lm", + ], + "//conditions:default": [], + }), ) cc_library( @@ -460,6 +473,12 @@ cc_test( "//tensorflow/lite/testing:util", "@com_google_googletest//:gtest", ], + linkopts = tflite_linkopts() + select({ + "//tensorflow:android": [ + "-lm" + ], + "//conditions:default": [], + }), ) cc_test( diff --git a/tensorflow/lite/kernels/kernel_util_test.cc b/tensorflow/lite/kernels/kernel_util_test.cc index db0cc3cb39c..cd49a7797ec 100644 --- a/tensorflow/lite/kernels/kernel_util_test.cc +++ b/tensorflow/lite/kernels/kernel_util_test.cc @@ -14,11 +14,11 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/lite/kernels/kernel_util.h" -#include #include #include #include +#include #include #include