mirror of
https://github.com/tensorflow/tensorflow.git
synced 2024-11-21 21:05:19 +00:00
f366199c00
All custom types shared between TF/TSL/JAX are upstreamed to the ml_dtypes package. This change adds that dependency. A follow-up will replace TF's existing definitions with the upstreamed ones. Moved the `pybind11_bazel` dependency since it is used both by `ml_dtypes` and `pybind11_abseil`. PiperOrigin-RevId: 539157226
18 lines
717 B
Python
18 lines
717 B
Python
"""Provides the repo macro to import pybind11_bazel.
|
|
|
|
pybind11_bazel requires pybind11 (which is loaded in another rule).
|
|
"""
|
|
|
|
load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
|
|
|
|
def repo():
|
|
PB_COMMIT = "72cbbf1fbc830e487e3012862b7b720001b70672"
|
|
PB_SHA256 = "516c1b3a10d87740d2b7de6f121f8e19dde2c372ecbfe59aef44cd1872c10395"
|
|
tf_http_archive(
|
|
name = "pybind11_bazel",
|
|
strip_prefix = "pybind11_bazel-{commit}".format(commit = PB_COMMIT),
|
|
sha256 = PB_SHA256,
|
|
patch_file = ["//third_party/pybind11_bazel:pybind11_bazel.patch"],
|
|
urls = tf_mirror_urls("https://github.com/pybind/pybind11_bazel/archive/{commit}.tar.gz".format(commit = PB_COMMIT)),
|
|
)
|