diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 8e2ddcb8249..a04ddf9f8a1 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -184,6 +184,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"}, diff --git a/third_party/aws/BUILD.bazel b/third_party/aws/BUILD.bazel index e59e80feaf1..26ff2282fd0 100644 --- a/third_party/aws/BUILD.bazel +++ b/third_party/aws/BUILD.bazel @@ -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", ]),