diff options
| author | Baptiste Jonglez <30461003+jonglezb@users.noreply.github.com> | 2021-10-29 20:00:48 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-10-29 13:00:48 -0500 |
| commit | b8bc030a3cacc8761d3030965e57339f5bd15e2e (patch) | |
| tree | d3b381e648d580ff15d2ee68ff94ebf527d49698 | |
| parent | 3eb52b48b8d0f3379e257ab0817fef4e1b7f4c5e (diff) | |
| download | spack-b8bc030a3cacc8761d3030965e57339f5bd15e2e.tar.gz spack-b8bc030a3cacc8761d3030965e57339f5bd15e2e.tar.bz2 spack-b8bc030a3cacc8761d3030965e57339f5bd15e2e.tar.xz spack-b8bc030a3cacc8761d3030965e57339f5bd15e2e.zip | |
py-torch: Add a breakpad variant, disable it for ppc64 and ppc64le (#26990)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 6 |
1 files changed, 6 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 f3e6c58051..19805f48a0 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -75,6 +75,7 @@ class PyTorch(PythonPackage, CudaPackage): variant('gloo', default=not is_darwin, description='Use Gloo') variant('tensorpipe', default=not is_darwin, description='Use TensorPipe') variant('onnx_ml', default=True, description='Enable traditional ONNX ML API') + variant('breakpad', default=True, description='Enable breakpad crash dump library') conflicts('+cuda', when='+rocm') conflicts('+cudnn', when='~cuda') @@ -98,6 +99,9 @@ class PyTorch(PythonPackage, CudaPackage): conflicts('+fbgemm', when='@:0.4,1.4.0') conflicts('+qnnpack', when='@:0.4') conflicts('+mkldnn', when='@:0.4') + conflicts('+breakpad', when='@:1.9') # Option appeared in 1.10.0 + conflicts('+breakpad', when='target=ppc64:', msg='Unsupported') + conflicts('+breakpad', when='target=ppc64le:', msg='Unsupported') conflicts('cuda_arch=none', when='+cuda', msg='Must specify CUDA compute capabilities of your GPU, see ' @@ -312,6 +316,8 @@ class PyTorch(PythonPackage, CudaPackage): enable_or_disable('kineto') enable_or_disable('magma') enable_or_disable('metal') + if self.spec.satisfies('@1.10:'): + enable_or_disable('breakpad') enable_or_disable('nccl') if '+nccl' in self.spec: |