Update XNNPACK and its dependencies
- Update XNNPACK, FXdiv, pthreadpool, FP16 libraries - Switch to using upstreamed BUILD.bazel for FXdiv and pthreadpool PiperOrigin-RevId: 306260724 Change-Id: I800f4827ca8da3e8151700e285f9de373230c4b5
This commit is contained in:
parent
517e0274df
commit
da4fba5d6d
@ -22,7 +22,6 @@ load(
|
|||||||
"def_file_filter_configure",
|
"def_file_filter_configure",
|
||||||
)
|
)
|
||||||
load("//third_party/FP16:workspace.bzl", FP16 = "repo")
|
load("//third_party/FP16:workspace.bzl", FP16 = "repo")
|
||||||
load("//third_party/FXdiv:workspace.bzl", FXdiv = "repo")
|
|
||||||
load("//third_party/aws:workspace.bzl", aws = "repo")
|
load("//third_party/aws:workspace.bzl", aws = "repo")
|
||||||
load("//third_party/clog:workspace.bzl", clog = "repo")
|
load("//third_party/clog:workspace.bzl", clog = "repo")
|
||||||
load("//third_party/cpuinfo:workspace.bzl", cpuinfo = "repo")
|
load("//third_party/cpuinfo:workspace.bzl", cpuinfo = "repo")
|
||||||
@ -38,7 +37,6 @@ load("//third_party/opencl_headers:workspace.bzl", opencl_headers = "repo")
|
|||||||
load("//third_party/kissfft:workspace.bzl", kissfft = "repo")
|
load("//third_party/kissfft:workspace.bzl", kissfft = "repo")
|
||||||
load("//third_party/pasta:workspace.bzl", pasta = "repo")
|
load("//third_party/pasta:workspace.bzl", pasta = "repo")
|
||||||
load("//third_party/psimd:workspace.bzl", psimd = "repo")
|
load("//third_party/psimd:workspace.bzl", psimd = "repo")
|
||||||
load("//third_party/pthreadpool:workspace.bzl", pthreadpool = "repo")
|
|
||||||
load("//third_party/ruy:workspace.bzl", ruy = "repo")
|
load("//third_party/ruy:workspace.bzl", ruy = "repo")
|
||||||
load("//third_party/sobol_data:workspace.bzl", sobol_data = "repo")
|
load("//third_party/sobol_data:workspace.bzl", sobol_data = "repo")
|
||||||
load("//third_party/vulkan_headers:workspace.bzl", vulkan_headers = "repo")
|
load("//third_party/vulkan_headers:workspace.bzl", vulkan_headers = "repo")
|
||||||
@ -47,7 +45,6 @@ load("//third_party/toolchains/remote_config:configs.bzl", "initialize_rbe_confi
|
|||||||
def initialize_third_party():
|
def initialize_third_party():
|
||||||
""" Load third party repositories. See above load() statements. """
|
""" Load third party repositories. See above load() statements. """
|
||||||
FP16()
|
FP16()
|
||||||
FXdiv()
|
|
||||||
aws()
|
aws()
|
||||||
clog()
|
clog()
|
||||||
cpuinfo()
|
cpuinfo()
|
||||||
@ -63,7 +60,6 @@ def initialize_third_party():
|
|||||||
opencl_headers()
|
opencl_headers()
|
||||||
pasta()
|
pasta()
|
||||||
psimd()
|
psimd()
|
||||||
pthreadpool()
|
|
||||||
sobol_data()
|
sobol_data()
|
||||||
vulkan_headers()
|
vulkan_headers()
|
||||||
ruy()
|
ruy()
|
||||||
@ -151,11 +147,31 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
|
|||||||
|
|
||||||
tf_http_archive(
|
tf_http_archive(
|
||||||
name = "XNNPACK",
|
name = "XNNPACK",
|
||||||
sha256 = "2afaaf5f866ec714358985b123c3115043b9e099638100937743997f02bbd8cb",
|
sha256 = "246aa56afc5263f1d41fc4a3437ecd51b56f78e16421818961cf79e39431c1df",
|
||||||
strip_prefix = "XNNPACK-05702cf4099ad019ad1abb8ba656bfe04304f32a",
|
strip_prefix = "XNNPACK-b9d07cfa38af15c2abf564c980e00c965857ba21",
|
||||||
urls = [
|
urls = [
|
||||||
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/google/XNNPACK/archive/05702cf4099ad019ad1abb8ba656bfe04304f32a.zip",
|
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/google/XNNPACK/archive/b9d07cfa38af15c2abf564c980e00c965857ba21.zip",
|
||||||
"https://github.com/google/XNNPACK/archive/05702cf4099ad019ad1abb8ba656bfe04304f32a.zip",
|
"https://github.com/google/XNNPACK/archive/b9d07cfa38af15c2abf564c980e00c965857ba21.zip",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
tf_http_archive(
|
||||||
|
name = "FXdiv",
|
||||||
|
sha256 = "8224ff187cdfa178b8c54d36eea70520391781eda16d13a418ab5ae53289e1ab",
|
||||||
|
strip_prefix = "FXdiv-561254d968e5679460e6a0a743206410284d9f46",
|
||||||
|
urls = [
|
||||||
|
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/Maratyszcza/FXdiv/archive/561254d968e5679460e6a0a743206410284d9f46.zip",
|
||||||
|
"https://github.com/Maratyszcza/FXdiv/archive/561254d968e5679460e6a0a743206410284d9f46.zip",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
tf_http_archive(
|
||||||
|
name = "pthreadpool",
|
||||||
|
sha256 = "f894d845cefc091291329712deec85ce7020546f6eaff200b690ae04b6094535",
|
||||||
|
strip_prefix = "pthreadpool-bfa3b9ce6cb71dc8b792e39d24717320a4f92572",
|
||||||
|
urls = [
|
||||||
|
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/Maratyszcza/pthreadpool/archive/bfa3b9ce6cb71dc8b792e39d24717320a4f92572.zip",
|
||||||
|
"https://github.com/Maratyszcza/pthreadpool/archive/bfa3b9ce6cb71dc8b792e39d24717320a4f92572.zip",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
8
third_party/FP16/workspace.bzl
vendored
8
third_party/FP16/workspace.bzl
vendored
@ -5,11 +5,11 @@ load("//third_party:repo.bzl", "third_party_http_archive")
|
|||||||
def repo():
|
def repo():
|
||||||
third_party_http_archive(
|
third_party_http_archive(
|
||||||
name = "FP16",
|
name = "FP16",
|
||||||
strip_prefix = "FP16-febbb1c163726b5db24bed55cc9dc42529068997",
|
strip_prefix = "FP16-3c54eacb74f6f5e39077300c5564156c424d77ba",
|
||||||
sha256 = "3e71681e0a67cd28552aa0bbb78ec6a6bd238216df15336dc1326280f7958de2",
|
sha256 = "0d56bb92f649ec294dbccb13e04865e3c82933b6f6735d1d7145de45da700156",
|
||||||
urls = [
|
urls = [
|
||||||
"https://mirror.bazel.build/github.com/Maratyszcza/FP16/archive/febbb1c163726b5db24bed55cc9dc42529068997.tar.gz",
|
"https://mirror.bazel.build/github.com/Maratyszcza/FP16/archive/3c54eacb74f6f5e39077300c5564156c424d77ba.zip",
|
||||||
"https://github.com/Maratyszcza/FP16/archive/febbb1c163726b5db24bed55cc9dc42529068997.tar.gz",
|
"https://github.com/Maratyszcza/FP16/archive/3c54eacb74f6f5e39077300c5564156c424d77ba.zip",
|
||||||
],
|
],
|
||||||
build_file = "//third_party/FP16:BUILD.bazel",
|
build_file = "//third_party/FP16:BUILD.bazel",
|
||||||
)
|
)
|
||||||
|
1
third_party/FXdiv/BUILD
vendored
1
third_party/FXdiv/BUILD
vendored
@ -1 +0,0 @@
|
|||||||
# This empty BUILD file is required to make Bazel treat this directory as a package.
|
|
15
third_party/FXdiv/BUILD.bazel
vendored
15
third_party/FXdiv/BUILD.bazel
vendored
@ -1,15 +0,0 @@
|
|||||||
# Description:
|
|
||||||
# C99/C++ library for division via fixed-point multiplication by inverse
|
|
||||||
|
|
||||||
package(default_visibility = ["//visibility:public"])
|
|
||||||
|
|
||||||
licenses(["notice"])
|
|
||||||
|
|
||||||
exports_files(["LICENSE"])
|
|
||||||
|
|
||||||
cc_library(
|
|
||||||
name = "FXdiv",
|
|
||||||
hdrs = glob(["include/fxdiv.h"]),
|
|
||||||
includes = ["include"],
|
|
||||||
strip_include_prefix = "include",
|
|
||||||
)
|
|
15
third_party/FXdiv/workspace.bzl
vendored
15
third_party/FXdiv/workspace.bzl
vendored
@ -1,15 +0,0 @@
|
|||||||
"""Loads the FXdiv library, used by XNNPACK & pthreadpool."""
|
|
||||||
|
|
||||||
load("//third_party:repo.bzl", "third_party_http_archive")
|
|
||||||
|
|
||||||
def repo():
|
|
||||||
third_party_http_archive(
|
|
||||||
name = "FXdiv",
|
|
||||||
strip_prefix = "FXdiv-f8c5354679ec2597792bc70a9e06eff50c508b9a",
|
|
||||||
sha256 = "7d3215bea832fe77091ec5666200b91156df6724da1e348205078346325fc45e",
|
|
||||||
urls = [
|
|
||||||
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/Maratyszcza/FXdiv/archive/f8c5354679ec2597792bc70a9e06eff50c508b9a.tar.gz",
|
|
||||||
"https://github.com/Maratyszcza/FXdiv/archive/f8c5354679ec2597792bc70a9e06eff50c508b9a.tar.gz",
|
|
||||||
],
|
|
||||||
build_file = "//third_party/FXdiv:BUILD.bazel",
|
|
||||||
)
|
|
1
third_party/pthreadpool/BUILD
vendored
1
third_party/pthreadpool/BUILD
vendored
@ -1 +0,0 @@
|
|||||||
# This empty BUILD file is required to make Bazel treat this directory as a package.
|
|
76
third_party/pthreadpool/BUILD.bazel
vendored
76
third_party/pthreadpool/BUILD.bazel
vendored
@ -1,76 +0,0 @@
|
|||||||
# Description:
|
|
||||||
# Portable pthread-based thread pool for C and C++
|
|
||||||
|
|
||||||
package(default_visibility = ["//visibility:public"])
|
|
||||||
|
|
||||||
licenses(["notice"])
|
|
||||||
|
|
||||||
exports_files(["LICENSE"])
|
|
||||||
|
|
||||||
cc_library(
|
|
||||||
name = "pthreadpool",
|
|
||||||
srcs = [
|
|
||||||
"src/threadpool-atomics.h",
|
|
||||||
"src/threadpool-pthreads.c",
|
|
||||||
"src/threadpool-utils.h",
|
|
||||||
],
|
|
||||||
hdrs = [
|
|
||||||
"include/pthreadpool.h",
|
|
||||||
],
|
|
||||||
copts = [
|
|
||||||
"-O2",
|
|
||||||
] + select({
|
|
||||||
":optimized_build": ["-O2"],
|
|
||||||
"//conditions:default": [],
|
|
||||||
}) + select({
|
|
||||||
":linux_aarch64": ["-DPTHREADPOOL_USE_CPUINFO=1"],
|
|
||||||
":android_arm64": ["-DPTHREADPOOL_USE_CPUINFO=1"],
|
|
||||||
":android_armv7": ["-DPTHREADPOOL_USE_CPUINFO=1"],
|
|
||||||
"//conditions:default": ["-DPTHREADPOOL_USE_CPUINFO=0"],
|
|
||||||
}),
|
|
||||||
defines = [
|
|
||||||
"PTHREADPOOL_NO_DEPRECATED_API",
|
|
||||||
],
|
|
||||||
includes = [
|
|
||||||
"include",
|
|
||||||
],
|
|
||||||
strip_include_prefix = "include",
|
|
||||||
deps = [
|
|
||||||
"@FXdiv",
|
|
||||||
] + select({
|
|
||||||
":linux_aarch64": ["@cpuinfo"],
|
|
||||||
":android_arm64": ["@cpuinfo"],
|
|
||||||
":android_armv7": ["@cpuinfo"],
|
|
||||||
"//conditions:default": [],
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
|
|
||||||
############################# Build configurations #############################
|
|
||||||
|
|
||||||
config_setting(
|
|
||||||
name = "optimized_build",
|
|
||||||
values = {
|
|
||||||
"compilation_mode": "opt",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
config_setting(
|
|
||||||
name = "linux_aarch64",
|
|
||||||
values = {"cpu": "aarch64"},
|
|
||||||
)
|
|
||||||
|
|
||||||
config_setting(
|
|
||||||
name = "android_armv7",
|
|
||||||
values = {
|
|
||||||
"crosstool_top": "//external:android/crosstool",
|
|
||||||
"cpu": "armeabi-v7a",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
|
|
||||||
config_setting(
|
|
||||||
name = "android_arm64",
|
|
||||||
values = {
|
|
||||||
"crosstool_top": "//external:android/crosstool",
|
|
||||||
"cpu": "arm64-v8a",
|
|
||||||
},
|
|
||||||
)
|
|
15
third_party/pthreadpool/workspace.bzl
vendored
15
third_party/pthreadpool/workspace.bzl
vendored
@ -1,15 +0,0 @@
|
|||||||
"""Loads the pthreadpool library, used by XNNPACK."""
|
|
||||||
|
|
||||||
load("//third_party:repo.bzl", "third_party_http_archive")
|
|
||||||
|
|
||||||
def repo():
|
|
||||||
third_party_http_archive(
|
|
||||||
name = "pthreadpool",
|
|
||||||
strip_prefix = "pthreadpool-76042155a8b1e189c8f141429fd72219472c32e1",
|
|
||||||
sha256 = "91c7b00c16c60c96f23d1966d524879c0f6044caf4bc5e9fc06518dda643e07e",
|
|
||||||
urls = [
|
|
||||||
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/Maratyszcza/pthreadpool/archive/76042155a8b1e189c8f141429fd72219472c32e1.tar.gz",
|
|
||||||
"https://github.com/Maratyszcza/pthreadpool/archive/76042155a8b1e189c8f141429fd72219472c32e1.tar.gz",
|
|
||||||
],
|
|
||||||
build_file = "//third_party/pthreadpool:BUILD.bazel",
|
|
||||||
)
|
|
Loading…
Reference in New Issue
Block a user