Rollback of 'Reland "Fix issues with 32-bit ARM builds"'
PiperOrigin-RevId: 320495531 Change-Id: I7aa19da802163f705bf706e42918766f61e8bab1
This commit is contained in:
parent
82e1db2bba
commit
ad35731ffd
@ -260,36 +260,6 @@ config_setting(
|
|||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
|
||||||
name = "armeabi",
|
|
||||||
values = {"cpu": "armeabi"},
|
|
||||||
visibility = ["//visibility:public"],
|
|
||||||
)
|
|
||||||
|
|
||||||
config_setting(
|
|
||||||
name = "armeabi-v7a",
|
|
||||||
values = {"cpu": "armeabi-v7a"},
|
|
||||||
visibility = ["//visibility:public"],
|
|
||||||
)
|
|
||||||
|
|
||||||
config_setting(
|
|
||||||
name = "arm64-v8a",
|
|
||||||
values = {"cpu": "arm64-v8a"},
|
|
||||||
visibility = ["//visibility:public"],
|
|
||||||
)
|
|
||||||
|
|
||||||
selects.config_setting_group(
|
|
||||||
name = "arm_any",
|
|
||||||
match_any = [
|
|
||||||
":arm",
|
|
||||||
":armeabi",
|
|
||||||
":armeabi-v7a",
|
|
||||||
":arm64-v8a",
|
|
||||||
":linux_aarch64",
|
|
||||||
":linux_armhf",
|
|
||||||
],
|
|
||||||
)
|
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "freebsd",
|
name = "freebsd",
|
||||||
values = {"cpu": "freebsd"},
|
values = {"cpu": "freebsd"},
|
||||||
|
@ -817,9 +817,10 @@ cc_library(
|
|||||||
srcs = ["eigen_contraction_kernel.cc"],
|
srcs = ["eigen_contraction_kernel.cc"],
|
||||||
hdrs = ["eigen_contraction_kernel.h"],
|
hdrs = ["eigen_contraction_kernel.h"],
|
||||||
defines = select({
|
defines = select({
|
||||||
"//tensorflow:android_x86": [],
|
"//tensorflow:android": [],
|
||||||
"//tensorflow:arm_any": [],
|
"//tensorflow:arm": [],
|
||||||
"//tensorflow:ios": [],
|
"//tensorflow:ios": [],
|
||||||
|
"//tensorflow:linux_aarch64": [],
|
||||||
"//tensorflow:linux_ppc64le": [],
|
"//tensorflow:linux_ppc64le": [],
|
||||||
"//conditions:default": [
|
"//conditions:default": [
|
||||||
"TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL",
|
"TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL",
|
||||||
@ -831,9 +832,10 @@ cc_library(
|
|||||||
"//third_party/eigen3",
|
"//third_party/eigen3",
|
||||||
"//tensorflow/core/platform:dynamic_annotations",
|
"//tensorflow/core/platform:dynamic_annotations",
|
||||||
] + select({
|
] + select({
|
||||||
"//tensorflow:android_x86": [],
|
"//tensorflow:android": [],
|
||||||
"//tensorflow:arm_any": [],
|
"//tensorflow:arm": [],
|
||||||
"//tensorflow:ios": [],
|
"//tensorflow:ios": [],
|
||||||
|
"//tensorflow:linux_aarch64": [],
|
||||||
"//tensorflow:linux_ppc64le": [],
|
"//tensorflow:linux_ppc64le": [],
|
||||||
"//conditions:default": ["@mkl_dnn//:mkldnn_single_threaded"],
|
"//conditions:default": ["@mkl_dnn//:mkldnn_single_threaded"],
|
||||||
}),
|
}),
|
||||||
@ -3222,8 +3224,8 @@ tf_cc_test(
|
|||||||
name = "eigen_mkldnn_contraction_kernel_test",
|
name = "eigen_mkldnn_contraction_kernel_test",
|
||||||
size = "small",
|
size = "small",
|
||||||
srcs = select({
|
srcs = select({
|
||||||
"//tensorflow:android_x86": [],
|
"//tensorflow:android": [],
|
||||||
"//tensorflow:arm_any": [],
|
"//tensorflow:arm": [],
|
||||||
"//tensorflow:ios": [],
|
"//tensorflow:ios": [],
|
||||||
"//tensorflow:linux_ppc64le": [],
|
"//tensorflow:linux_ppc64le": [],
|
||||||
":no_mkldnn_contraction_kernel": [],
|
":no_mkldnn_contraction_kernel": [],
|
||||||
|
@ -41,22 +41,18 @@ limitations under the License.
|
|||||||
#elif defined(_WIN32)
|
#elif defined(_WIN32)
|
||||||
#define PLATFORM_WINDOWS
|
#define PLATFORM_WINDOWS
|
||||||
|
|
||||||
|
#elif defined(__arm__)
|
||||||
|
#define PLATFORM_POSIX
|
||||||
|
|
||||||
#elif defined(__EMSCRIPTEN__)
|
#elif defined(__EMSCRIPTEN__)
|
||||||
#define PLATFORM_PORTABLE_GOOGLE
|
#define PLATFORM_PORTABLE_GOOGLE
|
||||||
#define PLATFORM_POSIX
|
#define PLATFORM_POSIX
|
||||||
// EMSCRIPTEN builds are considered "mobile" for the sake of portability.
|
|
||||||
#define IS_MOBILE_PLATFORM
|
|
||||||
|
|
||||||
#elif defined(__arm__) || defined(__aarch64__)
|
|
||||||
// If no platform specified, use:
|
|
||||||
#define PLATFORM_POSIX
|
|
||||||
|
|
||||||
// Require an outside macro to tell us if we're building for Raspberry Pi or
|
// Require an outside macro to tell us if we're building for Raspberry Pi or
|
||||||
// another ARM device that's not a mobile platform.
|
// another ARM device that's not a mobile platform.
|
||||||
#if !defined(RASPBERRY_PI) && !defined(ARM_NON_MOBILE) && \
|
#if !defined(RASPBERRY_PI) && !defined(ARM_NON_MOBILE)
|
||||||
!defined(PLATFORM_GOOGLE)
|
|
||||||
#define IS_MOBILE_PLATFORM
|
#define IS_MOBILE_PLATFORM
|
||||||
#endif
|
#endif // !defined(RASPBERRY_PI) && !defined(ARM_NON_MOBILE)
|
||||||
|
|
||||||
#else
|
#else
|
||||||
// If no platform specified, use:
|
// If no platform specified, use:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user