mirror of
https://github.com/google-ai-edge/LiteRT.git
synced 2026-01-09 06:21:22 +08:00
233 lines
6.6 KiB
Python
233 lines
6.6 KiB
Python
# buildifier: disable=load-on-top
|
|
|
|
workspace(name = "litert")
|
|
|
|
# buildifier: disable=load-on-top
|
|
|
|
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
|
|
|
|
http_archive(
|
|
name = "rules_shell",
|
|
sha256 = "bc61ef94facc78e20a645726f64756e5e285a045037c7a61f65af2941f4c25e1",
|
|
strip_prefix = "rules_shell-0.4.1",
|
|
url = "https://github.com/bazelbuild/rules_shell/releases/download/v0.4.1/rules_shell-v0.4.1.tar.gz",
|
|
)
|
|
|
|
load("@rules_shell//shell:repositories.bzl", "rules_shell_dependencies", "rules_shell_toolchains")
|
|
|
|
rules_shell_dependencies()
|
|
|
|
rules_shell_toolchains()
|
|
|
|
# Java rules
|
|
http_archive(
|
|
name = "rules_java",
|
|
sha256 = "c73336802d0b4882e40770666ad055212df4ea62cfa6edf9cb0f9d29828a0934",
|
|
url = "https://github.com/bazelbuild/rules_java/releases/download/5.3.5/rules_java-5.3.5.tar.gz",
|
|
)
|
|
|
|
# Load the custom repository rule to select either a local TensorFlow source or a remote http_archive.
|
|
load("//litert:tensorflow_source_rules.bzl", "tensorflow_source_repo")
|
|
|
|
tensorflow_source_repo(
|
|
name = "org_tensorflow",
|
|
sha256 = "",
|
|
strip_prefix = "tensorflow-master",
|
|
urls = [
|
|
"https://github.com/tensorflow/tensorflow/archive/master.tar.gz",
|
|
],
|
|
)
|
|
|
|
# Initialize the TensorFlow repository and all dependencies.
|
|
#
|
|
# The cascade of load() statements and tf_workspace?() calls works around the
|
|
# restriction that load() statements need to be at the top of .bzl files.
|
|
# E.g. we can not retrieve a new repository with http_archive and then load()
|
|
# a macro from that repository in the same file.
|
|
load("@org_tensorflow//tensorflow:workspace3.bzl", "tf_workspace3")
|
|
|
|
tf_workspace3()
|
|
|
|
# Initialize hermetic Python
|
|
load("@local_xla//third_party/py:python_init_rules.bzl", "python_init_rules")
|
|
|
|
python_init_rules()
|
|
|
|
load("@local_xla//third_party/py:python_init_repositories.bzl", "python_init_repositories")
|
|
|
|
python_init_repositories(
|
|
default_python_version = "system",
|
|
local_wheel_dist_folder = "dist",
|
|
local_wheel_inclusion_list = [
|
|
"tensorflow*",
|
|
"tf_nightly*",
|
|
],
|
|
local_wheel_workspaces = ["@org_tensorflow//:WORKSPACE"],
|
|
requirements = {
|
|
"3.9": "@org_tensorflow//:requirements_lock_3_9.txt",
|
|
"3.10": "@org_tensorflow//:requirements_lock_3_10.txt",
|
|
"3.11": "@org_tensorflow//:requirements_lock_3_11.txt",
|
|
"3.12": "@org_tensorflow//:requirements_lock_3_12.txt",
|
|
},
|
|
)
|
|
|
|
load("@local_xla//third_party/py:python_init_toolchains.bzl", "python_init_toolchains")
|
|
|
|
python_init_toolchains()
|
|
|
|
load("@local_xla//third_party/py:python_init_pip.bzl", "python_init_pip")
|
|
|
|
python_init_pip()
|
|
|
|
load("@pypi//:requirements.bzl", "install_deps")
|
|
|
|
install_deps()
|
|
# End hermetic Python initialization
|
|
|
|
load("@org_tensorflow//tensorflow:workspace2.bzl", "tf_workspace2")
|
|
|
|
tf_workspace2()
|
|
|
|
load("@org_tensorflow//tensorflow:workspace1.bzl", "tf_workspace1")
|
|
|
|
tf_workspace1()
|
|
|
|
load("@org_tensorflow//tensorflow:workspace0.bzl", "tf_workspace0")
|
|
|
|
tf_workspace0()
|
|
|
|
load(
|
|
"@local_xla//third_party/py:python_wheel.bzl",
|
|
"python_wheel_version_suffix_repository",
|
|
)
|
|
|
|
python_wheel_version_suffix_repository(name = "tf_wheel_version_suffix")
|
|
|
|
load(
|
|
"@local_xla//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl",
|
|
"cuda_json_init_repository",
|
|
)
|
|
|
|
cuda_json_init_repository()
|
|
|
|
load(
|
|
"@cuda_redist_json//:distributions.bzl",
|
|
"CUDA_REDISTRIBUTIONS",
|
|
"CUDNN_REDISTRIBUTIONS",
|
|
)
|
|
load(
|
|
"@local_xla//third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl",
|
|
"cuda_redist_init_repositories",
|
|
"cudnn_redist_init_repository",
|
|
)
|
|
|
|
cuda_redist_init_repositories(
|
|
cuda_redistributions = CUDA_REDISTRIBUTIONS,
|
|
)
|
|
|
|
cudnn_redist_init_repository(
|
|
cudnn_redistributions = CUDNN_REDISTRIBUTIONS,
|
|
)
|
|
|
|
load(
|
|
"@local_xla//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
|
|
"cuda_configure",
|
|
)
|
|
|
|
cuda_configure(name = "local_config_cuda")
|
|
|
|
load(
|
|
"@local_xla//third_party/nccl/hermetic:nccl_redist_init_repository.bzl",
|
|
"nccl_redist_init_repository",
|
|
)
|
|
|
|
nccl_redist_init_repository()
|
|
|
|
load(
|
|
"@local_xla//third_party/nccl/hermetic:nccl_configure.bzl",
|
|
"nccl_configure",
|
|
)
|
|
|
|
nccl_configure(name = "local_config_nccl")
|
|
|
|
load("//third_party/tqdm:workspace.bzl", tqdm = "repo")
|
|
|
|
tqdm()
|
|
|
|
load("//third_party/dawn:workspace.bzl", dawn = "repo")
|
|
|
|
dawn()
|
|
|
|
load("@rules_jvm_external//:defs.bzl", "maven_install")
|
|
|
|
maven_install(
|
|
name = "litert_maven",
|
|
artifacts = [
|
|
"androidx.lifecycle:lifecycle-common:2.8.7",
|
|
"com.google.android.play:ai-delivery:0.1.1-alpha01",
|
|
"com.google.guava:guava:33.4.6-android",
|
|
"org.jetbrains.kotlinx:kotlinx-coroutines-android:1.10.1",
|
|
"org.jetbrains.kotlinx:kotlinx-coroutines-guava:1.10.1",
|
|
"org.jetbrains.kotlinx:kotlinx-coroutines-play-services:1.10.1",
|
|
],
|
|
repositories = [
|
|
"https://jcenter.bintray.com",
|
|
"https://maven.google.com",
|
|
"https://dl.google.com/dl/android/maven2",
|
|
"https://repo1.maven.org/maven2",
|
|
],
|
|
version_conflict_policy = "pinned",
|
|
)
|
|
|
|
# Kotlin rules
|
|
http_archive(
|
|
name = "rules_kotlin",
|
|
sha256 = "e1448a56b2462407b2688dea86df5c375b36a0991bd478c2ddd94c97168125e2",
|
|
url = "https://github.com/bazelbuild/rules_kotlin/releases/download/v2.1.3/rules_kotlin-v2.1.3.tar.gz",
|
|
)
|
|
|
|
load("@rules_kotlin//kotlin:repositories.bzl", "kotlin_repositories")
|
|
|
|
kotlin_repositories() # if you want the default. Otherwise see custom kotlinc distribution below
|
|
|
|
load("@rules_kotlin//kotlin:core.bzl", "kt_register_toolchains")
|
|
|
|
kt_register_toolchains() # to use the default toolchain, otherwise see toolchains below
|
|
|
|
load("//third_party/stblib:workspace.bzl", stblib = "repo")
|
|
|
|
stblib()
|
|
|
|
# VENDOR SDKS ######################################################################################
|
|
|
|
# QUALCOMM ---------------------------------------------------------------------------------------
|
|
|
|
# The actual macro call will be set during configure for now.
|
|
load("//third_party/qairt:workspace.bzl", "qairt")
|
|
|
|
qairt()
|
|
|
|
# MEDIATEK ---------------------------------------------------------------------------------------
|
|
|
|
# Currently only works with local sdk
|
|
load("//third_party/neuro_pilot:workspace.bzl", "neuro_pilot")
|
|
|
|
neuro_pilot()
|
|
|
|
# GOOGLE TENSOR ----------------------------------------------------------------------------------
|
|
load("//third_party/google_tensor:workspace.bzl", "google_tensor")
|
|
|
|
google_tensor()
|
|
|
|
# LiteRT GPU ----------------------------------------------------------------------------------
|
|
load("//third_party/litert_gpu:workspace.bzl", "litert_gpu")
|
|
|
|
litert_gpu()
|
|
|
|
load("//third_party/intel_openvino:openvino.bzl", "openvino_configure")
|
|
|
|
openvino_configure(
|
|
name = "intel_openvino",
|
|
build_file = "//third_party/intel_openvino:openvino.bazel",
|
|
)
|