diff options
| -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 8190e01102..b7d6fa0c0e 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -49,6 +49,7 @@ class PyTorch(PythonPackage, CudaPackage): # All options are defined in CMakeLists.txt. # Some are listed in setup.py, but not all. + variant('debug', default=False, description="Build with debugging support") variant('caffe2', default=True, description='Build Caffe2', when='@1.7:') variant('test', default=False, description='Build C++ test binaries') variant('cuda', default=not is_darwin, description='Use CUDA') @@ -343,6 +344,11 @@ class PyTorch(PythonPackage, CudaPackage): enable_or_disable('gloo', newer=True) enable_or_disable('tensorpipe') + if '+debug' in self.spec: + env.set('DEBUG', 'ON') + else: + env.set('DEBUG', 'OFF') + if '+onnx_ml' in self.spec: env.set('ONNX_ML', 'ON') elif '~onnx_ml' in self.spec: |