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

py-torchvision: fix __version__ for stable releases (#31657)

Browse Source
This commit is contained in:
Adam J. Stewart 2022-07-21 00:36:11 -07:00 committed by GitHub
parent 609bc89fd9
commit 1af76946a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
var/spack/repos/builtin/packages/py-torchvision/package.py
View File

@ -18,7 +18,6 @@ class PyTorchvision(PythonPackage):
maintainers = ['adamjstewart']
version('main', branch='main')
version('master', branch='main', deprecated=True)
version('0.13.0', sha256='2fe9139150800820d02c867a0b64b7c7fbc964d48d76fae235d6ef9215eabcf4')
version('0.12.0', sha256='99e6d3d304184895ff4f6152e2d2ec1cbec89b3e057d9c940ae0125546b04e91')
version('0.11.3', sha256='b4c51d27589783e6e6941ecaa67b55f6f41633874ec37f80b64a0c92c3196e0c')
@ -62,7 +61,7 @@ class PyTorchvision(PythonPackage):
depends_on('py-six', when='@:0.5', type=('build', 'run'))
# https://github.com/pytorch/vision#installation
depends_on('py-torch@master', when='@master', type=('build', 'link', 'run'))
depends_on('py-torch@master', when='@main', type=('build', 'link', 'run'))
depends_on('py-torch@1.12.0', when='@0.13.0', type=('build', 'link', 'run'))
depends_on('py-torch@1.11.0', when='@0.12.0', type=('build', 'link', 'run'))
depends_on('py-torch@1.10.2', when='@0.11.3', type=('build', 'link', 'run'))
@ -124,6 +123,11 @@ def setup_build_environment(self, env):
env.set('CPATH', ':'.join(include))
env.set('LIBRARY_PATH', ':'.join(library))
# By default, version is read from `version.txt`, but this includes an `a0`
# suffix used for alpha builds. Override the version for stable releases.
if not self.spec.satisfies('@main'):
env.set('BUILD_VERSION', self.version)
if '+cuda' in self.spec['py-torch']:
env.set('FORCE_CUDA', 1)
env.set('CUDA_HOME', self.spec['cuda'].prefix)