diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-10-12 15:08:11 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-12 15:15:45 -0700 |
| commit | abb6b8e73cc0c7b566536df52e93a7eb6327b788 (patch) | |
| tree | 6cb36064392c51164654294a25f7256a38ebc807 /tensorflow | |
| parent | 7c2993a2358688f2a4e314ee8c15c40ebc4ce9f1 (diff) | |
Add linux_arm64 and linux_armhf config_setting.
PiperOrigin-RevId: 172018709
Diffstat (limited to 'tensorflow')
| -rw-r--r-- | tensorflow/BUILD | 16 | ||||
| -rw-r--r-- | tensorflow/compiler/aot/tfcompile.bzl | 2 | ||||
| -rw-r--r-- | tensorflow/tensorflow.bzl | 6 |
3 files changed, 24 insertions, 0 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 64758dee0e..9d07697d01 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -142,6 +142,22 @@ config_setting( ) config_setting( + name = "linux_armhf", + values = { + "cpu": "armeabi-v7a", + }, + visibility = ["//visibility:public"], +) + +config_setting( + name = "linux_arm64", + values = { + "cpu": "arm64-v8a", + }, + visibility = ["//visibility:public"], +) + +config_setting( name = "debug", values = { "compilation_mode": "dbg", diff --git a/tensorflow/compiler/aot/tfcompile.bzl b/tensorflow/compiler/aot/tfcompile.bzl index 4888760acd..c900d201d2 100644 --- a/tensorflow/compiler/aot/tfcompile.bzl +++ b/tensorflow/compiler/aot/tfcompile.bzl @@ -302,6 +302,8 @@ def target_llvm_triple(): "//tensorflow:android_arm": "armv7-none-android", "//tensorflow:android_arm64": "aarch64-none-android", "//tensorflow:android_x86": "i686-none-android", + "//tensorflow:linux_armhf": "armv7-none-linux-gnueabihf", + "//tensorflow:linux_arm64": "aarch64-none-linux-gnu", "//tensorflow:linux_ppc64le": "ppc64le-ibm-linux-gnu", "//tensorflow:darwin": "x86_64-none-darwin", "//conditions:default": "x86_64-pc-linux", diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 0f074151db..5ec31e492c 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -112,6 +112,9 @@ def if_ios(a): def if_mobile(a): return select({ clean_dep("//tensorflow:android"): a, + # Treat arm linux devices as mobile. + clean_dep("//tensorflow:linux_arm64"): a, + clean_dep("//tensorflow:linux_armhf"): a, clean_dep("//tensorflow:ios"): a, "//conditions:default": [], }) @@ -120,6 +123,9 @@ def if_mobile(a): def if_not_mobile(a): return select({ clean_dep("//tensorflow:android"): [], + # Treat arm linux devices as mobile. + clean_dep("//tensorflow:linux_arm64"): [], + clean_dep("//tensorflow:linux_armhf"): [], clean_dep("//tensorflow:ios"): [], "//conditions:default": a, }) |