experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Support "armv7a" cpu value in XNNPACK, pthreadpool, and cpuinfo

Browse Source 
PiperOrigin-RevId: 315781171
Change-Id: Ide3d8af216edcec3e4a332f3a24c1340ccb95f42
This commit is contained in:
Marat Dukhan 2020-06-10 15:27:31 -07:00 committed by TensorFlower Gardener
parent 3ba9124877
commit 94bd4f7cf6
2 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/workspace.bzl
View File

@ -164,11 +164,11 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
tf_http_archive( tf_http_archive(
name = "XNNPACK", name = "XNNPACK",
sha256 = "945c0e3c1c3a4cc58f99a5ccf3cdbcf710db99f2ef7f7821eb6be93e59e53534", sha256 = "ce44ddcf8b54d8ee87d7aee3a93a8cc949440547c41a8ab23fbf8a66314ef7ce",
strip_prefix = "XNNPACK-5fab409eeff66b0cca5d0482c69b322f2fff6880", strip_prefix = "XNNPACK-24567894452324bbc369e67b23fd86d87506511a",
urls = [ urls = [
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/google/XNNPACK/archive/5fab409eeff66b0cca5d0482c69b322f2fff6880.zip", "https://storage.googleapis.com/mirror.tensorflow.org/github.com/google/XNNPACK/archive/24567894452324bbc369e67b23fd86d87506511a.zip",
"https://github.com/google/XNNPACK/archive/5fab409eeff66b0cca5d0482c69b322f2fff6880.zip", "https://github.com/google/XNNPACK/archive/24567894452324bbc369e67b23fd86d87506511a.zip",
], ],
) )
@ -184,11 +184,11 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
tf_http_archive( tf_http_archive(
name = "pthreadpool", name = "pthreadpool",
sha256 = "9f5fb7f87dc778d9c1d638826344b762afa23884d0252526337ae710264faef3", sha256 = "c4d4a16053ec0e5125dbd8ae1d6d2ba99601d6fdcf8601a0d51d02a048c40348",
strip_prefix = "pthreadpool-18a7156cb9be8e534acefade42e46d4209600c35", strip_prefix = "pthreadpool-e1642461b3b0217d23d6664d839a060f54e4e652",
urls = [ urls = [
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/Maratyszcza/pthreadpool/archive/18a7156cb9be8e534acefade42e46d4209600c35.zip", "https://storage.googleapis.com/mirror.tensorflow.org/github.com/Maratyszcza/pthreadpool/archive/e1642461b3b0217d23d6664d839a060f54e4e652.zip",
"https://github.com/Maratyszcza/pthreadpool/archive/18a7156cb9be8e534acefade42e46d4209600c35.zip", "https://github.com/Maratyszcza/pthreadpool/archive/e1642461b3b0217d23d6664d839a060f54e4e652.zip",
], ],
) )

6
third_party/cpuinfo/BUILD.bazel vendored
View File

@ -99,6 +99,7 @@ cc_library(
":linux_x86_64": COMMON_SRCS + X86_SRCS + LINUX_SRCS + LINUX_X86_SRCS, ":linux_x86_64": COMMON_SRCS + X86_SRCS + LINUX_SRCS + LINUX_X86_SRCS,
":linux_arm": COMMON_SRCS + ARM_SRCS + LINUX_SRCS + LINUX_ARM32_SRCS, ":linux_arm": COMMON_SRCS + ARM_SRCS + LINUX_SRCS + LINUX_ARM32_SRCS,
":linux_armhf": COMMON_SRCS + ARM_SRCS + LINUX_SRCS + LINUX_ARM32_SRCS, ":linux_armhf": COMMON_SRCS + ARM_SRCS + LINUX_SRCS + LINUX_ARM32_SRCS,
":linux_armv7a": COMMON_SRCS + ARM_SRCS + LINUX_SRCS + LINUX_ARM32_SRCS,
":linux_aarch64": COMMON_SRCS + ARM_SRCS + LINUX_SRCS + LINUX_ARM64_SRCS, ":linux_aarch64": COMMON_SRCS + ARM_SRCS + LINUX_SRCS + LINUX_ARM64_SRCS,
":macos_x86_64": COMMON_SRCS + X86_SRCS + MACH_SRCS + MACH_X86_SRCS, ":macos_x86_64": COMMON_SRCS + X86_SRCS + MACH_SRCS + MACH_X86_SRCS,
":windows_x86_64": COMMON_SRCS + X86_SRCS + WINDOWS_X86_SRCS, ":windows_x86_64": COMMON_SRCS + X86_SRCS + WINDOWS_X86_SRCS,
@ -191,6 +192,11 @@ config_setting(
values = {"cpu": "armhf"}, values = {"cpu": "armhf"},
) )
config_setting(
name = "linux_armv7a",
values = {"cpu": "armv7a"},
)
config_setting( config_setting(
name = "linux_aarch64", name = "linux_aarch64",
values = {"cpu": "aarch64"}, values = {"cpu": "aarch64"},