zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-torch: rename master to main (#38858)

Browse Source
This commit is contained in:
Adam J. Stewart
2023-07-12 04:20:47 -05:00
committed by GitHub
parent fb223f034b
commit d2ed8c5226
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
var/spack/repos/builtin/packages/py-torch/package.py
View File

@@ -23,7 +23,8 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage):
# core libraries to ensure that the package was successfully installed. # core libraries to ensure that the package was successfully installed.
import_modules = ["torch", "torch.autograd", "torch.nn", "torch.utils"] import_modules = ["torch", "torch.autograd", "torch.nn", "torch.utils"]
version("master", branch="master", submodules=True) version("main", branch="main", submodules=True)
version("master", branch="main", submodules=True, deprecated=True)
version("2.0.1", tag="v2.0.1", submodules=True) version("2.0.1", tag="v2.0.1", submodules=True)
version("2.0.0", tag="v2.0.0", submodules=True) version("2.0.0", tag="v2.0.0", submodules=True)
version("1.13.1", tag="v1.13.1", submodules=True) version("1.13.1", tag="v1.13.1", submodules=True)
@@ -545,7 +546,7 @@ def enable_or_disable(variant, keyword="USE", var=None, newer=False):
elif "~onnx_ml" in self.spec: elif "~onnx_ml" in self.spec:
env.set("ONNX_ML", "OFF") env.set("ONNX_ML", "OFF")
if not self.spec.satisfies("@master"): if not self.spec.satisfies("@main,master"):
env.set("PYTORCH_BUILD_VERSION", self.version) env.set("PYTORCH_BUILD_VERSION", self.version)
env.set("PYTORCH_BUILD_NUMBER", 0) env.set("PYTORCH_BUILD_NUMBER", 0)