mirror of
https://github.com/zebrajr/tensorflow.git
synced 2026-01-15 12:15:41 +00:00
3eae65361b
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros. The JAX repository update will be done [in the next step](https://github.com/jax-ml/jax/pull/31709) PiperOrigin-RevId: 808689003
74 lines
1.9 KiB
Python
74 lines
1.9 KiB
Python
load("@bazel_skylib//rules:common_settings.bzl", "bool_flag")
|
|
load("@python_version_repo//:py_version.bzl", "REQUIREMENTS")
|
|
load("@rules_python//python:pip.bzl", "compile_pip_requirements")
|
|
|
|
compile_pip_requirements(
|
|
name = "requirements",
|
|
extra_args = [
|
|
"--allow-unsafe",
|
|
"--build-isolation",
|
|
],
|
|
generate_hashes = True,
|
|
requirements_in = "requirements.in",
|
|
requirements_txt = REQUIREMENTS,
|
|
)
|
|
|
|
compile_pip_requirements(
|
|
name = "requirements_nightly",
|
|
data = ["test-requirements.txt"],
|
|
extra_args = [
|
|
"--allow-unsafe",
|
|
"--build-isolation",
|
|
"--extra-index-url=https://pypi.anaconda.org/scientific-python-nightly-wheels/simple",
|
|
"--pre",
|
|
"--upgrade",
|
|
],
|
|
generate_hashes = False,
|
|
requirements_in = "requirements.in",
|
|
requirements_txt = REQUIREMENTS,
|
|
)
|
|
|
|
compile_pip_requirements(
|
|
name = "requirements_dev",
|
|
extra_args = [
|
|
"--allow-unsafe",
|
|
"--build-isolation",
|
|
"--upgrade",
|
|
],
|
|
generate_hashes = False,
|
|
requirements_in = "requirements.in",
|
|
requirements_txt = REQUIREMENTS,
|
|
)
|
|
|
|
# Flag indicating if the target requires pre-built wheel.
|
|
bool_flag(
|
|
name = "wheel_dependency",
|
|
build_setting_default = False,
|
|
)
|
|
|
|
config_setting(
|
|
name = "enable_wheel_dependency",
|
|
flag_values = {
|
|
":wheel_dependency": "True",
|
|
},
|
|
)
|
|
|
|
filegroup(
|
|
name = "manylinux_compliance_test",
|
|
srcs = ["manylinux_compliance_test.py"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
py_binary(
|
|
name = "unpack_wheel_and_unzip_archive_files",
|
|
srcs = ["unpack_wheel_and_unzip_archive_files.py"],
|
|
main = "unpack_wheel_and_unzip_archive_files.py",
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
py_library(
|
|
name = "setup_py_nvidia_dependencies_util",
|
|
srcs = ["setup_py_nvidia_dependencies_util.py"],
|
|
visibility = ["//visibility:public"],
|
|
)
|