Use if_google/oss() to handle config_setting differences in //tensorflow:.
PiperOrigin-RevId: 360925485 Change-Id: I9044df678a138b316cd10a621f47c510a2ac5727
This commit is contained in:
parent
1a61eebcc7
commit
033c60b5b9
151
tensorflow/BUILD
151
tensorflow/BUILD
@ -7,6 +7,7 @@ load("@bazel_skylib//rules:common_settings.bzl", "bool_flag", "bool_setting")
|
|||||||
load(
|
load(
|
||||||
"//tensorflow:tensorflow.bzl",
|
"//tensorflow:tensorflow.bzl",
|
||||||
"VERSION",
|
"VERSION",
|
||||||
|
"if_google",
|
||||||
"if_oss",
|
"if_oss",
|
||||||
"tf_cc_shared_object",
|
"tf_cc_shared_object",
|
||||||
"tf_custom_op_library_additional_deps_impl",
|
"tf_custom_op_library_additional_deps_impl",
|
||||||
@ -90,34 +91,59 @@ config_setting(
|
|||||||
# Config setting for determining if we are building for Android.
|
# Config setting for determining if we are building for Android.
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "android",
|
name = "android",
|
||||||
values = {"crosstool_top": "//external:android/crosstool"},
|
flag_values = if_google(
|
||||||
|
{"//tools/cpp:cc_target_os": "android"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
|
values = if_oss(
|
||||||
|
{"crosstool_top": "//external:android/crosstool"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "android_x86",
|
name = "android_x86",
|
||||||
values = {
|
flag_values = if_google(
|
||||||
"crosstool_top": "//external:android/crosstool",
|
{"//tools/cpp:cc_target_os": "android"},
|
||||||
"cpu": "x86",
|
{},
|
||||||
},
|
),
|
||||||
|
values = dict(
|
||||||
|
if_oss(
|
||||||
|
{"crosstool_top": "//external:android/crosstool"},
|
||||||
|
),
|
||||||
|
cpu = "x86",
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "android_x86_64",
|
name = "android_x86_64",
|
||||||
values = {
|
flag_values = if_google(
|
||||||
"crosstool_top": "//external:android/crosstool",
|
{"//tools/cpp:cc_target_os": "android"},
|
||||||
"cpu": "x86_64",
|
{},
|
||||||
},
|
),
|
||||||
|
values = dict(
|
||||||
|
if_oss(
|
||||||
|
{"crosstool_top": "//external:android/crosstool"},
|
||||||
|
),
|
||||||
|
cpu = "x86_64",
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "android_armeabi",
|
name = "android_armeabi",
|
||||||
values = {
|
flag_values = if_google(
|
||||||
"crosstool_top": "//external:android/crosstool",
|
{"//tools/cpp:cc_target_os": "android"},
|
||||||
"cpu": "armeabi",
|
{},
|
||||||
},
|
),
|
||||||
|
values = dict(
|
||||||
|
if_oss(
|
||||||
|
{"crosstool_top": "//external:android/crosstool"},
|
||||||
|
),
|
||||||
|
cpu = "armeabi",
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -146,7 +172,14 @@ config_setting(
|
|||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "emscripten",
|
name = "emscripten",
|
||||||
values = {"crosstool_top": "//external:android/emscripten"},
|
flag_values = if_google(
|
||||||
|
{"//tools/cpp:cc_target_os": "emscripten"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
|
values = if_oss(
|
||||||
|
{"crosstool_top": "//external:android/emscripten"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -161,19 +194,31 @@ config_setting(
|
|||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "android_arm",
|
name = "android_arm",
|
||||||
values = {
|
flag_values = if_google(
|
||||||
"crosstool_top": "//external:android/crosstool",
|
{"//tools/cpp:cc_target_os": "android"},
|
||||||
"cpu": "armeabi-v7a",
|
{},
|
||||||
},
|
),
|
||||||
|
values = dict(
|
||||||
|
if_oss(
|
||||||
|
{"crosstool_top": "//external:android/crosstool"},
|
||||||
|
),
|
||||||
|
cpu = "armeabi-v7a",
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "android_arm64",
|
name = "android_arm64",
|
||||||
values = {
|
flag_values = if_google(
|
||||||
"crosstool_top": "//external:android/crosstool",
|
{"//tools/cpp:cc_target_os": "android"},
|
||||||
"cpu": "arm64-v8a",
|
{},
|
||||||
},
|
),
|
||||||
|
values = dict(
|
||||||
|
if_oss(
|
||||||
|
{"crosstool_top": "//external:android/crosstool"},
|
||||||
|
),
|
||||||
|
cpu = "arm64-v8a",
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -198,9 +243,15 @@ config_setting(
|
|||||||
config_setting(
|
config_setting(
|
||||||
name = "windows",
|
name = "windows",
|
||||||
# Internal builds query the target OS.
|
# Internal builds query the target OS.
|
||||||
# copybara:uncomment flag_values = {"//tools/cpp:cc_target_os": "windows"},
|
flag_values = if_google(
|
||||||
|
{"//tools/cpp:cc_target_os": "windows"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
# OSS builds query the CPU type.
|
# OSS builds query the CPU type.
|
||||||
values = {"cpu": "x64_windows"}, # copybara:comment
|
values = if_oss(
|
||||||
|
{"cpu": "x64_windows"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -221,15 +272,23 @@ config_setting(
|
|||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "macos_x86_64",
|
name = "macos_x86_64",
|
||||||
|
flag_values = if_google(
|
||||||
|
{"//tools/cpp:cc_target_os": "apple"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
values = {
|
values = {
|
||||||
"apple_platform_type": "macos",
|
"apple_platform_type": "macos",
|
||||||
"cpu": "darwin",
|
"cpu": if_google("darwin_x86_64", "darwin"),
|
||||||
},
|
},
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "macos_arm64",
|
name = "macos_arm64",
|
||||||
|
flag_values = if_google(
|
||||||
|
{"//tools/cpp:cc_target_os": "apple"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
values = {
|
values = {
|
||||||
"apple_platform_type": "macos",
|
"apple_platform_type": "macos",
|
||||||
"cpu": "darwin_arm64",
|
"cpu": "darwin_arm64",
|
||||||
@ -248,28 +307,56 @@ selects.config_setting_group(
|
|||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "ios",
|
name = "ios",
|
||||||
values = {"apple_platform_type": "ios"},
|
flag_values = if_google(
|
||||||
|
{"//tools/cpp:cc_target_os": "apple"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
|
values = if_oss(
|
||||||
|
{"apple_platform_type": "ios"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "fuchsia",
|
name = "fuchsia",
|
||||||
values = {"cpu": "fuchsia"},
|
flag_values = if_google(
|
||||||
|
{"//tools/cpp:cc_target_os": "fuchsia"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
|
values = if_oss(
|
||||||
|
# TODO(b/149248802) When we have a Fuchsia Bazel SDK update to use the values it sets.
|
||||||
|
{"cpu": "fuchsia"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "ios_x86_64",
|
name = "ios_x86_64",
|
||||||
values = {
|
flag_values = if_google(
|
||||||
"crosstool_top": "//tools/osx/crosstool:crosstool",
|
{"//tools/cpp:cc_target_os": "apple"},
|
||||||
"cpu": "ios_x86_64",
|
{},
|
||||||
},
|
),
|
||||||
|
values = dict(
|
||||||
|
if_oss(
|
||||||
|
{"crosstool_top": "//tools/osx/crosstool:crosstool"},
|
||||||
|
),
|
||||||
|
cpu = "ios_x86_64",
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "chromiumos",
|
name = "chromiumos",
|
||||||
values = {"crosstool_top": "//external:android/chromiumos"},
|
flag_values = if_google(
|
||||||
|
{"//tools/cpp:cc_target_os": "chromiumos"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
|
values = if_oss(
|
||||||
|
{"crosstool_top": "//external:android/chromiumos"},
|
||||||
|
{},
|
||||||
|
),
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user