summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/py-torch/package.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py
index aa8f219d74..31bfb71657 100644
--- a/var/spack/repos/builtin/packages/py-torch/package.py
+++ b/var/spack/repos/builtin/packages/py-torch/package.py
@@ -51,7 +51,10 @@ class PyTorch(PythonPackage, CudaPackage):
]
version('master', branch='master', submodules=True)
- version('1.4.0', tag='v1.4.0', submodules=True)
+ version('1.4.1', tag='v1.4.1', submodules=True)
+ # see https://github.com/pytorch/pytorch/issues/35149
+ version('1.4.0', tag='v1.4.0', submodules=True,
+ submodules_delete=['third_party/fbgemm'])
version('1.3.1', tag='v1.3.1', submodules=True)
version('1.3.0', tag='v1.3.0', submodules=True)
version('1.2.0', tag='v1.2.0', submodules=True)
@@ -101,6 +104,8 @@ class PyTorch(PythonPackage, CudaPackage):
conflicts('+redis', when='@:1.0')
conflicts('+zstd', when='@:1.0')
conflicts('+tbb', when='@:1.1')
+ # see https://github.com/pytorch/pytorch/issues/35149
+ conflicts('+fbgemm', when='@1.4.0')
cuda_arch_conflict = ('This version of Torch/Caffe2 only supports compute '
'capabilities ')