[aarch64] make aws sdk work on aarch64

`bazel build //tensorflow/tools/pip_package:build_pip_package'
requires AWS SDK by default. but platform part was not built
on aarch64
This commit is contained in:
Koan-Sin Tan 2018-10-10 02:34:02 +00:00
parent 129bb5e845
commit 3f88ddb71b
2 changed files with 9 additions and 0 deletions

View File

@ -163,6 +163,12 @@ config_setting(
visibility = ["//visibility:public"],
)
config_setting(
name = "linux_aarch64",
values = {"cpu": "aarch64"},
visibility = ["//visibility:public"],
)
config_setting(
name = "linux_x86_64",
values = {"cpu": "k8"},

View File

@ -12,6 +12,9 @@ load("@org_tensorflow//third_party:common.bzl", "template_rule")
cc_library(
name = "aws",
srcs = select({
"@org_tensorflow//tensorflow:linux_aarch64": glob([
"aws-cpp-sdk-core/source/platform/linux-shared/*.cpp",
]),
"@org_tensorflow//tensorflow:linux_x86_64": glob([
"aws-cpp-sdk-core/source/platform/linux-shared/*.cpp",
]),