diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/py-torch/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 7022e562ff..19991da898 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -16,6 +16,9 @@ class PyTorch(PythonPackage, CudaPackage): maintainers = ['adamjstewart'] + # Exact set of modules is version- and variant-specific + import_modules = ['torch', 'torch.autograd', 'torch.nn', 'torch.utils'] + version('master', branch='master', submodules=True) version('1.7.1', tag='v1.7.1', submodules=True) version('1.7.0', tag='v1.7.0', submodules=True) |