No public description

PiperOrigin-RevId: 591370734
This commit is contained in:
Sebastian Schmidt 2023-12-15 15:08:19 -08:00 committed by Copybara-Service
parent 41db137d37
commit 4b471266b8
2 changed files with 40 additions and 50 deletions

View File

@ -1,7 +1,6 @@
workspace(name = "mediapipe") workspace(name = "mediapipe")
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive") load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
load("@//third_party:shared_dependencies.bzl", "mediapipe_absl", "mediapipe_sentencepiece", "mediapipe_flatbuffers", "mediapipe_tensorflow")
# Protobuf expects an //external:python_headers target # Protobuf expects an //external:python_headers target
bind( bind(
@ -22,6 +21,7 @@ bazel_skylib_workspace()
load("@bazel_skylib//lib:versions.bzl", "versions") load("@bazel_skylib//lib:versions.bzl", "versions")
versions.check(minimum_bazel_version = "3.7.2") versions.check(minimum_bazel_version = "3.7.2")
load("@//third_party:shared_dependencies.bzl", "mediapipe_absl", "mediapipe_sentencepiece")
mediapipe_absl() mediapipe_absl()
mediapipe_sentencepiece() mediapipe_sentencepiece()
@ -208,7 +208,8 @@ http_archive(
urls = ["https://github.com/protocolbuffers/protobuf/archive/v3.19.1.tar.gz"], urls = ["https://github.com/protocolbuffers/protobuf/archive/v3.19.1.tar.gz"],
) )
mediapipe_flatbuffers() load("@//third_party/flatbuffers:workspace.bzl", flatbuffers = "repo")
flatbuffers()
http_archive( http_archive(
name = "com_google_audio_tools", name = "com_google_audio_tools",
@ -228,6 +229,7 @@ http_archive(
build_file = "@//third_party:pffft.BUILD", build_file = "@//third_party:pffft.BUILD",
) )
http_archive( http_archive(
name = "darts_clone", name = "darts_clone",
build_file = "@//third_party:darts_clone.BUILD", build_file = "@//third_party:darts_clone.BUILD",
@ -461,7 +463,42 @@ maven_install(
version_conflict_policy = "pinned", version_conflict_policy = "pinned",
) )
mediapipe_tensorflow() # Needed by TensorFlow
http_archive(
name = "io_bazel_rules_closure",
sha256 = "e0a111000aeed2051f29fcc7a3f83be3ad8c6c93c186e64beb1ad313f0c7f9f9",
strip_prefix = "rules_closure-cf1e44edb908e9616030cc83d085989b8e6cd6df",
urls = [
"http://mirror.tensorflow.org/github.com/bazelbuild/rules_closure/archive/cf1e44edb908e9616030cc83d085989b8e6cd6df.tar.gz",
"https://github.com/bazelbuild/rules_closure/archive/cf1e44edb908e9616030cc83d085989b8e6cd6df.tar.gz", # 2019-04-04
],
)
# TensorFlow repo should always go after the other external dependencies.
# TF on 2023-07-26.
_TENSORFLOW_GIT_COMMIT = "e92261fd4cec0b726692081c4d2966b75abf31dd"
# curl -L https://github.com/tensorflow/tensorflow/archive/<TENSORFLOW_GIT_COMMIT>.tar.gz | shasum -a 256
_TENSORFLOW_SHA256 = "478a229bd4ec70a5b568ac23b5ea013d9fca46a47d6c43e30365a0412b9febf4"
http_archive(
name = "org_tensorflow",
urls = [
"https://github.com/tensorflow/tensorflow/archive/%s.tar.gz" % _TENSORFLOW_GIT_COMMIT,
],
patches = [
"@//third_party:org_tensorflow_compatibility_fixes.diff",
"@//third_party:org_tensorflow_system_python.diff",
# Diff is generated with a script, don't update it manually.
"@//third_party:org_tensorflow_custom_ops.diff",
# Works around Bazel issue with objc_library.
# See https://github.com/bazelbuild/bazel/issues/19912
"@//third_party:org_tensorflow_objc_build_fixes.diff",
],
patch_args = [
"-p1",
],
strip_prefix = "tensorflow-%s" % _TENSORFLOW_GIT_COMMIT,
sha256 = _TENSORFLOW_SHA256,
)
load("@org_tensorflow//tensorflow:workspace3.bzl", "tf_workspace3") load("@org_tensorflow//tensorflow:workspace3.bzl", "tf_workspace3")
tf_workspace3() tf_workspace3()

View File

@ -1,18 +1,9 @@
"""MediaPipe's shared dependencies that can be used by dependent projects. Includes build patches.""" """MediaPipe's shared dependencies that can be used by dependent projects. Includes build patches."""
load("@//third_party/flatbuffers:workspace.bzl", flatbuffers = "repo")
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive") load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
# TensorFlow repo should always go after the other external dependencies.
# TF on 2023-07-26.
_TENSORFLOW_GIT_COMMIT = "e92261fd4cec0b726692081c4d2966b75abf31dd"
# curl -L https://github.com/tensorflow/tensorflow/archive/<TENSORFLOW_GIT_COMMIT>.tar.gz | shasum -a 256
_TENSORFLOW_SHA256 = "478a229bd4ec70a5b568ac23b5ea013d9fca46a47d6c43e30365a0412b9febf4"
# ABSL cpp library lts_2023_01_25. # ABSL cpp library lts_2023_01_25.
def mediapipe_absl(): def mediapipe_absl():
"""Exports the ABSL depedency on TensorFlow."""
http_archive( http_archive(
name = "com_google_absl", name = "com_google_absl",
urls = [ urls = [
@ -29,7 +20,6 @@ def mediapipe_absl():
) )
def mediapipe_sentencepiece(): def mediapipe_sentencepiece():
"""Exports the Semtencepiece depedency on TensorFlow."""
http_archive( http_archive(
name = "com_google_sentencepiece", name = "com_google_sentencepiece",
strip_prefix = "sentencepiece-0.1.96", strip_prefix = "sentencepiece-0.1.96",
@ -41,40 +31,3 @@ def mediapipe_sentencepiece():
patches = ["@//third_party:com_google_sentencepiece.diff"], patches = ["@//third_party:com_google_sentencepiece.diff"],
patch_args = ["-p1"], patch_args = ["-p1"],
) )
def mediapipe_flatbuffers():
"""Exports the FlatBuffers depedency on TensorFlow."""
flatbuffers()
def mediapipe_tensorflow():
"""Exports the MediaPipe depedency on TensorFlow."""
# Needed by TensorFlow
http_archive(
name = "io_bazel_rules_closure",
sha256 = "e0a111000aeed2051f29fcc7a3f83be3ad8c6c93c186e64beb1ad313f0c7f9f9",
strip_prefix = "rules_closure-cf1e44edb908e9616030cc83d085989b8e6cd6df",
urls = [
"http://mirror.tensorflow.org/github.com/bazelbuild/rules_closure/archive/cf1e44edb908e9616030cc83d085989b8e6cd6df.tar.gz",
"https://github.com/bazelbuild/rules_closure/archive/cf1e44edb908e9616030cc83d085989b8e6cd6df.tar.gz", # 2019-04-04
],
)
http_archive(
name = "org_tensorflow",
urls = ["https://github.com/tensorflow/tensorflow/archive/%s.tar.gz" % _TENSORFLOW_GIT_COMMIT],
patches = [
"@//third_party:org_tensorflow_compatibility_fixes.diff",
"@//third_party:org_tensorflow_system_python.diff",
# Diff is generated with a script, don't update it manually.
"@//third_party:org_tensorflow_custom_ops.diff",
# Works around Bazel issue with objc_library.
# See https://github.com/bazelbuild/bazel/issues/19912
"@//third_party:org_tensorflow_objc_build_fixes.diff",
],
patch_args = [
"-p1",
],
strip_prefix = "tensorflow-%s" % _TENSORFLOW_GIT_COMMIT,
sha256 = _TENSORFLOW_SHA256,
)