Skip to content

Commit

Permalink
[xla:cpu] Add an xnn_threadpool for wrapping ParallelLoopRunner as pt…
Browse files Browse the repository at this point in the history
…hreadpool API

Update xnnpack version to the latest one required by XLA:CPU

PiperOrigin-RevId: 706816929
  • Loading branch information
ezhulenev authored and copybara-github committed Dec 16, 2024
1 parent ec0078a commit 936ef9e
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions workspace2.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -111,9 +111,9 @@ def _tf_repositories():
# LINT.IfChange
tf_http_archive(
name = "XNNPACK",
sha256 = "ca3a5316b8161214f8f22a578fb638f1fccd0585eee40301363ffd026310379a",
strip_prefix = "XNNPACK-a50369c0fdd15f0f35b1a91c964644327a88d480",
urls = tf_mirror_urls("https://github.com/google/XNNPACK/archive/a50369c0fdd15f0f35b1a91c964644327a88d480.zip"),
sha256 = "3306f4178c8594b689165d385e644f03a3154c3be044f6ae36dd170fbf182cf5",
strip_prefix = "XNNPACK-983d013300f19fd3f4e33220b6401408e97a8d12",
urls = tf_mirror_urls("https://github.com/google/XNNPACK/archive/983d013300f19fd3f4e33220b6401408e97a8d12.zip"),
)
# LINT.ThenChange(//tensorflow/lite/tools/cmake/modules/xnnpack.cmake)

Expand All @@ -126,16 +126,16 @@ def _tf_repositories():

tf_http_archive(
name = "pthreadpool",
sha256 = "b96413b10dd8edaa4f6c0a60c6cf5ef55eebeef78164d5d69294c8173457f0ec",
strip_prefix = "pthreadpool-b8374f80e42010941bda6c85b0e3f1a1bd77a1e0",
urls = tf_mirror_urls("https://github.com/Maratyszcza/pthreadpool/archive/b8374f80e42010941bda6c85b0e3f1a1bd77a1e0.zip"),
sha256 = "a4cf06de57bfdf8d7b537c61f1c3071bce74e57524fe053e0bbd2332feca7f95",
strip_prefix = "pthreadpool-4fe0e1e183925bf8cfa6aae24237e724a96479b8",
urls = tf_mirror_urls("https://github.com/Maratyszcza/pthreadpool/archive/4fe0e1e183925bf8cfa6aae24237e724a96479b8.zip"),
)

tf_http_archive(
name = "cpuinfo",
strip_prefix = "cpuinfo-5e63739504f0f8e18e941bd63b2d6d42536c7d90",
sha256 = "18eca9bc8d9c4ce5496d0d2be9f456d55cbbb5f0639a551ce9c8bac2e84d85fe",
urls = tf_mirror_urls("https://github.com/pytorch/cpuinfo/archive/5e63739504f0f8e18e941bd63b2d6d42536c7d90.tar.gz"),
sha256 = "52e0ffd7998d8cb3a927d8a6e1145763744d866d2be09c4eccea27fc157b6bb0",
strip_prefix = "cpuinfo-cebb0933058d7f181c979afd50601dc311e1bf8c",
urls = tf_mirror_urls("https://github.com/pytorch/cpuinfo/archive/cebb0933058d7f181c979afd50601dc311e1bf8c.zip"),
)

tf_http_archive(
Expand Down

0 comments on commit 936ef9e

Please sign in to comment.