Skip to content

Commit 936ef9e

Browse files
ezhulenevcopybara-github
authored andcommitted
[xla:cpu] Add an xnn_threadpool for wrapping ParallelLoopRunner as pthreadpool API
Update xnnpack version to the latest one required by XLA:CPU PiperOrigin-RevId: 706816929
1 parent ec0078a commit 936ef9e

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

workspace2.bzl

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -111,9 +111,9 @@ def _tf_repositories():
111111
# LINT.IfChange
112112
tf_http_archive(
113113
name = "XNNPACK",
114-
sha256 = "ca3a5316b8161214f8f22a578fb638f1fccd0585eee40301363ffd026310379a",
115-
strip_prefix = "XNNPACK-a50369c0fdd15f0f35b1a91c964644327a88d480",
116-
urls = tf_mirror_urls("https://github.com/google/XNNPACK/archive/a50369c0fdd15f0f35b1a91c964644327a88d480.zip"),
114+
sha256 = "3306f4178c8594b689165d385e644f03a3154c3be044f6ae36dd170fbf182cf5",
115+
strip_prefix = "XNNPACK-983d013300f19fd3f4e33220b6401408e97a8d12",
116+
urls = tf_mirror_urls("https://github.com/google/XNNPACK/archive/983d013300f19fd3f4e33220b6401408e97a8d12.zip"),
117117
)
118118
# LINT.ThenChange(//tensorflow/lite/tools/cmake/modules/xnnpack.cmake)
119119

@@ -126,16 +126,16 @@ def _tf_repositories():
126126

127127
tf_http_archive(
128128
name = "pthreadpool",
129-
sha256 = "b96413b10dd8edaa4f6c0a60c6cf5ef55eebeef78164d5d69294c8173457f0ec",
130-
strip_prefix = "pthreadpool-b8374f80e42010941bda6c85b0e3f1a1bd77a1e0",
131-
urls = tf_mirror_urls("https://github.com/Maratyszcza/pthreadpool/archive/b8374f80e42010941bda6c85b0e3f1a1bd77a1e0.zip"),
129+
sha256 = "a4cf06de57bfdf8d7b537c61f1c3071bce74e57524fe053e0bbd2332feca7f95",
130+
strip_prefix = "pthreadpool-4fe0e1e183925bf8cfa6aae24237e724a96479b8",
131+
urls = tf_mirror_urls("https://github.com/Maratyszcza/pthreadpool/archive/4fe0e1e183925bf8cfa6aae24237e724a96479b8.zip"),
132132
)
133133

134134
tf_http_archive(
135135
name = "cpuinfo",
136-
strip_prefix = "cpuinfo-5e63739504f0f8e18e941bd63b2d6d42536c7d90",
137-
sha256 = "18eca9bc8d9c4ce5496d0d2be9f456d55cbbb5f0639a551ce9c8bac2e84d85fe",
138-
urls = tf_mirror_urls("https://github.com/pytorch/cpuinfo/archive/5e63739504f0f8e18e941bd63b2d6d42536c7d90.tar.gz"),
136+
sha256 = "52e0ffd7998d8cb3a927d8a6e1145763744d866d2be09c4eccea27fc157b6bb0",
137+
strip_prefix = "cpuinfo-cebb0933058d7f181c979afd50601dc311e1bf8c",
138+
urls = tf_mirror_urls("https://github.com/pytorch/cpuinfo/archive/cebb0933058d7f181c979afd50601dc311e1bf8c.zip"),
139139
)
140140

141141
tf_http_archive(

0 commit comments

Comments
 (0)