OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Update py_import macros for the ability to unpack additional wheels in the same folder as the main wheel.

Browse Source 
Usage example: provide NVIDIA wheel dependencies for ML wheels that have rpaths pointing to NVIDIA folders. When a user executes `pip install tensorflow[and_cuda]`, NVIDIA wheels are installed together with Tensorflow wheel. To reproduce this behavior in hermetic Python approach, we need to define `py_import` as follows (provided NVIDIA dependencies are defined in `requirements.in` and requirements lock files):

        py_import(
            name = "tf_py_import",
            wheel = ":wheel",
            deps = [
                "@pypi_absl_py//:pkg",
                "@pypi_astunparse//:pkg",
                "@pypi_flatbuffers//:pkg",
                "@pypi_gast//:pkg",
                "@pypi_ml_dtypes//:pkg",
                "@pypi_numpy//:pkg",
                "@pypi_opt_einsum//:pkg",
                "@pypi_packaging//:pkg",
                "@pypi_protobuf//:pkg",
                "@pypi_requests//:pkg",
                "@pypi_termcolor//:pkg",
                "@pypi_typing_extensions//:pkg",
                "@pypi_wrapt//:pkg",
            ],
            wheel_deps = [
                "@pypi_nvidia_cublas_cu12//:whl",
                "@pypi_nvidia_cuda_cupti_cu12//:whl",
                "@pypi_nvidia_cuda_nvcc_cu12//:whl",
                "@pypi_nvidia_cuda_nvrtc_cu12//:whl",
                "@pypi_nvidia_cuda_runtime_cu12//:whl",
                "@pypi_nvidia_cudnn_cu12//:whl",
                "@pypi_nvidia_cufft_cu12//:whl",
                "@pypi_nvidia_curand_cu12//:whl",
                "@pypi_nvidia_cusolver_cu12//:whl",
                "@pypi_nvidia_cusparse_cu12//:whl",
                "@pypi_nvidia_nccl_cu12//:whl",
                "@pypi_nvidia_nvjitlink_cu12//:whl",
            ],
        )

PiperOrigin-RevId: 705666137
This commit is contained in:
A. Unique TensorFlower 2024-12-12 16:24:18 -08:00 committed by TensorFlower Gardener
parent 3b501c046a
commit d24c61b7bd
36 changed files with 702 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ci/official/requirements_updater/numpy1_requirements/requirements.in
View File

@ -28,6 +28,20 @@ requests >= 2.31.0
packaging==23.2
setuptools==70.0.0
jax==0.4.7
# NVIDIA CUDA dependencies
# Note that the wheels are downloaded only when the targets in bazel command
# contain dependencies on these wheels.
nvidia-cublas-cu12 == 12.5.3.2
nvidia-cuda-cupti-cu12 == 12.5.82
nvidia-cuda-nvrtc-cu12 == 12.5.82
nvidia-cuda-runtime-cu12 == 12.5.82
nvidia-cudnn-cu12 == 9.3.0.75
nvidia-cufft-cu12 == 11.2.3.61
nvidia-curand-cu12 == 10.3.6.82
nvidia-cusolver-cu12 == 11.6.3.83
nvidia-cusparse-cu12 == 12.5.1.3
nvidia-nccl-cu12 == 2.23.4
nvidia-nvjitlink-cu12 == 12.5.82
# The dependencies below are needed for TF wheel testing.
tensorflow-io-gcs-filesystem==0.37.1
libclang >= 13.0.0

63
ci/official/requirements_updater/numpy1_requirements/requirements_lock_3_10.txt
View File

@ -430,6 +430,69 @@ numpy==1.26.4 \
# opt-einsum
# scipy
# tb-nightly
nvidia-cublas-cu12==12.5.3.2 \
--hash=sha256:4960f3dc5f39699acadf76fa6d94b10a2a00f2956c2c442efa299fb22b0748f3 \
--hash=sha256:7d0191251180de606023d396b94d66f66470a0ae96d1dbb906c7656ea0f71eda \
--hash=sha256:ca070ad70e9fa6654084575d01bd001f30cc4665e33d4bb9fc8e0f321caa034b
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.5.82 \
--hash=sha256:4f835281cf492e2bedd153f5c3de9da8f1d775a419468305e64ce73b3b0c6dc3 \
--hash=sha256:bde77a5feb66752ec61db2adfe47f56b941842825b4c7e2068aff27c9d107953 \
--hash=sha256:d32c06490c6ba35c4323730820c7d0c4c126c04ed58d2f57275adb8d54b138fe
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-nvrtc-cu12==12.5.82 \
--hash=sha256:3dbd97b0104b4bfbc3c4f8c79cd2496307c89c43c29a9f83125f1d76296ff3fd \
--hash=sha256:5bb6a0eb01d4974bb7ca3d48bd3859472debb3c3057a5e7de2b08fbdf35eed7e \
--hash=sha256:e5db37e990056c70953b7772dd778336ef9da0a0b5bb28f9f2a61c2e42b51d78
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-runtime-cu12==12.5.82 \
--hash=sha256:0fd5fbca289bceb9f0690aa9858f06187b554fdeb7e2711dfd5bb3ce58900b46 \
--hash=sha256:3e79a060e126df40fd3a068f3f787eb000fa51b251ec6cd97d09579632687115 \
--hash=sha256:71f015dbf9df05dd71f7480132c6ebf47a6ceb2ab53d7db8e08e4b30ebb87e14
# via -r ci/official/requirements_updater/requirements.in
nvidia-cudnn-cu12==9.3.0.75 \
--hash=sha256:9ad9c6929ebb5295eb4a1728024666d1c88283373e265a0c5c883e6f9d5cd76d \
--hash=sha256:c5cf7ff3415e446adf195a5b7dd2ba56cd00c3ee78bfdc566e51698931aa4b7f \
--hash=sha256:c819e82eed8cf564b9d37478ea4eab9e87194bb3b7f7f8098bc1f67c9b80f1b6
# via -r ci/official/requirements_updater/requirements.in
nvidia-cufft-cu12==11.2.3.61 \
--hash=sha256:4a8f6f0ce93c52a50ee83422a80472b5f376054a63f38532d0eab4007e7ef28b \
--hash=sha256:6d45b48a5ee7599e57131129cda2c58544d9b78b95064d3ec3e5c6b96e2b58cc \
--hash=sha256:9a6e8df162585750f61983a638104a48c756aa13f9f48e19ab079b38e3c828b8
# via -r ci/official/requirements_updater/requirements.in
nvidia-curand-cu12==10.3.6.82 \
--hash=sha256:0631ba65231260ad832ce233ddda57e7b3b7158eabf000d78e46cbb5bd5b7aae \
--hash=sha256:2823fb27de4e44dbb22394a6adf53aa6e1b013aca0f8c22867d1cfae58405536 \
--hash=sha256:36aabeb5990297bbce3df324ea7c7c13c3aabb140c86d50ab3b23e4ec61672f1
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusolver-cu12==11.6.3.83 \
--hash=sha256:1b8b77d2fe8abe72bb722dafb708cceaeb81f1a03999477f20b33b34f46ab885 \
--hash=sha256:6224732963cba312a84c78114b9a38c4ffabb2e2a6a120923ac99ba6f895c8cf \
--hash=sha256:93cfafacde4428b71778eeb092ec615a02a3d05404da1bcf91c53e3fa1bce42b
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusparse-cu12==12.5.1.3 \
--hash=sha256:016df8e993c437e8301e62739f01775cba988fd5253cd4c64173f8e8d2f8e752 \
--hash=sha256:33520db374e2f5ebc976d6faa1852b98c398a57e6f71150fe59705928596ffd1 \
--hash=sha256:7b97fd01f0a61628af99d0efd52132fccc8c18fc5c509f13802dccf0574a19c2
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.23.4 \
--hash=sha256:aa946c8327e22ced28e7cef508a334673abc42064ec85f02d005ba1785ea4cec \
--hash=sha256:b097258d9aab2fa9f686e33c6fe40ae57b27df60cedbd15d139701bb5509e0c1
# via -r ci/official/requirements_updater/requirements.in
nvidia-nvjitlink-cu12==12.5.82 \
--hash=sha256:98103729cc5226e13ca319a10bbf9433bbbd44ef64fe72f45f067cacc14b8d27 \
--hash=sha256:e782564d705ff0bf61ac3e1bf730166da66dd2fe9012f111ede5fc49b64ae697 \
--hash=sha256:f9b37bc5c8cf7509665cb6ada5aaa0ce65618f2332b7d3e78e9790511f111212
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
opt-einsum==3.3.0 \
--hash=sha256:2455e59e3947d3c275477df7f5205b30635e266fe6dc300e3d9f9646bfcea147 \
--hash=sha256:59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549

63
ci/official/requirements_updater/numpy1_requirements/requirements_lock_3_11.txt
View File

@ -430,6 +430,69 @@ numpy==1.26.4 \
# opt-einsum
# scipy
# tb-nightly
nvidia-cublas-cu12==12.5.3.2 \
--hash=sha256:4960f3dc5f39699acadf76fa6d94b10a2a00f2956c2c442efa299fb22b0748f3 \
--hash=sha256:7d0191251180de606023d396b94d66f66470a0ae96d1dbb906c7656ea0f71eda \
--hash=sha256:ca070ad70e9fa6654084575d01bd001f30cc4665e33d4bb9fc8e0f321caa034b
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.5.82 \
--hash=sha256:4f835281cf492e2bedd153f5c3de9da8f1d775a419468305e64ce73b3b0c6dc3 \
--hash=sha256:bde77a5feb66752ec61db2adfe47f56b941842825b4c7e2068aff27c9d107953 \
--hash=sha256:d32c06490c6ba35c4323730820c7d0c4c126c04ed58d2f57275adb8d54b138fe
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-nvrtc-cu12==12.5.82 \
--hash=sha256:3dbd97b0104b4bfbc3c4f8c79cd2496307c89c43c29a9f83125f1d76296ff3fd \
--hash=sha256:5bb6a0eb01d4974bb7ca3d48bd3859472debb3c3057a5e7de2b08fbdf35eed7e \
--hash=sha256:e5db37e990056c70953b7772dd778336ef9da0a0b5bb28f9f2a61c2e42b51d78
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-runtime-cu12==12.5.82 \
--hash=sha256:0fd5fbca289bceb9f0690aa9858f06187b554fdeb7e2711dfd5bb3ce58900b46 \
--hash=sha256:3e79a060e126df40fd3a068f3f787eb000fa51b251ec6cd97d09579632687115 \
--hash=sha256:71f015dbf9df05dd71f7480132c6ebf47a6ceb2ab53d7db8e08e4b30ebb87e14
# via -r ci/official/requirements_updater/requirements.in
nvidia-cudnn-cu12==9.3.0.75 \
--hash=sha256:9ad9c6929ebb5295eb4a1728024666d1c88283373e265a0c5c883e6f9d5cd76d \
--hash=sha256:c5cf7ff3415e446adf195a5b7dd2ba56cd00c3ee78bfdc566e51698931aa4b7f \
--hash=sha256:c819e82eed8cf564b9d37478ea4eab9e87194bb3b7f7f8098bc1f67c9b80f1b6
# via -r ci/official/requirements_updater/requirements.in
nvidia-cufft-cu12==11.2.3.61 \
--hash=sha256:4a8f6f0ce93c52a50ee83422a80472b5f376054a63f38532d0eab4007e7ef28b \
--hash=sha256:6d45b48a5ee7599e57131129cda2c58544d9b78b95064d3ec3e5c6b96e2b58cc \
--hash=sha256:9a6e8df162585750f61983a638104a48c756aa13f9f48e19ab079b38e3c828b8
# via -r ci/official/requirements_updater/requirements.in
nvidia-curand-cu12==10.3.6.82 \
--hash=sha256:0631ba65231260ad832ce233ddda57e7b3b7158eabf000d78e46cbb5bd5b7aae \
--hash=sha256:2823fb27de4e44dbb22394a6adf53aa6e1b013aca0f8c22867d1cfae58405536 \
--hash=sha256:36aabeb5990297bbce3df324ea7c7c13c3aabb140c86d50ab3b23e4ec61672f1
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusolver-cu12==11.6.3.83 \
--hash=sha256:1b8b77d2fe8abe72bb722dafb708cceaeb81f1a03999477f20b33b34f46ab885 \
--hash=sha256:6224732963cba312a84c78114b9a38c4ffabb2e2a6a120923ac99ba6f895c8cf \
--hash=sha256:93cfafacde4428b71778eeb092ec615a02a3d05404da1bcf91c53e3fa1bce42b
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusparse-cu12==12.5.1.3 \
--hash=sha256:016df8e993c437e8301e62739f01775cba988fd5253cd4c64173f8e8d2f8e752 \
--hash=sha256:33520db374e2f5ebc976d6faa1852b98c398a57e6f71150fe59705928596ffd1 \
--hash=sha256:7b97fd01f0a61628af99d0efd52132fccc8c18fc5c509f13802dccf0574a19c2
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.23.4 \
--hash=sha256:aa946c8327e22ced28e7cef508a334673abc42064ec85f02d005ba1785ea4cec \
--hash=sha256:b097258d9aab2fa9f686e33c6fe40ae57b27df60cedbd15d139701bb5509e0c1
# via -r ci/official/requirements_updater/requirements.in
nvidia-nvjitlink-cu12==12.5.82 \
--hash=sha256:98103729cc5226e13ca319a10bbf9433bbbd44ef64fe72f45f067cacc14b8d27 \
--hash=sha256:e782564d705ff0bf61ac3e1bf730166da66dd2fe9012f111ede5fc49b64ae697 \
--hash=sha256:f9b37bc5c8cf7509665cb6ada5aaa0ce65618f2332b7d3e78e9790511f111212
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
opt-einsum==3.3.0 \
--hash=sha256:2455e59e3947d3c275477df7f5205b30635e266fe6dc300e3d9f9646bfcea147 \
--hash=sha256:59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549

63
ci/official/requirements_updater/numpy1_requirements/requirements_lock_3_12.txt
View File

@ -430,6 +430,69 @@ numpy==1.26.4 \
# opt-einsum
# scipy
# tb-nightly
nvidia-cublas-cu12==12.5.3.2 \
--hash=sha256:4960f3dc5f39699acadf76fa6d94b10a2a00f2956c2c442efa299fb22b0748f3 \
--hash=sha256:7d0191251180de606023d396b94d66f66470a0ae96d1dbb906c7656ea0f71eda \
--hash=sha256:ca070ad70e9fa6654084575d01bd001f30cc4665e33d4bb9fc8e0f321caa034b
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.5.82 \
--hash=sha256:4f835281cf492e2bedd153f5c3de9da8f1d775a419468305e64ce73b3b0c6dc3 \
--hash=sha256:bde77a5feb66752ec61db2adfe47f56b941842825b4c7e2068aff27c9d107953 \
--hash=sha256:d32c06490c6ba35c4323730820c7d0c4c126c04ed58d2f57275adb8d54b138fe
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-nvrtc-cu12==12.5.82 \
--hash=sha256:3dbd97b0104b4bfbc3c4f8c79cd2496307c89c43c29a9f83125f1d76296ff3fd \
--hash=sha256:5bb6a0eb01d4974bb7ca3d48bd3859472debb3c3057a5e7de2b08fbdf35eed7e \
--hash=sha256:e5db37e990056c70953b7772dd778336ef9da0a0b5bb28f9f2a61c2e42b51d78
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-runtime-cu12==12.5.82 \
--hash=sha256:0fd5fbca289bceb9f0690aa9858f06187b554fdeb7e2711dfd5bb3ce58900b46 \
--hash=sha256:3e79a060e126df40fd3a068f3f787eb000fa51b251ec6cd97d09579632687115 \
--hash=sha256:71f015dbf9df05dd71f7480132c6ebf47a6ceb2ab53d7db8e08e4b30ebb87e14
# via -r ci/official/requirements_updater/requirements.in
nvidia-cudnn-cu12==9.3.0.75 \
--hash=sha256:9ad9c6929ebb5295eb4a1728024666d1c88283373e265a0c5c883e6f9d5cd76d \
--hash=sha256:c5cf7ff3415e446adf195a5b7dd2ba56cd00c3ee78bfdc566e51698931aa4b7f \
--hash=sha256:c819e82eed8cf564b9d37478ea4eab9e87194bb3b7f7f8098bc1f67c9b80f1b6
# via -r ci/official/requirements_updater/requirements.in
nvidia-cufft-cu12==11.2.3.61 \
--hash=sha256:4a8f6f0ce93c52a50ee83422a80472b5f376054a63f38532d0eab4007e7ef28b \
--hash=sha256:6d45b48a5ee7599e57131129cda2c58544d9b78b95064d3ec3e5c6b96e2b58cc \
--hash=sha256:9a6e8df162585750f61983a638104a48c756aa13f9f48e19ab079b38e3c828b8
# via -r ci/official/requirements_updater/requirements.in
nvidia-curand-cu12==10.3.6.82 \
--hash=sha256:0631ba65231260ad832ce233ddda57e7b3b7158eabf000d78e46cbb5bd5b7aae \
--hash=sha256:2823fb27de4e44dbb22394a6adf53aa6e1b013aca0f8c22867d1cfae58405536 \
--hash=sha256:36aabeb5990297bbce3df324ea7c7c13c3aabb140c86d50ab3b23e4ec61672f1
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusolver-cu12==11.6.3.83 \
--hash=sha256:1b8b77d2fe8abe72bb722dafb708cceaeb81f1a03999477f20b33b34f46ab885 \
--hash=sha256:6224732963cba312a84c78114b9a38c4ffabb2e2a6a120923ac99ba6f895c8cf \
--hash=sha256:93cfafacde4428b71778eeb092ec615a02a3d05404da1bcf91c53e3fa1bce42b
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusparse-cu12==12.5.1.3 \
--hash=sha256:016df8e993c437e8301e62739f01775cba988fd5253cd4c64173f8e8d2f8e752 \
--hash=sha256:33520db374e2f5ebc976d6faa1852b98c398a57e6f71150fe59705928596ffd1 \
--hash=sha256:7b97fd01f0a61628af99d0efd52132fccc8c18fc5c509f13802dccf0574a19c2
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.23.4 \
--hash=sha256:aa946c8327e22ced28e7cef508a334673abc42064ec85f02d005ba1785ea4cec \
--hash=sha256:b097258d9aab2fa9f686e33c6fe40ae57b27df60cedbd15d139701bb5509e0c1
# via -r ci/official/requirements_updater/requirements.in
nvidia-nvjitlink-cu12==12.5.82 \
--hash=sha256:98103729cc5226e13ca319a10bbf9433bbbd44ef64fe72f45f067cacc14b8d27 \
--hash=sha256:e782564d705ff0bf61ac3e1bf730166da66dd2fe9012f111ede5fc49b64ae697 \
--hash=sha256:f9b37bc5c8cf7509665cb6ada5aaa0ce65618f2332b7d3e78e9790511f111212
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
opt-einsum==3.3.0 \
--hash=sha256:2455e59e3947d3c275477df7f5205b30635e266fe6dc300e3d9f9646bfcea147 \
--hash=sha256:59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549

63
ci/official/requirements_updater/numpy1_requirements/requirements_lock_3_9.txt
View File

@ -434,6 +434,69 @@ numpy==1.26.4 \
# opt-einsum
# scipy
# tb-nightly
nvidia-cublas-cu12==12.5.3.2 \
--hash=sha256:4960f3dc5f39699acadf76fa6d94b10a2a00f2956c2c442efa299fb22b0748f3 \
--hash=sha256:7d0191251180de606023d396b94d66f66470a0ae96d1dbb906c7656ea0f71eda \
--hash=sha256:ca070ad70e9fa6654084575d01bd001f30cc4665e33d4bb9fc8e0f321caa034b
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.5.82 \
--hash=sha256:4f835281cf492e2bedd153f5c3de9da8f1d775a419468305e64ce73b3b0c6dc3 \
--hash=sha256:bde77a5feb66752ec61db2adfe47f56b941842825b4c7e2068aff27c9d107953 \
--hash=sha256:d32c06490c6ba35c4323730820c7d0c4c126c04ed58d2f57275adb8d54b138fe
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-nvrtc-cu12==12.5.82 \
--hash=sha256:3dbd97b0104b4bfbc3c4f8c79cd2496307c89c43c29a9f83125f1d76296ff3fd \
--hash=sha256:5bb6a0eb01d4974bb7ca3d48bd3859472debb3c3057a5e7de2b08fbdf35eed7e \
--hash=sha256:e5db37e990056c70953b7772dd778336ef9da0a0b5bb28f9f2a61c2e42b51d78
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-runtime-cu12==12.5.82 \
--hash=sha256:0fd5fbca289bceb9f0690aa9858f06187b554fdeb7e2711dfd5bb3ce58900b46 \
--hash=sha256:3e79a060e126df40fd3a068f3f787eb000fa51b251ec6cd97d09579632687115 \
--hash=sha256:71f015dbf9df05dd71f7480132c6ebf47a6ceb2ab53d7db8e08e4b30ebb87e14
# via -r ci/official/requirements_updater/requirements.in
nvidia-cudnn-cu12==9.3.0.75 \
--hash=sha256:9ad9c6929ebb5295eb4a1728024666d1c88283373e265a0c5c883e6f9d5cd76d \
--hash=sha256:c5cf7ff3415e446adf195a5b7dd2ba56cd00c3ee78bfdc566e51698931aa4b7f \
--hash=sha256:c819e82eed8cf564b9d37478ea4eab9e87194bb3b7f7f8098bc1f67c9b80f1b6
# via -r ci/official/requirements_updater/requirements.in
nvidia-cufft-cu12==11.2.3.61 \
--hash=sha256:4a8f6f0ce93c52a50ee83422a80472b5f376054a63f38532d0eab4007e7ef28b \
--hash=sha256:6d45b48a5ee7599e57131129cda2c58544d9b78b95064d3ec3e5c6b96e2b58cc \
--hash=sha256:9a6e8df162585750f61983a638104a48c756aa13f9f48e19ab079b38e3c828b8
# via -r ci/official/requirements_updater/requirements.in
nvidia-curand-cu12==10.3.6.82 \
--hash=sha256:0631ba65231260ad832ce233ddda57e7b3b7158eabf000d78e46cbb5bd5b7aae \
--hash=sha256:2823fb27de4e44dbb22394a6adf53aa6e1b013aca0f8c22867d1cfae58405536 \
--hash=sha256:36aabeb5990297bbce3df324ea7c7c13c3aabb140c86d50ab3b23e4ec61672f1
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusolver-cu12==11.6.3.83 \
--hash=sha256:1b8b77d2fe8abe72bb722dafb708cceaeb81f1a03999477f20b33b34f46ab885 \
--hash=sha256:6224732963cba312a84c78114b9a38c4ffabb2e2a6a120923ac99ba6f895c8cf \
--hash=sha256:93cfafacde4428b71778eeb092ec615a02a3d05404da1bcf91c53e3fa1bce42b
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusparse-cu12==12.5.1.3 \
--hash=sha256:016df8e993c437e8301e62739f01775cba988fd5253cd4c64173f8e8d2f8e752 \
--hash=sha256:33520db374e2f5ebc976d6faa1852b98c398a57e6f71150fe59705928596ffd1 \
--hash=sha256:7b97fd01f0a61628af99d0efd52132fccc8c18fc5c509f13802dccf0574a19c2
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.23.4 \
--hash=sha256:aa946c8327e22ced28e7cef508a334673abc42064ec85f02d005ba1785ea4cec \
--hash=sha256:b097258d9aab2fa9f686e33c6fe40ae57b27df60cedbd15d139701bb5509e0c1
# via -r ci/official/requirements_updater/requirements.in
nvidia-nvjitlink-cu12==12.5.82 \
--hash=sha256:98103729cc5226e13ca319a10bbf9433bbbd44ef64fe72f45f067cacc14b8d27 \
--hash=sha256:e782564d705ff0bf61ac3e1bf730166da66dd2fe9012f111ede5fc49b64ae697 \
--hash=sha256:f9b37bc5c8cf7509665cb6ada5aaa0ce65618f2332b7d3e78e9790511f111212
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
opt-einsum==3.3.0 \
--hash=sha256:2455e59e3947d3c275477df7f5205b30635e266fe6dc300e3d9f9646bfcea147 \
--hash=sha256:59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549

14
ci/official/requirements_updater/requirements.in
View File

@ -28,6 +28,20 @@ requests >= 2.31.0
packaging==23.2
setuptools==70.0.0
jax==0.4.7
# NVIDIA CUDA dependencies
# Note that the wheels are downloaded only when the targets in bazel command
# contain dependencies on these wheels.
nvidia-cublas-cu12 == 12.5.3.2
nvidia-cuda-cupti-cu12 == 12.5.82
nvidia-cuda-nvrtc-cu12 == 12.5.82
nvidia-cuda-runtime-cu12 == 12.5.82
nvidia-cudnn-cu12 == 9.3.0.75
nvidia-cufft-cu12 == 11.2.3.61
nvidia-curand-cu12 == 10.3.6.82
nvidia-cusolver-cu12 == 11.6.3.83
nvidia-cusparse-cu12 == 12.5.1.3
nvidia-nccl-cu12 == 2.23.4
nvidia-nvjitlink-cu12 == 12.5.82
# The dependencies below are needed for TF wheel testing.
tensorflow-io-gcs-filesystem==0.37.1
libclang >= 13.0.0

63
requirements_lock_3_10.txt
View File

@ -447,6 +447,69 @@ numpy==2.1.1 \
# opt-einsum
# scipy
# tb-nightly
nvidia-cublas-cu12==12.5.3.2 \
--hash=sha256:4960f3dc5f39699acadf76fa6d94b10a2a00f2956c2c442efa299fb22b0748f3 \
--hash=sha256:7d0191251180de606023d396b94d66f66470a0ae96d1dbb906c7656ea0f71eda \
--hash=sha256:ca070ad70e9fa6654084575d01bd001f30cc4665e33d4bb9fc8e0f321caa034b
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.5.82 \
--hash=sha256:4f835281cf492e2bedd153f5c3de9da8f1d775a419468305e64ce73b3b0c6dc3 \
--hash=sha256:bde77a5feb66752ec61db2adfe47f56b941842825b4c7e2068aff27c9d107953 \
--hash=sha256:d32c06490c6ba35c4323730820c7d0c4c126c04ed58d2f57275adb8d54b138fe
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-nvrtc-cu12==12.5.82 \
--hash=sha256:3dbd97b0104b4bfbc3c4f8c79cd2496307c89c43c29a9f83125f1d76296ff3fd \
--hash=sha256:5bb6a0eb01d4974bb7ca3d48bd3859472debb3c3057a5e7de2b08fbdf35eed7e \
--hash=sha256:e5db37e990056c70953b7772dd778336ef9da0a0b5bb28f9f2a61c2e42b51d78
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-runtime-cu12==12.5.82 \
--hash=sha256:0fd5fbca289bceb9f0690aa9858f06187b554fdeb7e2711dfd5bb3ce58900b46 \
--hash=sha256:3e79a060e126df40fd3a068f3f787eb000fa51b251ec6cd97d09579632687115 \
--hash=sha256:71f015dbf9df05dd71f7480132c6ebf47a6ceb2ab53d7db8e08e4b30ebb87e14
# via -r ci/official/requirements_updater/requirements.in
nvidia-cudnn-cu12==9.3.0.75 \
--hash=sha256:9ad9c6929ebb5295eb4a1728024666d1c88283373e265a0c5c883e6f9d5cd76d \
--hash=sha256:c5cf7ff3415e446adf195a5b7dd2ba56cd00c3ee78bfdc566e51698931aa4b7f \
--hash=sha256:c819e82eed8cf564b9d37478ea4eab9e87194bb3b7f7f8098bc1f67c9b80f1b6
# via -r ci/official/requirements_updater/requirements.in
nvidia-cufft-cu12==11.2.3.61 \
--hash=sha256:4a8f6f0ce93c52a50ee83422a80472b5f376054a63f38532d0eab4007e7ef28b \
--hash=sha256:6d45b48a5ee7599e57131129cda2c58544d9b78b95064d3ec3e5c6b96e2b58cc \
--hash=sha256:9a6e8df162585750f61983a638104a48c756aa13f9f48e19ab079b38e3c828b8
# via -r ci/official/requirements_updater/requirements.in
nvidia-curand-cu12==10.3.6.82 \
--hash=sha256:0631ba65231260ad832ce233ddda57e7b3b7158eabf000d78e46cbb5bd5b7aae \
--hash=sha256:2823fb27de4e44dbb22394a6adf53aa6e1b013aca0f8c22867d1cfae58405536 \
--hash=sha256:36aabeb5990297bbce3df324ea7c7c13c3aabb140c86d50ab3b23e4ec61672f1
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusolver-cu12==11.6.3.83 \
--hash=sha256:1b8b77d2fe8abe72bb722dafb708cceaeb81f1a03999477f20b33b34f46ab885 \
--hash=sha256:6224732963cba312a84c78114b9a38c4ffabb2e2a6a120923ac99ba6f895c8cf \
--hash=sha256:93cfafacde4428b71778eeb092ec615a02a3d05404da1bcf91c53e3fa1bce42b
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusparse-cu12==12.5.1.3 \
--hash=sha256:016df8e993c437e8301e62739f01775cba988fd5253cd4c64173f8e8d2f8e752 \
--hash=sha256:33520db374e2f5ebc976d6faa1852b98c398a57e6f71150fe59705928596ffd1 \
--hash=sha256:7b97fd01f0a61628af99d0efd52132fccc8c18fc5c509f13802dccf0574a19c2
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.23.4 \
--hash=sha256:aa946c8327e22ced28e7cef508a334673abc42064ec85f02d005ba1785ea4cec \
--hash=sha256:b097258d9aab2fa9f686e33c6fe40ae57b27df60cedbd15d139701bb5509e0c1
# via -r ci/official/requirements_updater/requirements.in
nvidia-nvjitlink-cu12==12.5.82 \
--hash=sha256:98103729cc5226e13ca319a10bbf9433bbbd44ef64fe72f45f067cacc14b8d27 \
--hash=sha256:e782564d705ff0bf61ac3e1bf730166da66dd2fe9012f111ede5fc49b64ae697 \
--hash=sha256:f9b37bc5c8cf7509665cb6ada5aaa0ce65618f2332b7d3e78e9790511f111212
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
opt-einsum==3.3.0 \
--hash=sha256:2455e59e3947d3c275477df7f5205b30635e266fe6dc300e3d9f9646bfcea147 \
--hash=sha256:59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549

63
requirements_lock_3_11.txt
View File

@ -447,6 +447,69 @@ numpy==2.1.1 \
# opt-einsum
# scipy
# tb-nightly
nvidia-cublas-cu12==12.5.3.2 \
--hash=sha256:4960f3dc5f39699acadf76fa6d94b10a2a00f2956c2c442efa299fb22b0748f3 \
--hash=sha256:7d0191251180de606023d396b94d66f66470a0ae96d1dbb906c7656ea0f71eda \
--hash=sha256:ca070ad70e9fa6654084575d01bd001f30cc4665e33d4bb9fc8e0f321caa034b
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.5.82 \
--hash=sha256:4f835281cf492e2bedd153f5c3de9da8f1d775a419468305e64ce73b3b0c6dc3 \
--hash=sha256:bde77a5feb66752ec61db2adfe47f56b941842825b4c7e2068aff27c9d107953 \
--hash=sha256:d32c06490c6ba35c4323730820c7d0c4c126c04ed58d2f57275adb8d54b138fe
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-nvrtc-cu12==12.5.82 \
--hash=sha256:3dbd97b0104b4bfbc3c4f8c79cd2496307c89c43c29a9f83125f1d76296ff3fd \
--hash=sha256:5bb6a0eb01d4974bb7ca3d48bd3859472debb3c3057a5e7de2b08fbdf35eed7e \
--hash=sha256:e5db37e990056c70953b7772dd778336ef9da0a0b5bb28f9f2a61c2e42b51d78
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-runtime-cu12==12.5.82 \
--hash=sha256:0fd5fbca289bceb9f0690aa9858f06187b554fdeb7e2711dfd5bb3ce58900b46 \
--hash=sha256:3e79a060e126df40fd3a068f3f787eb000fa51b251ec6cd97d09579632687115 \
--hash=sha256:71f015dbf9df05dd71f7480132c6ebf47a6ceb2ab53d7db8e08e4b30ebb87e14
# via -r ci/official/requirements_updater/requirements.in
nvidia-cudnn-cu12==9.3.0.75 \
--hash=sha256:9ad9c6929ebb5295eb4a1728024666d1c88283373e265a0c5c883e6f9d5cd76d \
--hash=sha256:c5cf7ff3415e446adf195a5b7dd2ba56cd00c3ee78bfdc566e51698931aa4b7f \
--hash=sha256:c819e82eed8cf564b9d37478ea4eab9e87194bb3b7f7f8098bc1f67c9b80f1b6
# via -r ci/official/requirements_updater/requirements.in
nvidia-cufft-cu12==11.2.3.61 \
--hash=sha256:4a8f6f0ce93c52a50ee83422a80472b5f376054a63f38532d0eab4007e7ef28b \
--hash=sha256:6d45b48a5ee7599e57131129cda2c58544d9b78b95064d3ec3e5c6b96e2b58cc \
--hash=sha256:9a6e8df162585750f61983a638104a48c756aa13f9f48e19ab079b38e3c828b8
# via -r ci/official/requirements_updater/requirements.in
nvidia-curand-cu12==10.3.6.82 \
--hash=sha256:0631ba65231260ad832ce233ddda57e7b3b7158eabf000d78e46cbb5bd5b7aae \
--hash=sha256:2823fb27de4e44dbb22394a6adf53aa6e1b013aca0f8c22867d1cfae58405536 \
--hash=sha256:36aabeb5990297bbce3df324ea7c7c13c3aabb140c86d50ab3b23e4ec61672f1
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusolver-cu12==11.6.3.83 \
--hash=sha256:1b8b77d2fe8abe72bb722dafb708cceaeb81f1a03999477f20b33b34f46ab885 \
--hash=sha256:6224732963cba312a84c78114b9a38c4ffabb2e2a6a120923ac99ba6f895c8cf \
--hash=sha256:93cfafacde4428b71778eeb092ec615a02a3d05404da1bcf91c53e3fa1bce42b
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusparse-cu12==12.5.1.3 \
--hash=sha256:016df8e993c437e8301e62739f01775cba988fd5253cd4c64173f8e8d2f8e752 \
--hash=sha256:33520db374e2f5ebc976d6faa1852b98c398a57e6f71150fe59705928596ffd1 \
--hash=sha256:7b97fd01f0a61628af99d0efd52132fccc8c18fc5c509f13802dccf0574a19c2
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.23.4 \
--hash=sha256:aa946c8327e22ced28e7cef508a334673abc42064ec85f02d005ba1785ea4cec \
--hash=sha256:b097258d9aab2fa9f686e33c6fe40ae57b27df60cedbd15d139701bb5509e0c1
# via -r ci/official/requirements_updater/requirements.in
nvidia-nvjitlink-cu12==12.5.82 \
--hash=sha256:98103729cc5226e13ca319a10bbf9433bbbd44ef64fe72f45f067cacc14b8d27 \
--hash=sha256:e782564d705ff0bf61ac3e1bf730166da66dd2fe9012f111ede5fc49b64ae697 \
--hash=sha256:f9b37bc5c8cf7509665cb6ada5aaa0ce65618f2332b7d3e78e9790511f111212
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
opt-einsum==3.3.0 \
--hash=sha256:2455e59e3947d3c275477df7f5205b30635e266fe6dc300e3d9f9646bfcea147 \
--hash=sha256:59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549

63
requirements_lock_3_12.txt
View File

@ -447,6 +447,69 @@ numpy==2.1.1 \
# opt-einsum
# scipy
# tb-nightly
nvidia-cublas-cu12==12.5.3.2 \
--hash=sha256:4960f3dc5f39699acadf76fa6d94b10a2a00f2956c2c442efa299fb22b0748f3 \
--hash=sha256:7d0191251180de606023d396b94d66f66470a0ae96d1dbb906c7656ea0f71eda \
--hash=sha256:ca070ad70e9fa6654084575d01bd001f30cc4665e33d4bb9fc8e0f321caa034b
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.5.82 \
--hash=sha256:4f835281cf492e2bedd153f5c3de9da8f1d775a419468305e64ce73b3b0c6dc3 \
--hash=sha256:bde77a5feb66752ec61db2adfe47f56b941842825b4c7e2068aff27c9d107953 \
--hash=sha256:d32c06490c6ba35c4323730820c7d0c4c126c04ed58d2f57275adb8d54b138fe
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-nvrtc-cu12==12.5.82 \
--hash=sha256:3dbd97b0104b4bfbc3c4f8c79cd2496307c89c43c29a9f83125f1d76296ff3fd \
--hash=sha256:5bb6a0eb01d4974bb7ca3d48bd3859472debb3c3057a5e7de2b08fbdf35eed7e \
--hash=sha256:e5db37e990056c70953b7772dd778336ef9da0a0b5bb28f9f2a61c2e42b51d78
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-runtime-cu12==12.5.82 \
--hash=sha256:0fd5fbca289bceb9f0690aa9858f06187b554fdeb7e2711dfd5bb3ce58900b46 \
--hash=sha256:3e79a060e126df40fd3a068f3f787eb000fa51b251ec6cd97d09579632687115 \
--hash=sha256:71f015dbf9df05dd71f7480132c6ebf47a6ceb2ab53d7db8e08e4b30ebb87e14
# via -r ci/official/requirements_updater/requirements.in
nvidia-cudnn-cu12==9.3.0.75 \
--hash=sha256:9ad9c6929ebb5295eb4a1728024666d1c88283373e265a0c5c883e6f9d5cd76d \
--hash=sha256:c5cf7ff3415e446adf195a5b7dd2ba56cd00c3ee78bfdc566e51698931aa4b7f \
--hash=sha256:c819e82eed8cf564b9d37478ea4eab9e87194bb3b7f7f8098bc1f67c9b80f1b6
# via -r ci/official/requirements_updater/requirements.in
nvidia-cufft-cu12==11.2.3.61 \
--hash=sha256:4a8f6f0ce93c52a50ee83422a80472b5f376054a63f38532d0eab4007e7ef28b \
--hash=sha256:6d45b48a5ee7599e57131129cda2c58544d9b78b95064d3ec3e5c6b96e2b58cc \
--hash=sha256:9a6e8df162585750f61983a638104a48c756aa13f9f48e19ab079b38e3c828b8
# via -r ci/official/requirements_updater/requirements.in
nvidia-curand-cu12==10.3.6.82 \
--hash=sha256:0631ba65231260ad832ce233ddda57e7b3b7158eabf000d78e46cbb5bd5b7aae \
--hash=sha256:2823fb27de4e44dbb22394a6adf53aa6e1b013aca0f8c22867d1cfae58405536 \
--hash=sha256:36aabeb5990297bbce3df324ea7c7c13c3aabb140c86d50ab3b23e4ec61672f1
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusolver-cu12==11.6.3.83 \
--hash=sha256:1b8b77d2fe8abe72bb722dafb708cceaeb81f1a03999477f20b33b34f46ab885 \
--hash=sha256:6224732963cba312a84c78114b9a38c4ffabb2e2a6a120923ac99ba6f895c8cf \
--hash=sha256:93cfafacde4428b71778eeb092ec615a02a3d05404da1bcf91c53e3fa1bce42b
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusparse-cu12==12.5.1.3 \
--hash=sha256:016df8e993c437e8301e62739f01775cba988fd5253cd4c64173f8e8d2f8e752 \
--hash=sha256:33520db374e2f5ebc976d6faa1852b98c398a57e6f71150fe59705928596ffd1 \
--hash=sha256:7b97fd01f0a61628af99d0efd52132fccc8c18fc5c509f13802dccf0574a19c2
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.23.4 \
--hash=sha256:aa946c8327e22ced28e7cef508a334673abc42064ec85f02d005ba1785ea4cec \
--hash=sha256:b097258d9aab2fa9f686e33c6fe40ae57b27df60cedbd15d139701bb5509e0c1
# via -r ci/official/requirements_updater/requirements.in
nvidia-nvjitlink-cu12==12.5.82 \
--hash=sha256:98103729cc5226e13ca319a10bbf9433bbbd44ef64fe72f45f067cacc14b8d27 \
--hash=sha256:e782564d705ff0bf61ac3e1bf730166da66dd2fe9012f111ede5fc49b64ae697 \
--hash=sha256:f9b37bc5c8cf7509665cb6ada5aaa0ce65618f2332b7d3e78e9790511f111212
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
opt-einsum==3.3.0 \
--hash=sha256:2455e59e3947d3c275477df7f5205b30635e266fe6dc300e3d9f9646bfcea147 \
--hash=sha256:59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549

63
requirements_lock_3_9.txt
View File

@ -443,6 +443,69 @@ numpy==2.0.2 \
# opt-einsum
# scipy
# tb-nightly
nvidia-cublas-cu12==12.5.3.2 \
--hash=sha256:4960f3dc5f39699acadf76fa6d94b10a2a00f2956c2c442efa299fb22b0748f3 \
--hash=sha256:7d0191251180de606023d396b94d66f66470a0ae96d1dbb906c7656ea0f71eda \
--hash=sha256:ca070ad70e9fa6654084575d01bd001f30cc4665e33d4bb9fc8e0f321caa034b
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.5.82 \
--hash=sha256:4f835281cf492e2bedd153f5c3de9da8f1d775a419468305e64ce73b3b0c6dc3 \
--hash=sha256:bde77a5feb66752ec61db2adfe47f56b941842825b4c7e2068aff27c9d107953 \
--hash=sha256:d32c06490c6ba35c4323730820c7d0c4c126c04ed58d2f57275adb8d54b138fe
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-nvrtc-cu12==12.5.82 \
--hash=sha256:3dbd97b0104b4bfbc3c4f8c79cd2496307c89c43c29a9f83125f1d76296ff3fd \
--hash=sha256:5bb6a0eb01d4974bb7ca3d48bd3859472debb3c3057a5e7de2b08fbdf35eed7e \
--hash=sha256:e5db37e990056c70953b7772dd778336ef9da0a0b5bb28f9f2a61c2e42b51d78
# via -r ci/official/requirements_updater/requirements.in
nvidia-cuda-runtime-cu12==12.5.82 \
--hash=sha256:0fd5fbca289bceb9f0690aa9858f06187b554fdeb7e2711dfd5bb3ce58900b46 \
--hash=sha256:3e79a060e126df40fd3a068f3f787eb000fa51b251ec6cd97d09579632687115 \
--hash=sha256:71f015dbf9df05dd71f7480132c6ebf47a6ceb2ab53d7db8e08e4b30ebb87e14
# via -r ci/official/requirements_updater/requirements.in
nvidia-cudnn-cu12==9.3.0.75 \
--hash=sha256:9ad9c6929ebb5295eb4a1728024666d1c88283373e265a0c5c883e6f9d5cd76d \
--hash=sha256:c5cf7ff3415e446adf195a5b7dd2ba56cd00c3ee78bfdc566e51698931aa4b7f \
--hash=sha256:c819e82eed8cf564b9d37478ea4eab9e87194bb3b7f7f8098bc1f67c9b80f1b6
# via -r ci/official/requirements_updater/requirements.in
nvidia-cufft-cu12==11.2.3.61 \
--hash=sha256:4a8f6f0ce93c52a50ee83422a80472b5f376054a63f38532d0eab4007e7ef28b \
--hash=sha256:6d45b48a5ee7599e57131129cda2c58544d9b78b95064d3ec3e5c6b96e2b58cc \
--hash=sha256:9a6e8df162585750f61983a638104a48c756aa13f9f48e19ab079b38e3c828b8
# via -r ci/official/requirements_updater/requirements.in
nvidia-curand-cu12==10.3.6.82 \
--hash=sha256:0631ba65231260ad832ce233ddda57e7b3b7158eabf000d78e46cbb5bd5b7aae \
--hash=sha256:2823fb27de4e44dbb22394a6adf53aa6e1b013aca0f8c22867d1cfae58405536 \
--hash=sha256:36aabeb5990297bbce3df324ea7c7c13c3aabb140c86d50ab3b23e4ec61672f1
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusolver-cu12==11.6.3.83 \
--hash=sha256:1b8b77d2fe8abe72bb722dafb708cceaeb81f1a03999477f20b33b34f46ab885 \
--hash=sha256:6224732963cba312a84c78114b9a38c4ffabb2e2a6a120923ac99ba6f895c8cf \
--hash=sha256:93cfafacde4428b71778eeb092ec615a02a3d05404da1bcf91c53e3fa1bce42b
# via -r ci/official/requirements_updater/requirements.in
nvidia-cusparse-cu12==12.5.1.3 \
--hash=sha256:016df8e993c437e8301e62739f01775cba988fd5253cd4c64173f8e8d2f8e752 \
--hash=sha256:33520db374e2f5ebc976d6faa1852b98c398a57e6f71150fe59705928596ffd1 \
--hash=sha256:7b97fd01f0a61628af99d0efd52132fccc8c18fc5c509f13802dccf0574a19c2
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.23.4 \
--hash=sha256:aa946c8327e22ced28e7cef508a334673abc42064ec85f02d005ba1785ea4cec \
--hash=sha256:b097258d9aab2fa9f686e33c6fe40ae57b27df60cedbd15d139701bb5509e0c1
# via -r ci/official/requirements_updater/requirements.in
nvidia-nvjitlink-cu12==12.5.82 \
--hash=sha256:98103729cc5226e13ca319a10bbf9433bbbd44ef64fe72f45f067cacc14b8d27 \
--hash=sha256:e782564d705ff0bf61ac3e1bf730166da66dd2fe9012f111ede5fc49b64ae697 \
--hash=sha256:f9b37bc5c8cf7509665cb6ada5aaa0ce65618f2332b7d3e78e9790511f111212
# via
# -r ci/official/requirements_updater/requirements.in
# nvidia-cufft-cu12
# nvidia-cusolver-cu12
# nvidia-cusparse-cu12
opt-einsum==3.3.0 \
--hash=sha256:2455e59e3947d3c275477df7f5205b30635e266fe6dc300e3d9f9646bfcea147 \
--hash=sha256:59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549

44
tensorflow/tools/pip_package/BUILD
View File

@ -7,10 +7,15 @@ load(
"@local_tsl//third_party/py:py_import.bzl",
"py_import",
)
load("@local_xla//xla/tsl:tsl.bzl", "if_cuda_libs")
load("@local_xla//xla/tsl/mkl:build_defs.bzl", "if_enable_mkl", "if_mkl", "if_mkl_ml")
load("//tensorflow:tensorflow.bzl", "if_wheel_dependency", "if_with_tpu_support", "transitive_hdrs")
load("//tensorflow/core/platform:build_config_root.bzl", "if_pywrap", "tf_additional_license_deps")
load(
"//tensorflow/core/platform:build_config_root.bzl",
"if_pywrap",
"tf_additional_license_deps",
"tf_cuda_tests_tags",
"tf_exec_properties",
)
load("//tensorflow/tools/pip_package/utils:data_deps.bzl", "collect_data_files")
load("//tensorflow/tools/pip_package/utils:py_deps.bzl", "transitive_py_deps")
load("//tensorflow/tools/pip_package/utils:tf_wheel.bzl", "tf_wheel", "tf_wheel_dep")
@ -339,6 +344,10 @@ py_test(
["import_api_packages_test.py"],
[":empty_test"],
),
exec_properties = if_cuda(
tf_exec_properties({"tags": tf_cuda_tests_tags()}),
{},
),
main = if_wheel_dependency("import_api_packages_test.py", "empty_test.py"),
tags = [
"cpu",
@ -351,6 +360,10 @@ py_test(
py_test(
name = "import_api_packages_test",
srcs = ["import_api_packages_test.py"],
exec_properties = if_cuda(
tf_exec_properties({"tags": tf_cuda_tests_tags()}),
{},
),
main = "import_api_packages_test.py",
tags = [
"cpu",
@ -364,21 +377,20 @@ py_test(
py_import(
name = "tf_py_import",
cc_deps = if_cuda_libs([
"@cuda_cublas//:cublas",
"@cuda_cublas//:cublasLt",
"@cuda_cudart//:cudart",
"@cuda_cudnn//:cudnn",
"@cuda_cufft//:cufft",
"@cuda_cupti//:cupti",
"@cuda_curand//:curand",
"@cuda_cusolver//:cusolver",
"@cuda_cusparse//:cusparse",
"@cuda_nccl//:nccl",
"@cuda_nvjitlink//:nvjitlink",
"@cuda_nvrtc//:nvrtc",
]),
wheel = ":wheel",
wheel_deps = if_cuda([
"@pypi_nvidia_cublas_cu12//:whl",
"@pypi_nvidia_cuda_cupti_cu12//:whl",
"@pypi_nvidia_cuda_nvrtc_cu12//:whl",
"@pypi_nvidia_cuda_runtime_cu12//:whl",
"@pypi_nvidia_cudnn_cu12//:whl",
"@pypi_nvidia_cufft_cu12//:whl",
"@pypi_nvidia_curand_cu12//:whl",
"@pypi_nvidia_cusolver_cu12//:whl",
"@pypi_nvidia_cusparse_cu12//:whl",
"@pypi_nvidia_nccl_cu12//:whl",
"@pypi_nvidia_nvjitlink_cu12//:whl",
]),
deps = [
"@pypi_absl_py//:pkg",
"@pypi_astunparse//:pkg",

6
third_party/gpus/cuda/hermetic/cuda_cublas.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -21,12 +25,14 @@ cc_library(
name = "cublas",
visibility = ["//visibility:public"],
%{comment}deps = [":cublas_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cublas/lib"),
)
cc_library(
name = "cublasLt",
visibility = ["//visibility:public"],
%{comment}deps = [":cublasLt_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cublas/lib"),
)
cc_library(

5
third_party/gpus/cuda/hermetic/cuda_cudart.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -36,6 +40,7 @@ cc_library(
%{comment}}) + [
%{comment}":cudart_shared_library",
%{comment}],
%{comment}linkopts = cuda_rpath_flags("nvidia/cuda_runtime/lib"),
visibility = ["//visibility:public"],
)

5
third_party/gpus/cuda/hermetic/cuda_cudnn.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -58,6 +62,7 @@ cc_library(
%{comment}"@cuda_nvrtc//:nvrtc",
%{comment}":cudnn_main",
%{comment}],
%{comment}linkopts = cuda_rpath_flags("nvidia/cudnn/lib"),
visibility = ["//visibility:public"],
)

5
third_party/gpus/cuda/hermetic/cuda_cudnn9.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -65,6 +69,7 @@ cc_library(
%{comment}"@cuda_nvrtc//:nvrtc",
%{comment}":cudnn_main",
%{comment}],
%{comment}linkopts = cuda_rpath_flags("nvidia/cudnn/lib"),
visibility = ["//visibility:public"],
)

5
third_party/gpus/cuda/hermetic/cuda_cufft.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -13,6 +17,7 @@ cc_import(
cc_library(
name = "cufft",
%{comment}deps = [":cufft_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cufft/lib"),
visibility = ["//visibility:public"],
)

6
third_party/gpus/cuda/hermetic/cuda_cupti.BUILD.tpl vendored
View File

@ -1,5 +1,10 @@
licenses(["restricted"]) # NVIDIA proprietary license
load("@local_config_cuda//cuda:build_defs.bzl", "if_version_equal_or_greater_than")
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
])
@ -13,6 +18,7 @@ cc_import(
cc_library(
name = "cupti",
%{comment}deps = [":cupti_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cuda_cupti/lib"),
visibility = ["//visibility:public"],
)

5
third_party/gpus/cuda/hermetic/cuda_curand.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -13,6 +17,7 @@ cc_import(
cc_library(
name = "curand",
%{comment}deps = [":curand_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/curand/lib"),
visibility = ["//visibility:public"],
)

5
third_party/gpus/cuda/hermetic/cuda_cusolver.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -19,6 +23,7 @@ cc_import(
cc_library(
name = "cusolver",
%{comment}deps = [":cusolver_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cusolver/lib"),
visibility = ["//visibility:public"],
)

5
third_party/gpus/cuda/hermetic/cuda_cusparse.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -14,6 +18,7 @@ cc_import(
cc_library(
name = "cusparse",
%{comment}deps = [":cusparse_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cusparse/lib"),
visibility = ["//visibility:public"],
)

5
third_party/gpus/cuda/hermetic/cuda_nvjitlink.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -13,6 +17,7 @@ cc_import(
cc_library(
name = "nvjitlink",
%{comment}deps = [":nvjitlink_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/nvjitlink/lib"),
visibility = ["//visibility:public"],
)

6
third_party/gpus/cuda/hermetic/cuda_nvrtc.BUILD.tpl vendored
View File

@ -1,4 +1,9 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
%{multiline_comment}
cc_import(
name = "nvrtc_main",
@ -16,5 +21,6 @@ cc_library(
%{comment}":nvrtc_main",
%{comment}":nvrtc_builtins",
%{comment}],
%{comment}linkopts = cuda_rpath_flags("nvidia/cuda_nvrtc/lib"),
visibility = ["//visibility:public"],
)

5
third_party/nccl/hermetic/cuda_nccl.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -14,6 +18,7 @@ cc_import(
cc_library(
name = "nccl",
%{comment}deps = [":nccl_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/nccl/lib"),
visibility = ["//visibility:public"],
)

6
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_cublas.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -21,12 +25,14 @@ cc_library(
name = "cublas",
visibility = ["//visibility:public"],
%{comment}deps = [":cublas_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cublas/lib"),
)
cc_library(
name = "cublasLt",
visibility = ["//visibility:public"],
%{comment}deps = [":cublasLt_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cublas/lib"),
)
cc_library(

5
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_cudart.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -36,6 +40,7 @@ cc_library(
%{comment}}) + [
%{comment}":cudart_shared_library",
%{comment}],
%{comment}linkopts = cuda_rpath_flags("nvidia/cuda_runtime/lib"),
visibility = ["//visibility:public"],
)

5
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_cudnn.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -58,6 +62,7 @@ cc_library(
%{comment}"@cuda_nvrtc//:nvrtc",
%{comment}":cudnn_main",
%{comment}],
%{comment}linkopts = cuda_rpath_flags("nvidia/cudnn/lib"),
visibility = ["//visibility:public"],
)

5
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_cudnn9.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -65,6 +69,7 @@ cc_library(
%{comment}"@cuda_nvrtc//:nvrtc",
%{comment}":cudnn_main",
%{comment}],
%{comment}linkopts = cuda_rpath_flags("nvidia/cudnn/lib"),
visibility = ["//visibility:public"],
)

5
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_cufft.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -13,6 +17,7 @@ cc_import(
cc_library(
name = "cufft",
%{comment}deps = [":cufft_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cufft/lib"),
visibility = ["//visibility:public"],
)

6
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_cupti.BUILD.tpl vendored
View File

@ -1,5 +1,10 @@
licenses(["restricted"]) # NVIDIA proprietary license
load("@local_config_cuda//cuda:build_defs.bzl", "if_version_equal_or_greater_than")
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
])
@ -13,6 +18,7 @@ cc_import(
cc_library(
name = "cupti",
%{comment}deps = [":cupti_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cuda_cupti/lib"),
visibility = ["//visibility:public"],
)

5
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_curand.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -13,6 +17,7 @@ cc_import(
cc_library(
name = "curand",
%{comment}deps = [":curand_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/curand/lib"),
visibility = ["//visibility:public"],
)

5
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_cusolver.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -19,6 +23,7 @@ cc_import(
cc_library(
name = "cusolver",
%{comment}deps = [":cusolver_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cusolver/lib"),
visibility = ["//visibility:public"],
)

5
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_cusparse.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -14,6 +18,7 @@ cc_import(
cc_library(
name = "cusparse",
%{comment}deps = [":cusparse_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/cusparse/lib"),
visibility = ["//visibility:public"],
)

5
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_nvjitlink.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -13,6 +17,7 @@ cc_import(
cc_library(
name = "nvjitlink",
%{comment}deps = [":nvjitlink_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/nvjitlink/lib"),
visibility = ["//visibility:public"],
)

6
third_party/xla/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_nvrtc.BUILD.tpl vendored
View File

@ -1,4 +1,9 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
%{multiline_comment}
cc_import(
name = "nvrtc_main",
@ -16,5 +21,6 @@ cc_library(
%{comment}":nvrtc_main",
%{comment}":nvrtc_builtins",
%{comment}],
%{comment}linkopts = cuda_rpath_flags("nvidia/cuda_nvrtc/lib"),
visibility = ["//visibility:public"],
)

5
third_party/xla/third_party/tsl/third_party/nccl/hermetic/cuda_nccl.BUILD.tpl vendored
View File

@ -1,4 +1,8 @@
licenses(["restricted"]) # NVIDIA proprietary license
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
exports_files([
"version.txt",
@ -14,6 +18,7 @@ cc_import(
cc_library(
name = "nccl",
%{comment}deps = [":nccl_shared_library"],
%{comment}linkopts = cuda_rpath_flags("nvidia/nccl/lib"),
visibility = ["//visibility:public"],
)

32
third_party/xla/third_party/tsl/third_party/py/py_import.bzl vendored
View File

@ -2,13 +2,6 @@
def _unpacked_wheel_impl(ctx):
output_dir = ctx.actions.declare_directory(ctx.label.name)
libs = []
for dep in ctx.attr.cc_deps:
linker_inputs = dep[CcInfo].linking_context.linker_inputs.to_list()
for linker_input in linker_inputs:
if linker_input.libraries and linker_input.libraries[0].dynamic_library:
lib = linker_input.libraries[0].dynamic_library
libs.append(lib)
wheel = None
for w in ctx.files.wheel_rule_outputs:
if w.basename.endswith(".whl"):
@ -16,17 +9,20 @@ def _unpacked_wheel_impl(ctx):
break
script = """
{zipper} x {wheel} -d {output}
for lib in {libs}; do
cp $lib {output}/tensorflow
for wheel_dep in {wheel_deps}; do
{zipper} x $wheel_dep -d {output}
done
""".format(
zipper = ctx.executable.zipper.path,
wheel = wheel.path,
output = output_dir.path,
libs = " ".join(["'%s'" % lib.path for lib in libs]),
wheel_deps = " ".join([
"'%s'" % wheel_dep.path
for wheel_dep in ctx.files.wheel_deps
]),
)
ctx.actions.run_shell(
inputs = ctx.files.wheel_rule_outputs + libs,
inputs = ctx.files.wheel_rule_outputs + ctx.files.wheel_deps,
command = script,
outputs = [output_dir],
tools = [ctx.executable.zipper],
@ -45,16 +41,20 @@ _unpacked_wheel = rule(
cfg = "exec",
executable = True,
),
"cc_deps": attr.label_list(providers = [CcInfo]),
"wheel_deps": attr.label_list(allow_files = True),
},
)
def py_import(name, wheel, deps = [], cc_deps = []):
def py_import(
name,
wheel,
deps = [],
wheel_deps = []):
unpacked_wheel_name = name + "_unpacked_wheel"
_unpacked_wheel(
name = unpacked_wheel_name,
wheel_rule_outputs = wheel,
cc_deps = cc_deps,
wheel_deps = wheel_deps,
)
native.py_library(
name = name,
@ -68,6 +68,6 @@ def py_import(name, wheel, deps = [], cc_deps = []):
Args:
wheel: wheel file to unpack.
deps: dependencies of the py_library.
cc_deps: dependencies that will be copied in the folder
with the unpacked wheel content.
wheel_deps: additional wheels to unpack. These wheels will be unpacked in the
same folder as the wheel.
""" # buildifier: disable=no-effect