diff options
author | Adam J. Stewart <ajstewart426@gmail.com> | 2024-05-01 13:30:46 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-05-01 13:30:46 +0200 |
commit | 05b04cd4c3e213c4aef99aac6a261a38b1439b5b (patch) | |
tree | 6f342eaa0505c07a739564a7aee1484046ec3431 | |
parent | be48f762a9b2482641c075f729c11ac45f427e56 (diff) | |
download | spack-05b04cd4c3e213c4aef99aac6a261a38b1439b5b.tar.gz spack-05b04cd4c3e213c4aef99aac6a261a38b1439b5b.tar.bz2 spack-05b04cd4c3e213c4aef99aac6a261a38b1439b5b.tar.xz spack-05b04cd4c3e213c4aef99aac6a261a38b1439b5b.zip |
Update PyTorch ecosystem (#43823)
* Update PyTorch ecosystem
* Don't expose protobuf namespace to other packages, breaks torchtext build
* Revert "Don't expose protobuf namespace to other packages, breaks torchtext build"
This reverts commit 50a4b08f655cac53dd75c3c7cc01817d3f5f92f7.
7 files changed, 135 insertions, 109 deletions
diff --git a/var/spack/repos/builtin/packages/cpuinfo/package.py b/var/spack/repos/builtin/packages/cpuinfo/package.py index 375d48208a..4d1cb8fc22 100644 --- a/var/spack/repos/builtin/packages/cpuinfo/package.py +++ b/var/spack/repos/builtin/packages/cpuinfo/package.py @@ -17,7 +17,8 @@ class Cpuinfo(CMakePackage): license("BSD-2-Clause") version("main", branch="main") - version("2023-01-13", commit="6481e8bef08f606ddd627e4d3be89f64d62e1b8a") # py-torch@2.1: + version("2023-11-04", commit="d6860c477c99f1fce9e28eb206891af3c0e1a1d7") # py-torch@2.3: + version("2023-01-13", commit="6481e8bef08f606ddd627e4d3be89f64d62e1b8a") # py-torch@2.1:2.2 version("2022-08-19", commit="8ec7bd91ad0470e61cf38f618cc1f270dede599c") # py-torch@1.13:2.0 version("2020-12-17", commit="5916273f79a21551890fd3d56fc5375a78d1598d") # py-torch@1.8:1.12 version("2020-06-11", commit="63b254577ed77a8004a9be6ac707f3dccc4e1fd9") # py-torch@1.6:1.7 diff --git a/var/spack/repos/builtin/packages/gloo/package.py b/var/spack/repos/builtin/packages/gloo/package.py index dcf9a7e3ce..c01d99a877 100644 --- a/var/spack/repos/builtin/packages/gloo/package.py +++ b/var/spack/repos/builtin/packages/gloo/package.py @@ -15,7 +15,8 @@ class Gloo(CMakePackage, CudaPackage): license("BSD-3-Clause") version("master", branch="master") - version("2023-05-19", commit="597accfd79f5b0f9d57b228dec088ca996686475") # py-torch@2.1: + version("2023-12-03", commit="5354032ea08eadd7fc4456477f7f7c6308818509") # py-torch@2.3: + version("2023-05-19", commit="597accfd79f5b0f9d57b228dec088ca996686475") # py-torch@2.1:2.2 version("2023-01-17", commit="10909297fedab0a680799211a299203e53515032") # py-torch@2.0 version("2022-05-18", commit="5b143513263133af2b95547e97c07cebeb72bf72") # py-torch@1.13 version("2021-05-21", commit="c22a5cfba94edf8ea4f53a174d38aa0c629d070f") # py-torch@1.10:1.12 diff --git a/var/spack/repos/builtin/packages/py-protobuf/package.py b/var/spack/repos/builtin/packages/py-protobuf/package.py index b5f0171d01..f3ed5a0971 100644 --- a/var/spack/repos/builtin/packages/py-protobuf/package.py +++ b/var/spack/repos/builtin/packages/py-protobuf/package.py @@ -42,6 +42,7 @@ class PyProtobuf(PythonPackage): version("3.15.6", sha256="2b974519a2ae83aa1e31cff9018c70bbe0e303a46a598f982943c49ae1d4fcd3") version("3.15.5", sha256="be8a929c6178bb6cbe9e2c858be62fa08966a39ae758a8493a88f0ed1efb6097") version("3.15.1", sha256="824dbae3390fcc3ea1bf96748e6da951a601802894cf7e1465e72b4732538cab") + version("3.13.0", sha256="6a82e0c8bb2bf58f606040cc5814e07715b2094caeba281e2e7d0b0e2e397db5") version("3.12.2", sha256="49ef8ab4c27812a89a76fa894fe7a08f42f2147078392c0dee51d4a444ef6df5") version("3.11.2", sha256="3d7a7d8d20b4e7a8f63f62de2d192cfd8b7a53c56caba7ece95367ca2b80c574") version("3.11.1", sha256="aecdf12ef6dc7fd91713a6da93a86c2f2a8fe54840a3b1670853a2b7402e77c9") diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index f5340fcebd..2677105c64 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -26,6 +26,7 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage): license("BSD-3-Clause") version("main", branch="main") + version("2.3.0", tag="v2.3.0", commit="97ff6cfd9c86c5c09d7ce775ab64ec5c99230f5d") version("2.2.2", tag="v2.2.2", commit="39901f229520a5256505ec24782f716ee7ddc843") version("2.2.1", tag="v2.2.1", commit="6c8c5ad5eaf47a62fafbb4a2747198cbffbf1ff0") version("2.2.0", tag="v2.2.0", commit="8ac9b20d4b090c213799e81acf48a55ea8d437d6") @@ -137,89 +138,122 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage): # Required dependencies # Based on PyPI wheel availability - depends_on("python@3.8:3.12", when="@2.2:", type=("build", "link", "run")) - depends_on("python@3.8:3.11", when="@2.0:2.1", type=("build", "link", "run")) - depends_on("python@:3.10", when="@1.11:1", type=("build", "link", "run")) - depends_on("python@:3.9", when="@1.7.1:1.10", type=("build", "link", "run")) - depends_on("python@:3.8", when="@1.4:1.7.0", type=("build", "link", "run")) + with default_args(type=("build", "link", "run")): + depends_on("python@3.8:3.12", when="@2.2:") + depends_on("python@3.8:3.11", when="@2.0:2.1") + depends_on("python@:3.10", when="@1.11:1") + depends_on("python@:3.9", when="@1.7.1:1.10") + depends_on("python@:3.8", when="@1.4:1.7.0") # CMakelists.txt - depends_on("cmake@3.18:", when="@2:", type="build") - depends_on("cmake@3.13:", when="@1.11:", type="build") - depends_on("cmake@3.10:", when="@1.10:", type="build") - depends_on("cmake@3.5:", type="build") - - # pyproject.toml - depends_on("py-setuptools", type=("build", "run")) - depends_on("py-astunparse", when="@1.13:", type=("build", "run")) - depends_on("py-numpy@1.16.6:", type=("build", "run")) - depends_on("ninja@1.5:", when="@1.1:", type="build") - depends_on("py-pyyaml", type=("build", "run")) - depends_on("py-requests", when="@1.13:", type=("build", "run")) - depends_on("py-cffi", when="@:1", type=("build", "run")) - depends_on("py-future", when="@1.5:1", type=("build", "run")) - depends_on("py-six", when="@1.13:1", type=("build", "run")) - - # setup.py - depends_on("py-filelock", when="@2:", type=("build", "run")) - depends_on("py-typing-extensions@4.8:", when="@2.2:", type=("build", "run")) - depends_on("py-typing-extensions@3.6.2.1:", when="@1.7:", type=("build", "run")) - depends_on("py-sympy", when="@2:", type=("build", "run")) - depends_on("py-networkx", when="@2:", type=("build", "run")) - depends_on("py-jinja2", when="@2:", type=("build", "run")) - depends_on("py-fsspec", when="@2.1:", type=("build", "run")) + with default_args(type="build"): + depends_on("cmake@3.18:", when="@2:") + depends_on("cmake@3.13:", when="@1.11:") + depends_on("cmake@3.10:", when="@1.10:") + depends_on("cmake@3.5:") + depends_on("ninja@1.5:") + + with default_args(type=("build", "run")): + # setup.py + depends_on("py-filelock", when="@2:") + depends_on("py-typing-extensions@4.8:", when="@2.2:") + depends_on("py-typing-extensions@3.6.2.1:", when="@1.7:") + depends_on("py-sympy", when="@2:") + depends_on("py-networkx", when="@2:") + depends_on("py-jinja2", when="@2:") + depends_on("py-fsspec", when="@2.1:") + depends_on("mkl@2021.1.1:2021.4.0", when="@2.3: platform=windows") + + # pyproject.toml + depends_on("py-setuptools") + depends_on("py-astunparse", when="@1.13:") + depends_on("py-numpy@1.16.6:") + depends_on("py-pyyaml") + depends_on("py-requests", when="@1.13:") # Undocumented dependencies depends_on("py-tqdm", type="run") depends_on("blas") depends_on("lapack") - # third_party - depends_on("py-pybind11@2.11.0", when="@2.1:", type=("build", "link", "run")) - depends_on("py-pybind11@2.10.1", when="@2.0", type=("build", "link", "run")) - depends_on("py-pybind11@2.10.0", when="@1.13:1", type=("build", "link", "run")) - depends_on("py-pybind11@2.6.2", when="@1.8:1.12", type=("build", "link", "run")) - depends_on("py-pybind11@2.3.0", when="@:1.7", type=("build", "link", "run")) - with when("~custom-protobuf"): - depends_on("py-protobuf@3.12.2:", when="@1.10:", type=("build", "run")) - depends_on("py-protobuf@:3.14", when="@:1.9", type=("build", "run")) - depends_on("protobuf@3.12.2:", when="@1.10:") - depends_on("protobuf@:3.14", when="@:1.9") - # https://github.com/protocolbuffers/protobuf/issues/10051 - # https://github.com/pytorch/pytorch/issues/78362 - depends_on("py-protobuf@:3", type=("build", "run")) - depends_on("protobuf@:3") - depends_on("eigen") - depends_on("cpuinfo@2023-01-13", when="@2.1:") + # Third party dependencies + depends_on("fp16@2020-05-14", when="@1.6:") + depends_on("fxdiv@2020-04-17", when="@1.6:") + # https://github.com/pytorch/pytorch/issues/60332 + # depends_on("xnnpack@2024-02-29", when="@2.3:+xnnpack") + # depends_on("xnnpack@2022-12-21", when="@2.0:2.2+xnnpack") + # depends_on("xnnpack@2022-02-16", when="@1.12:1+xnnpack") + # depends_on("xnnpack@2021-06-21", when="@1.10:1.11+xnnpack") + # depends_on("xnnpack@2021-02-22", when="@1.8:1.9+xnnpack") + # depends_on("xnnpack@2020-03-23", when="@1.6:1.7+xnnpack") + depends_on("benchmark", when="@1.6:+test") + depends_on("cpuinfo@2023-11-04", when="@2.3:") + depends_on("cpuinfo@2023-01-13", when="@2.1:2.2") depends_on("cpuinfo@2022-08-19", when="@1.13:2.0") depends_on("cpuinfo@2020-12-17", when="@1.8:1.12") depends_on("cpuinfo@2020-06-11", when="@1.6:1.7") - depends_on("sleef@3.5.1_2020-12-22", when="@1.8:") - depends_on("sleef@3.4.0_2019-07-30", when="@1.6:1.7") - depends_on("fp16@2020-05-14", when="@1.6:") + depends_on("eigen") + depends_on("gloo@2023-12-03", when="@2.3:+gloo") + depends_on("gloo@2023-05-19", when="@2.1:2.2+gloo") + depends_on("gloo@2023-01-17", when="@2.0+gloo") + depends_on("gloo@2022-05-18", when="@1.13:1+gloo") + depends_on("gloo@2021-05-21", when="@1.10:1.12+gloo") + depends_on("gloo@2021-05-04", when="@1.9+gloo") + depends_on("gloo@2020-09-18", when="@1.7:1.8+gloo") + depends_on("gloo@2020-03-17", when="@1.6+gloo") + depends_on("gloo+cuda", when="@1.6:+gloo+cuda") + depends_on("nccl", when="+nccl+cuda") + # https://github.com/pytorch/pytorch/issues/60331 + # depends_on("onnx@1.16.0", when="@2.3:+onnx_ml") + # depends_on("onnx@1.15.0", when="@2.2+onnx_ml") + # depends_on("onnx@1.14.1", when="@2.1+onnx_ml") + # depends_on("onnx@1.13.1", when="@2.0+onnx_ml") + # depends_on("onnx@1.12.0", when="@1.13:1+onnx_ml") + # depends_on("onnx@1.11.0", when="@1.12+onnx_ml") + # depends_on("onnx@1.10.1_2021-10-08", when="@1.11+onnx_ml") + # depends_on("onnx@1.10.1", when="@1.10+onnx_ml") + # depends_on("onnx@1.8.0_2020-11-03", when="@1.8:1.9+onnx_ml") + # depends_on("onnx@1.7.0_2020-05-31", when="@1.6:1.7+onnx_ml") + with when("~custom-protobuf"): + depends_on("protobuf@3.13.0", when="@1.10:") + depends_on("protobuf@3.11.4", when="@1.6:1.9") + depends_on("protobuf@3.6.1", when="@1.1:1.5") + depends_on("protobuf@3.5.0", when="@1.0") + with default_args(type=("build", "run")): + depends_on("py-protobuf@3.13", when="@1.10:") + depends_on("py-protobuf@3.11", when="@1.6:1.9") + depends_on("py-protobuf@3.6", when="@1.1:1.5") + depends_on("py-protobuf@3.5", when="@1.0") + depends_on("psimd@2020-05-17", when="@1.6:") depends_on("pthreadpool@2023-08-29", when="@2.2:") depends_on("pthreadpool@2021-04-13", when="@1.9:2.1") depends_on("pthreadpool@2020-10-05", when="@1.8") depends_on("pthreadpool@2020-06-15", when="@1.6:1.7") - depends_on("psimd@2020-05-17", when="@1.6:") - depends_on("fxdiv@2020-04-17", when="@1.6:") - depends_on("benchmark", when="@1.6:+test") + with default_args(type=("build", "link", "run")): + depends_on("py-pybind11@2.12.0", when="@2.3:") + depends_on("py-pybind11@2.11.0", when="@2.1:2.2") + depends_on("py-pybind11@2.10.1", when="@2.0") + depends_on("py-pybind11@2.10.0", when="@1.13:1") + depends_on("py-pybind11@2.6.2", when="@1.8:1.12") + depends_on("py-pybind11@2.3.0", when="@:1.7") + depends_on("sleef@3.5.1_2020-12-22", when="@1.8:") + depends_on("sleef@3.4.0_2019-07-30", when="@1.6:1.7") # Optional dependencies - # cmake/public/cuda.cmake - depends_on("cuda@11:", when="@2:+cuda", type=("build", "link", "run")) - depends_on("cuda@10.2:", when="@1.11:1+cuda", type=("build", "link", "run")) - # https://discuss.pytorch.org/t/compiling-1-10-1-from-source-with-gcc-11-and-cuda-11-5/140971 - depends_on("cuda@10.2:11.4", when="@1.10+cuda", type=("build", "link", "run")) - depends_on("cuda@9.2:11.4", when="@1.6:1.9+cuda", type=("build", "link", "run")) - depends_on("cuda@9:11.4", when="@:1.5+cuda", type=("build", "link", "run")) + with default_args(type=("build", "link", "run")): + # cmake/public/cuda.cmake + depends_on("cuda@11:", when="@2:+cuda") + depends_on("cuda@10.2:", when="@1.11:1+cuda") + # https://discuss.pytorch.org/t/compiling-1-10-1-from-source-with-gcc-11-and-cuda-11-5/140971 + depends_on("cuda@10.2:11.4", when="@1.10+cuda") + depends_on("cuda@9.2:11.4", when="@1.6:1.9+cuda") + depends_on("cuda@9:11.4", when="@:1.5+cuda") # https://github.com/pytorch/pytorch#prerequisites depends_on("cudnn@8.5:", when="@2.3:+cudnn") depends_on("cudnn@7:", when="@1.6:+cudnn") depends_on("cudnn@7", when="@:1.5+cudnn") depends_on("magma+cuda", when="+magma+cuda") depends_on("magma+rocm", when="+magma+rocm") - depends_on("nccl", when="+nccl+cuda") depends_on("numactl", when="+numa") depends_on("llvm-openmp", when="%apple-clang +openmp") depends_on("valgrind", when="+valgrind") @@ -238,37 +272,20 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage): depends_on("rocblas") depends_on("miopen-hip") depends_on("rocminfo") - # https://github.com/pytorch/pytorch/issues/60332 - # depends_on("xnnpack@2022-12-21", when="@2:+xnnpack") - # depends_on("xnnpack@2022-02-16", when="@1.12:1+xnnpack") - # depends_on("xnnpack@2021-06-21", when="@1.10:1.11+xnnpack") - # depends_on("xnnpack@2021-02-22", when="@1.8:1.9+xnnpack") - # depends_on("xnnpack@2020-03-23", when="@1.6:1.7+xnnpack") depends_on("mpi", when="+mpi") - depends_on("gloo@2023-05-19", when="@2.1:+gloo") - depends_on("gloo@2023-01-17", when="@2.0+gloo") - depends_on("gloo@2022-05-18", when="@1.13:1+gloo") - depends_on("gloo@2021-05-21", when="@1.10:1.12+gloo") - depends_on("gloo@2021-05-04", when="@1.9+gloo") - depends_on("gloo@2020-09-18", when="@1.7:1.8+gloo") - depends_on("gloo@2020-03-17", when="@1.6+gloo") - depends_on("gloo+cuda", when="@1.6:+gloo+cuda") - # https://github.com/pytorch/pytorch/issues/60331 - # depends_on("onnx@1.15.0", when="@2.2:+onnx_ml") - # depends_on("onnx@1.14.1", when="@2.1+onnx_ml") - # depends_on("onnx@1.13.1", when="@2.0+onnx_ml") - # depends_on("onnx@1.12.0", when="@1.13:1+onnx_ml") - # depends_on("onnx@1.11.0", when="@1.12+onnx_ml") - # depends_on("onnx@1.10.1_2021-10-08", when="@1.11+onnx_ml") - # depends_on("onnx@1.10.1", when="@1.10+onnx_ml") - # depends_on("onnx@1.8.0_2020-11-03", when="@1.8:1.9+onnx_ml") - # depends_on("onnx@1.7.0_2020-05-31", when="@1.6:1.7+onnx_ml") depends_on("mkl", when="+mkldnn") # Test dependencies - depends_on("py-hypothesis", type="test") - depends_on("py-six", type="test") - depends_on("py-psutil", type="test") + with default_args(type="test"): + depends_on("py-hypothesis") + depends_on("py-six") + depends_on("py-psutil") + + # Historical dependencies + with default_args(type=("build", "run")): + depends_on("py-cffi", when="@:1") + depends_on("py-future", when="@1.5:1") + depends_on("py-six", when="@1.13:1") conflicts("%gcc@:9.3", when="@2.2:", msg="C++17 support required") @@ -620,30 +637,30 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage): env.set("WITH_BLAS", "generic") # Don't use vendored third-party libraries when possible + # env.set("USE_SYSTEM_LIBS", "ON") + env.set("USE_SYSTEM_BENCHMARK", "ON") + env.set("USE_SYSTEM_CPUINFO", "ON") + env.set("USE_SYSTEM_EIGEN_INSTALL", "ON") + env.set("USE_SYSTEM_FP16", "ON") + env.set("USE_SYSTEM_FXDIV", "ON") + env.set("USE_SYSTEM_GLOO", "ON") + env.set("USE_SYSTEM_NCCL", "ON") + # https://github.com/pytorch/pytorch/issues/60331 + # env.set("USE_SYSTEM_ONNX", "ON") + env.set("USE_SYSTEM_PSIMD", "ON") + env.set("USE_SYSTEM_PTHREADPOOL", "ON") + env.set("USE_SYSTEM_PYBIND11", "ON") + env.set("USE_SYSTEM_SLEEF", "ON") + # env.set("USE_SYSTEM_TBB", "ON") + # env.set("USE_SYSTEM_UCC", "ON") + # https://github.com/pytorch/pytorch/issues/60332 + # env.set("USE_SYSTEM_XNNPACK", "ON") + # env.set("USE_SYSTEM_ZSTD", "ON") + if self.spec.satisfies("+custom-protobuf"): env.set("BUILD_CUSTOM_PROTOBUF", "ON") else: env.set("BUILD_CUSTOM_PROTOBUF", "OFF") - env.set("USE_SYSTEM_NCCL", "ON") - env.set("USE_SYSTEM_EIGEN_INSTALL", "ON") - env.set("pybind11_DIR", self.spec["py-pybind11"].prefix) - env.set("pybind11_INCLUDE_DIR", self.spec["py-pybind11"].prefix.include) - if self.spec.satisfies("@1.10:"): - env.set("USE_SYSTEM_PYBIND11", "ON") - if self.spec.satisfies("@1.6:"): - # env.set("USE_SYSTEM_LIBS", "ON") - env.set("USE_SYSTEM_CPUINFO", "ON") - env.set("USE_SYSTEM_SLEEF", "ON") - env.set("USE_SYSTEM_GLOO", "ON") - env.set("USE_SYSTEM_FP16", "ON") - env.set("USE_SYSTEM_PTHREADPOOL", "ON") - env.set("USE_SYSTEM_PSIMD", "ON") - env.set("USE_SYSTEM_FXDIV", "ON") - env.set("USE_SYSTEM_BENCHMARK", "ON") - # https://github.com/pytorch/pytorch/issues/60331 - # env.set("USE_SYSTEM_ONNX", "ON") - # https://github.com/pytorch/pytorch/issues/60332 - # env.set("USE_SYSTEM_XNNPACK", "ON") # https://github.com/pytorch/pytorch/issues/111086 if self.spec.satisfies("%apple-clang@15:"): diff --git a/var/spack/repos/builtin/packages/py-torchaudio/package.py b/var/spack/repos/builtin/packages/py-torchaudio/package.py index 5cc3b302b4..e877afc10c 100644 --- a/var/spack/repos/builtin/packages/py-torchaudio/package.py +++ b/var/spack/repos/builtin/packages/py-torchaudio/package.py @@ -18,6 +18,7 @@ class PyTorchaudio(PythonPackage): maintainers("adamjstewart") version("main", branch="main") + version("2.3.0", tag="v2.3.0", commit="952ea7457bcc3ed0669e7741ff23015c426d6322") version("2.2.2", tag="v2.2.2", commit="cefdb369247668e1dba74de503d4d996124b6b11") version("2.2.1", tag="v2.2.1", commit="06ea59c97d56868e487490702d01b3cf59103b9c") version("2.2.0", tag="v2.2.0", commit="08901ade5d17d3e3cf6fc039cbd601cbd2853686") @@ -55,6 +56,7 @@ class PyTorchaudio(PythonPackage): depends_on("python@:3.8", when="@:0.7.0") depends_on("py-torch@main", when="@main") + depends_on("py-torch@2.3.0", when="@2.3.0") depends_on("py-torch@2.2.2", when="@2.2.2") depends_on("py-torch@2.2.1", when="@2.2.1") depends_on("py-torch@2.2.0", when="@2.2.0") diff --git a/var/spack/repos/builtin/packages/py-torchtext/package.py b/var/spack/repos/builtin/packages/py-torchtext/package.py index f18edc3dea..5369401bba 100644 --- a/var/spack/repos/builtin/packages/py-torchtext/package.py +++ b/var/spack/repos/builtin/packages/py-torchtext/package.py @@ -19,6 +19,7 @@ class PyTorchtext(PythonPackage): license("BSD-3-Clause") version("main", branch="main") + version("0.18.0", tag="v0.18.0", commit="9bed85d7a7ae13cf8c28598a88d8e461fe1afcb4") version("0.17.2", tag="v0.17.2", commit="5c34b86897a93ad6564543130661c260a760b356") version("0.17.1", tag="v0.17.1", commit="15e55dd73b5de8c179c7bd5cc9e2cc813830fb34") version("0.17.0", tag="v0.17.0", commit="400da5c61bab4abaaeaeca91744ca031ad9b2edf") @@ -51,6 +52,7 @@ class PyTorchtext(PythonPackage): # https://github.com/pytorch/text#installation depends_on("py-torch@main", when="@main") + depends_on("py-torch@2.3.0", when="@0.18.0") depends_on("py-torch@2.2.2", when="@0.17.2") depends_on("py-torch@2.2.1", when="@0.17.1") depends_on("py-torch@2.2.0", when="@0.17.0") diff --git a/var/spack/repos/builtin/packages/py-torchvision/package.py b/var/spack/repos/builtin/packages/py-torchvision/package.py index a9a97202d5..08e6182300 100644 --- a/var/spack/repos/builtin/packages/py-torchvision/package.py +++ b/var/spack/repos/builtin/packages/py-torchvision/package.py @@ -19,6 +19,7 @@ class PyTorchvision(PythonPackage): license("BSD-3-Clause") version("main", branch="main") + version("0.18.0", sha256="3e61cbac33986a862a59cd733fd65da8b2c2a6160a66556cfa0e850f62fd43c7") version("0.17.2", sha256="0f9304acd77aafb7cfaf3fd5e318b2986ecc73547394b971d710eacd59f3e78e") version("0.17.1", sha256="a01c7bce4098c41b62cd3a08d87569113e25d12994b1370f0fd5f531952b6cef") version("0.17.0", sha256="55e395d5c7d9bf7658c82ac633cac2224aa168e1bfe8bb5b2b2a296c792a3500") @@ -66,6 +67,7 @@ class PyTorchvision(PythonPackage): # https://github.com/pytorch/vision#installation depends_on("py-torch@main", when="@main") + depends_on("py-torch@2.3.0", when="@0.18.0") depends_on("py-torch@2.2.2", when="@0.17.2") depends_on("py-torch@2.2.1", when="@0.17.1") depends_on("py-torch@2.2.0", when="@0.17.0") |