mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add NVIDIA wheel requirements only for Linux builds.

Browse Source 
PiperOrigin-RevId: 735850240
This commit is contained in:
jax authors 2025-03-11 12:32:58 -07:00
parent 7ac6355262
commit 0db14aa342
7 changed files with 63 additions and 63 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
build/BUILD.bazel
View File

@ -29,7 +29,7 @@ compile_pip_requirements(
requirements_in = "requirements.in",
requirements_txt = REQUIREMENTS,
generate_hashes = True,
data = ["test-requirements.txt"]
data = ["test-requirements.txt", "gpu-test-requirements.txt"]
)
compile_pip_requirements(
@ -44,7 +44,7 @@ compile_pip_requirements(
requirements_in = "requirements.in",
requirements_txt = REQUIREMENTS,
generate_hashes = False,
data = ["test-requirements.txt"]
data = ["test-requirements.txt", "gpu-test-requirements.txt"]
)
compile_pip_requirements(
@ -58,7 +58,7 @@ compile_pip_requirements(
requirements_in = "requirements.in",
requirements_txt = REQUIREMENTS,
generate_hashes = False,
data = ["test-requirements.txt"]
data = ["test-requirements.txt", "gpu-test-requirements.txt"]
)
py_library(

20
build/gpu-test-requirements.txt
View File

@ -1,13 +1,13 @@
# 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.1.3.1
nvidia-cuda-cupti-cu12>=12.1.105
nvidia-cuda-nvcc-cu12>=12.6.85
nvidia-cuda-runtime-cu12>=12.1.105
nvidia-cudnn-cu12>=9.1,<10.0
nvidia-cufft-cu12>=11.0.2.54
nvidia-cusolver-cu12>=11.4.5.107
nvidia-cusparse-cu12>=12.1.0.106
nvidia-nccl-cu12>=2.18.1
nvidia-nvjitlink-cu12>=12.1.105
nvidia-cublas-cu12>=12.1.3.1 ; sys_platform == "linux"
nvidia-cuda-cupti-cu12>=12.1.105 ; sys_platform == "linux"
nvidia-cuda-nvcc-cu12>=12.6.85 ; sys_platform == "linux"
nvidia-cuda-runtime-cu12>=12.1.105 ; sys_platform == "linux"
nvidia-cudnn-cu12>=9.1,<10.0 ; sys_platform == "linux"
nvidia-cufft-cu12>=11.0.2.54 ; sys_platform == "linux"
nvidia-cusolver-cu12>=11.4.5.107 ; sys_platform == "linux"
nvidia-cusparse-cu12>=12.1.0.106 ; sys_platform == "linux"
nvidia-nccl-cu12>=2.18.1 ; sys_platform == "linux"
nvidia-nvjitlink-cu12>=12.1.105 ; sys_platform == "linux"

20
build/requirements_lock_3_10.txt
View File

@ -380,7 +380,7 @@ numpy==2.0.0 ; python_version <= "3.12" \
# ml-dtypes
# opt-einsum
# scipy
nvidia-cublas-cu12==12.8.3.14 \
nvidia-cublas-cu12==12.8.3.14 ; sys_platform == "linux" \
--hash=sha256:3f0e05e7293598cf61933258b73e66a160c27d59c4422670bf0b79348c04be44 \
--hash=sha256:93a4e0e386cc7f6e56c822531396de8170ed17068a1e18f987574895044cd8c3 \
--hash=sha256:9ae5eae500aead01fc4bdfc458209df638b1a3551557ce11a78eea9ece602ae9
@ -388,48 +388,48 @@ nvidia-cublas-cu12==12.8.3.14 \
# via -r build/test-requirements.txt
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.8.57 \
nvidia-cuda-cupti-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:8e0b2eb847de260739bee4a3f66fac31378f4ff49538ff527a38a01a9a39f950 \
--hash=sha256:bbed719c52a476958a74cfc42f2b95a3fd6b3fd94eb40134acc4601feb4acac3 \
--hash=sha256:ff154211724fd824e758ce176b66007b558eea19c9a5135fc991827ee147e317
# via -r build/test-requirements.txt
nvidia-cuda-nvcc-cu12==12.8.61 \
nvidia-cuda-nvcc-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:171f605044ba17bc455d19cad289946c3dbea029a90c60dfa7b88e545bc8e329 \
--hash=sha256:28604ec42aaa09035b0fb7111432e5121bc385580b30c55d2acfb7d644b16548 \
--hash=sha256:4524739cfc080e9c9e53032912be8f020058e0a7186746d19acef3b6d916ea0b
# via -r build/test-requirements.txt
nvidia-cuda-runtime-cu12==12.8.57 \
nvidia-cuda-runtime-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:534ccebd967b6a44292678fa5da4f00666029cb2ed07a79515ea41ef31fe3ec7 \
--hash=sha256:75342e28567340b7428ce79a5d6bb6ca5ff9d07b69e7ce00d2c7b4dc23eff0be \
--hash=sha256:89be637e3ee967323865b85e0f147d75f9a5bd98360befa37481b02dd57af8f5
# via -r build/test-requirements.txt
nvidia-cudnn-cu12==9.7.1.26 \
nvidia-cudnn-cu12==9.7.1.26 ; sys_platform == "linux" \
--hash=sha256:6d011159a158f3cfc47bf851aea79e31bcff60d530b70ef70474c84cac484d07 \
--hash=sha256:7b805b9a4cf9f3da7c5f4ea4a9dff7baf62d1a612d6154a7e0d2ea51ed296241 \
--hash=sha256:848a61d40ef3b32bd4e1fadb599f0cf04a4b942fbe5fb3be572ad75f9b8c53ef
# via -r build/test-requirements.txt
nvidia-cufft-cu12==11.3.3.41 \
nvidia-cufft-cu12==11.3.3.41 ; sys_platform == "linux" \
--hash=sha256:68509dcd7e3306e69d0e2d8a6d21c8b25ed62e6df8aac192ce752f17677398b5 \
--hash=sha256:da650080ab79fcdf7a4b06aa1b460e99860646b176a43f6208099bdc17836b6a \
--hash=sha256:f9760612886786601d27a0993bb29ce1f757e6b8b173499d0ecfa850d31b50f8
# via -r build/test-requirements.txt
nvidia-cusolver-cu12==11.7.2.55 \
nvidia-cusolver-cu12==11.7.2.55 ; sys_platform == "linux" \
--hash=sha256:0fd9e98246f43c15bee5561147ad235dfdf2d037f5d07c9d41af3f7f72feb7cc \
--hash=sha256:4d1354102f1e922cee9db51920dba9e2559877cf6ff5ad03a00d853adafb191b \
--hash=sha256:a5a516c55da5c5aba98420d9bc9bcab18245f21ec87338cc1f930eb18dd411ac
# via -r build/test-requirements.txt
nvidia-cusparse-cu12==12.5.7.53 \
nvidia-cusparse-cu12==12.5.7.53 ; sys_platform == "linux" \
--hash=sha256:3c1b61eb8c85257ea07e9354606b26397612627fdcd327bfd91ccf6155e7c86d \
--hash=sha256:82c201d6781bacf6bb7c654f0446728d0fe596dfdd82ef4a04c204ce3e107441 \
--hash=sha256:d869c6146ca80f4305b62e02d924b4aaced936f8173e3cef536a67eed2a91af1
# via
# via -r build/test-requirements.txt
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.25.1 \
nvidia-nccl-cu12==2.25.1 ; sys_platform == "linux" \
--hash=sha256:362aed5963fb9ea2ed2f264409baae30143498fd0e5c503aeaa1badd88cdc54a \
--hash=sha256:4ab428bc915785cc66e8c57cb34c7a64cf739c46702b8db748b6ad6cc7180cf8
# via -r build/test-requirements.txt
nvidia-nvjitlink-cu12==12.8.61 \
nvidia-nvjitlink-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:1166a964d25fdc0eae497574d38824305195a5283324a21ccb0ce0c802cbf41c \
--hash=sha256:45fd79f2ae20bd67e8bc411055939049873bfd8fac70ff13bd4865e0b9bdab17 \
--hash=sha256:9b80ecab31085dda3ce3b41d043be0ec739216c3fc633b8abe212d5a30026df0

20
build/requirements_lock_3_11.txt
View File

@ -375,7 +375,7 @@ numpy==2.0.0 ; python_version <= "3.12" \
# ml-dtypes
# opt-einsum
# scipy
nvidia-cublas-cu12==12.8.3.14 \
nvidia-cublas-cu12==12.8.3.14 ; sys_platform == "linux" \
--hash=sha256:3f0e05e7293598cf61933258b73e66a160c27d59c4422670bf0b79348c04be44 \
--hash=sha256:93a4e0e386cc7f6e56c822531396de8170ed17068a1e18f987574895044cd8c3 \
--hash=sha256:9ae5eae500aead01fc4bdfc458209df638b1a3551557ce11a78eea9ece602ae9
@ -383,48 +383,48 @@ nvidia-cublas-cu12==12.8.3.14 \
# -r build/test-requirements.txt
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.8.57 \
nvidia-cuda-cupti-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:8e0b2eb847de260739bee4a3f66fac31378f4ff49538ff527a38a01a9a39f950 \
--hash=sha256:bbed719c52a476958a74cfc42f2b95a3fd6b3fd94eb40134acc4601feb4acac3 \
--hash=sha256:ff154211724fd824e758ce176b66007b558eea19c9a5135fc991827ee147e317
# via -r build/test-requirements.txt
nvidia-cuda-nvcc-cu12==12.8.61 \
nvidia-cuda-nvcc-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:171f605044ba17bc455d19cad289946c3dbea029a90c60dfa7b88e545bc8e329 \
--hash=sha256:28604ec42aaa09035b0fb7111432e5121bc385580b30c55d2acfb7d644b16548 \
--hash=sha256:4524739cfc080e9c9e53032912be8f020058e0a7186746d19acef3b6d916ea0b
# via -r build/test-requirements.txt
nvidia-cuda-runtime-cu12==12.8.57 \
nvidia-cuda-runtime-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:534ccebd967b6a44292678fa5da4f00666029cb2ed07a79515ea41ef31fe3ec7 \
--hash=sha256:75342e28567340b7428ce79a5d6bb6ca5ff9d07b69e7ce00d2c7b4dc23eff0be \
--hash=sha256:89be637e3ee967323865b85e0f147d75f9a5bd98360befa37481b02dd57af8f5
# via -r build/test-requirements.txt
nvidia-cudnn-cu12==9.7.1.26 \
nvidia-cudnn-cu12==9.7.1.26 ; sys_platform == "linux" \
--hash=sha256:6d011159a158f3cfc47bf851aea79e31bcff60d530b70ef70474c84cac484d07 \
--hash=sha256:7b805b9a4cf9f3da7c5f4ea4a9dff7baf62d1a612d6154a7e0d2ea51ed296241 \
--hash=sha256:848a61d40ef3b32bd4e1fadb599f0cf04a4b942fbe5fb3be572ad75f9b8c53ef
# via -r build/test-requirements.txt
nvidia-cufft-cu12==11.3.3.41 \
nvidia-cufft-cu12==11.3.3.41 ; sys_platform == "linux" \
--hash=sha256:68509dcd7e3306e69d0e2d8a6d21c8b25ed62e6df8aac192ce752f17677398b5 \
--hash=sha256:da650080ab79fcdf7a4b06aa1b460e99860646b176a43f6208099bdc17836b6a \
--hash=sha256:f9760612886786601d27a0993bb29ce1f757e6b8b173499d0ecfa850d31b50f8
# via -r build/test-requirements.txt
nvidia-cusolver-cu12==11.7.2.55 \
nvidia-cusolver-cu12==11.7.2.55 ; sys_platform == "linux" \
--hash=sha256:0fd9e98246f43c15bee5561147ad235dfdf2d037f5d07c9d41af3f7f72feb7cc \
--hash=sha256:4d1354102f1e922cee9db51920dba9e2559877cf6ff5ad03a00d853adafb191b \
--hash=sha256:a5a516c55da5c5aba98420d9bc9bcab18245f21ec87338cc1f930eb18dd411ac
# via -r build/test-requirements.txt
nvidia-cusparse-cu12==12.5.7.53 \
nvidia-cusparse-cu12==12.5.7.53 ; sys_platform == "linux" \
--hash=sha256:3c1b61eb8c85257ea07e9354606b26397612627fdcd327bfd91ccf6155e7c86d \
--hash=sha256:82c201d6781bacf6bb7c654f0446728d0fe596dfdd82ef4a04c204ce3e107441 \
--hash=sha256:d869c6146ca80f4305b62e02d924b4aaced936f8173e3cef536a67eed2a91af1
# via
# -r build/test-requirements.txt
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.25.1 \
nvidia-nccl-cu12==2.25.1 ; sys_platform == "linux" \
--hash=sha256:362aed5963fb9ea2ed2f264409baae30143498fd0e5c503aeaa1badd88cdc54a \
--hash=sha256:4ab428bc915785cc66e8c57cb34c7a64cf739c46702b8db748b6ad6cc7180cf8
# via -r build/test-requirements.txt
nvidia-nvjitlink-cu12==12.8.61 \
nvidia-nvjitlink-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:1166a964d25fdc0eae497574d38824305195a5283324a21ccb0ce0c802cbf41c \
--hash=sha256:45fd79f2ae20bd67e8bc411055939049873bfd8fac70ff13bd4865e0b9bdab17 \
--hash=sha256:9b80ecab31085dda3ce3b41d043be0ec739216c3fc633b8abe212d5a30026df0

20
build/requirements_lock_3_12.txt
View File

@ -375,7 +375,7 @@ numpy==2.0.0 ; python_version <= "3.12" \
# ml-dtypes
# opt-einsum
# scipy
nvidia-cublas-cu12==12.8.3.14 \
nvidia-cublas-cu12==12.8.3.14 ; sys_platform == "linux" \
--hash=sha256:3f0e05e7293598cf61933258b73e66a160c27d59c4422670bf0b79348c04be44 \
--hash=sha256:93a4e0e386cc7f6e56c822531396de8170ed17068a1e18f987574895044cd8c3 \
--hash=sha256:9ae5eae500aead01fc4bdfc458209df638b1a3551557ce11a78eea9ece602ae9
@ -383,48 +383,48 @@ nvidia-cublas-cu12==12.8.3.14 \
# -r build/test-requirements.txt
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.8.57 \
nvidia-cuda-cupti-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:8e0b2eb847de260739bee4a3f66fac31378f4ff49538ff527a38a01a9a39f950 \
--hash=sha256:bbed719c52a476958a74cfc42f2b95a3fd6b3fd94eb40134acc4601feb4acac3 \
--hash=sha256:ff154211724fd824e758ce176b66007b558eea19c9a5135fc991827ee147e317
# via -r build/test-requirements.txt
nvidia-cuda-nvcc-cu12==12.8.61 \
nvidia-cuda-nvcc-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:171f605044ba17bc455d19cad289946c3dbea029a90c60dfa7b88e545bc8e329 \
--hash=sha256:28604ec42aaa09035b0fb7111432e5121bc385580b30c55d2acfb7d644b16548 \
--hash=sha256:4524739cfc080e9c9e53032912be8f020058e0a7186746d19acef3b6d916ea0b
# via -r build/test-requirements.txt
nvidia-cuda-runtime-cu12==12.8.57 \
nvidia-cuda-runtime-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:534ccebd967b6a44292678fa5da4f00666029cb2ed07a79515ea41ef31fe3ec7 \
--hash=sha256:75342e28567340b7428ce79a5d6bb6ca5ff9d07b69e7ce00d2c7b4dc23eff0be \
--hash=sha256:89be637e3ee967323865b85e0f147d75f9a5bd98360befa37481b02dd57af8f5
# via -r build/test-requirements.txt
nvidia-cudnn-cu12==9.7.1.26 \
nvidia-cudnn-cu12==9.7.1.26 ; sys_platform == "linux" \
--hash=sha256:6d011159a158f3cfc47bf851aea79e31bcff60d530b70ef70474c84cac484d07 \
--hash=sha256:7b805b9a4cf9f3da7c5f4ea4a9dff7baf62d1a612d6154a7e0d2ea51ed296241 \
--hash=sha256:848a61d40ef3b32bd4e1fadb599f0cf04a4b942fbe5fb3be572ad75f9b8c53ef
# via -r build/test-requirements.txt
nvidia-cufft-cu12==11.3.3.41 \
nvidia-cufft-cu12==11.3.3.41 ; sys_platform == "linux" \
--hash=sha256:68509dcd7e3306e69d0e2d8a6d21c8b25ed62e6df8aac192ce752f17677398b5 \
--hash=sha256:da650080ab79fcdf7a4b06aa1b460e99860646b176a43f6208099bdc17836b6a \
--hash=sha256:f9760612886786601d27a0993bb29ce1f757e6b8b173499d0ecfa850d31b50f8
# via -r build/test-requirements.txt
nvidia-cusolver-cu12==11.7.2.55 \
nvidia-cusolver-cu12==11.7.2.55 ; sys_platform == "linux" \
--hash=sha256:0fd9e98246f43c15bee5561147ad235dfdf2d037f5d07c9d41af3f7f72feb7cc \
--hash=sha256:4d1354102f1e922cee9db51920dba9e2559877cf6ff5ad03a00d853adafb191b \
--hash=sha256:a5a516c55da5c5aba98420d9bc9bcab18245f21ec87338cc1f930eb18dd411ac
# via -r build/test-requirements.txt
nvidia-cusparse-cu12==12.5.7.53 \
nvidia-cusparse-cu12==12.5.7.53 ; sys_platform == "linux" \
--hash=sha256:3c1b61eb8c85257ea07e9354606b26397612627fdcd327bfd91ccf6155e7c86d \
--hash=sha256:82c201d6781bacf6bb7c654f0446728d0fe596dfdd82ef4a04c204ce3e107441 \
--hash=sha256:d869c6146ca80f4305b62e02d924b4aaced936f8173e3cef536a67eed2a91af1
# via
# -r build/test-requirements.txt
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.25.1 \
nvidia-nccl-cu12==2.25.1 ; sys_platform == "linux" \
--hash=sha256:362aed5963fb9ea2ed2f264409baae30143498fd0e5c503aeaa1badd88cdc54a \
--hash=sha256:4ab428bc915785cc66e8c57cb34c7a64cf739c46702b8db748b6ad6cc7180cf8
# via -r build/test-requirements.txt
nvidia-nvjitlink-cu12==12.8.61 \
nvidia-nvjitlink-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:1166a964d25fdc0eae497574d38824305195a5283324a21ccb0ce0c802cbf41c \
--hash=sha256:45fd79f2ae20bd67e8bc411055939049873bfd8fac70ff13bd4865e0b9bdab17 \
--hash=sha256:9b80ecab31085dda3ce3b41d043be0ec739216c3fc633b8abe212d5a30026df0

20
build/requirements_lock_3_13.txt
View File

@ -434,7 +434,7 @@ numpy==2.1.2 ; python_version >= "3.13" \
# matplotlib
# ml-dtypes
# scipy
nvidia-cublas-cu12==12.8.3.14 \
nvidia-cublas-cu12==12.8.3.14 ; sys_platform == "linux" \
--hash=sha256:3f0e05e7293598cf61933258b73e66a160c27d59c4422670bf0b79348c04be44 \
--hash=sha256:93a4e0e386cc7f6e56c822531396de8170ed17068a1e18f987574895044cd8c3 \
--hash=sha256:9ae5eae500aead01fc4bdfc458209df638b1a3551557ce11a78eea9ece602ae9
@ -442,48 +442,48 @@ nvidia-cublas-cu12==12.8.3.14 \
# -r build/test-requirements.txt
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.8.57 \
nvidia-cuda-cupti-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:8e0b2eb847de260739bee4a3f66fac31378f4ff49538ff527a38a01a9a39f950 \
--hash=sha256:bbed719c52a476958a74cfc42f2b95a3fd6b3fd94eb40134acc4601feb4acac3 \
--hash=sha256:ff154211724fd824e758ce176b66007b558eea19c9a5135fc991827ee147e317
# via -r build/test-requirements.txt
nvidia-cuda-nvcc-cu12==12.8.61 \
nvidia-cuda-nvcc-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:171f605044ba17bc455d19cad289946c3dbea029a90c60dfa7b88e545bc8e329 \
--hash=sha256:28604ec42aaa09035b0fb7111432e5121bc385580b30c55d2acfb7d644b16548 \
--hash=sha256:4524739cfc080e9c9e53032912be8f020058e0a7186746d19acef3b6d916ea0b
# via -r build/test-requirements.txt
nvidia-cuda-runtime-cu12==12.8.57 \
nvidia-cuda-runtime-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:534ccebd967b6a44292678fa5da4f00666029cb2ed07a79515ea41ef31fe3ec7 \
--hash=sha256:75342e28567340b7428ce79a5d6bb6ca5ff9d07b69e7ce00d2c7b4dc23eff0be \
--hash=sha256:89be637e3ee967323865b85e0f147d75f9a5bd98360befa37481b02dd57af8f5
# via -r build/test-requirements.txt
nvidia-cudnn-cu12==9.7.1.26 \
nvidia-cudnn-cu12==9.7.1.26 ; sys_platform == "linux" \
--hash=sha256:6d011159a158f3cfc47bf851aea79e31bcff60d530b70ef70474c84cac484d07 \
--hash=sha256:7b805b9a4cf9f3da7c5f4ea4a9dff7baf62d1a612d6154a7e0d2ea51ed296241 \
--hash=sha256:848a61d40ef3b32bd4e1fadb599f0cf04a4b942fbe5fb3be572ad75f9b8c53ef
# via -r build/test-requirements.txt
nvidia-cufft-cu12==11.3.3.41 \
nvidia-cufft-cu12==11.3.3.41 ; sys_platform == "linux" \
--hash=sha256:68509dcd7e3306e69d0e2d8a6d21c8b25ed62e6df8aac192ce752f17677398b5 \
--hash=sha256:da650080ab79fcdf7a4b06aa1b460e99860646b176a43f6208099bdc17836b6a \
--hash=sha256:f9760612886786601d27a0993bb29ce1f757e6b8b173499d0ecfa850d31b50f8
# via -r build/test-requirements.txt
nvidia-cusolver-cu12==11.7.2.55 \
nvidia-cusolver-cu12==11.7.2.55 ; sys_platform == "linux" \
--hash=sha256:0fd9e98246f43c15bee5561147ad235dfdf2d037f5d07c9d41af3f7f72feb7cc \
--hash=sha256:4d1354102f1e922cee9db51920dba9e2559877cf6ff5ad03a00d853adafb191b \
--hash=sha256:a5a516c55da5c5aba98420d9bc9bcab18245f21ec87338cc1f930eb18dd411ac
# via -r build/test-requirements.txt
nvidia-cusparse-cu12==12.5.7.53 \
nvidia-cusparse-cu12==12.5.7.53 ; sys_platform == "linux" \
--hash=sha256:3c1b61eb8c85257ea07e9354606b26397612627fdcd327bfd91ccf6155e7c86d \
--hash=sha256:82c201d6781bacf6bb7c654f0446728d0fe596dfdd82ef4a04c204ce3e107441 \
--hash=sha256:d869c6146ca80f4305b62e02d924b4aaced936f8173e3cef536a67eed2a91af1
# via
# -r build/test-requirements.txt
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.25.1 \
nvidia-nccl-cu12==2.25.1 ; sys_platform == "linux" \
--hash=sha256:362aed5963fb9ea2ed2f264409baae30143498fd0e5c503aeaa1badd88cdc54a \
--hash=sha256:4ab428bc915785cc66e8c57cb34c7a64cf739c46702b8db748b6ad6cc7180cf8
# via -r build/test-requirements.txt
nvidia-nvjitlink-cu12==12.8.61 \
nvidia-nvjitlink-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:1166a964d25fdc0eae497574d38824305195a5283324a21ccb0ce0c802cbf41c \
--hash=sha256:45fd79f2ae20bd67e8bc411055939049873bfd8fac70ff13bd4865e0b9bdab17 \
--hash=sha256:9b80ecab31085dda3ce3b41d043be0ec739216c3fc633b8abe212d5a30026df0

20
build/requirements_lock_3_13_ft.txt
View File

@ -390,7 +390,7 @@ numpy==2.2.1 ; python_version >= "3.13" \
# matplotlib
# ml-dtypes
# scipy
nvidia-cublas-cu12==12.8.3.14 \
nvidia-cublas-cu12==12.8.3.14 ; sys_platform == "linux" \
--hash=sha256:3f0e05e7293598cf61933258b73e66a160c27d59c4422670bf0b79348c04be44 \
--hash=sha256:93a4e0e386cc7f6e56c822531396de8170ed17068a1e18f987574895044cd8c3 \
--hash=sha256:9ae5eae500aead01fc4bdfc458209df638b1a3551557ce11a78eea9ece602ae9
@ -398,48 +398,48 @@ nvidia-cublas-cu12==12.8.3.14 \
# -r build/test-requirements.txt
# nvidia-cudnn-cu12
# nvidia-cusolver-cu12
nvidia-cuda-cupti-cu12==12.8.57 \
nvidia-cuda-cupti-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:8e0b2eb847de260739bee4a3f66fac31378f4ff49538ff527a38a01a9a39f950 \
--hash=sha256:bbed719c52a476958a74cfc42f2b95a3fd6b3fd94eb40134acc4601feb4acac3 \
--hash=sha256:ff154211724fd824e758ce176b66007b558eea19c9a5135fc991827ee147e317
# via -r build/test-requirements.txt
nvidia-cuda-nvcc-cu12==12.8.61 \
nvidia-cuda-nvcc-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:171f605044ba17bc455d19cad289946c3dbea029a90c60dfa7b88e545bc8e329 \
--hash=sha256:28604ec42aaa09035b0fb7111432e5121bc385580b30c55d2acfb7d644b16548 \
--hash=sha256:4524739cfc080e9c9e53032912be8f020058e0a7186746d19acef3b6d916ea0b
# via -r build/test-requirements.txt
nvidia-cuda-runtime-cu12==12.8.57 \
nvidia-cuda-runtime-cu12==12.8.57 ; sys_platform == "linux" \
--hash=sha256:534ccebd967b6a44292678fa5da4f00666029cb2ed07a79515ea41ef31fe3ec7 \
--hash=sha256:75342e28567340b7428ce79a5d6bb6ca5ff9d07b69e7ce00d2c7b4dc23eff0be \
--hash=sha256:89be637e3ee967323865b85e0f147d75f9a5bd98360befa37481b02dd57af8f5
# via -r build/test-requirements.txt
nvidia-cudnn-cu12==9.7.1.26 \
nvidia-cudnn-cu12==9.7.1.26 ; sys_platform == "linux" \
--hash=sha256:6d011159a158f3cfc47bf851aea79e31bcff60d530b70ef70474c84cac484d07 \
--hash=sha256:7b805b9a4cf9f3da7c5f4ea4a9dff7baf62d1a612d6154a7e0d2ea51ed296241 \
--hash=sha256:848a61d40ef3b32bd4e1fadb599f0cf04a4b942fbe5fb3be572ad75f9b8c53ef
# via -r build/test-requirements.txt
nvidia-cufft-cu12==11.3.3.41 \
nvidia-cufft-cu12==11.3.3.41 ; sys_platform == "linux" \
--hash=sha256:68509dcd7e3306e69d0e2d8a6d21c8b25ed62e6df8aac192ce752f17677398b5 \
--hash=sha256:da650080ab79fcdf7a4b06aa1b460e99860646b176a43f6208099bdc17836b6a \
--hash=sha256:f9760612886786601d27a0993bb29ce1f757e6b8b173499d0ecfa850d31b50f8
# via -r build/test-requirements.txt
nvidia-cusolver-cu12==11.7.2.55 \
nvidia-cusolver-cu12==11.7.2.55 ; sys_platform == "linux" \
--hash=sha256:0fd9e98246f43c15bee5561147ad235dfdf2d037f5d07c9d41af3f7f72feb7cc \
--hash=sha256:4d1354102f1e922cee9db51920dba9e2559877cf6ff5ad03a00d853adafb191b \
--hash=sha256:a5a516c55da5c5aba98420d9bc9bcab18245f21ec87338cc1f930eb18dd411ac
# via -r build/test-requirements.txt
nvidia-cusparse-cu12==12.5.7.53 \
nvidia-cusparse-cu12==12.5.7.53 ; sys_platform == "linux" \
--hash=sha256:3c1b61eb8c85257ea07e9354606b26397612627fdcd327bfd91ccf6155e7c86d \
--hash=sha256:82c201d6781bacf6bb7c654f0446728d0fe596dfdd82ef4a04c204ce3e107441 \
--hash=sha256:d869c6146ca80f4305b62e02d924b4aaced936f8173e3cef536a67eed2a91af1
# via
# -r build/test-requirements.txt
# nvidia-cusolver-cu12
nvidia-nccl-cu12==2.25.1 \
nvidia-nccl-cu12==2.25.1 ; sys_platform == "linux" \
--hash=sha256:362aed5963fb9ea2ed2f264409baae30143498fd0e5c503aeaa1badd88cdc54a \
--hash=sha256:4ab428bc915785cc66e8c57cb34c7a64cf739c46702b8db748b6ad6cc7180cf8
# via -r build/test-requirements.txt
nvidia-nvjitlink-cu12==12.8.61 \
nvidia-nvjitlink-cu12==12.8.61 ; sys_platform == "linux" \
--hash=sha256:1166a964d25fdc0eae497574d38824305195a5283324a21ccb0ce0c802cbf41c \
--hash=sha256:45fd79f2ae20bd67e8bc411055939049873bfd8fac70ff13bd4865e0b9bdab17 \
--hash=sha256:9b80ecab31085dda3ce3b41d043be0ec739216c3fc633b8abe212d5a30026df0