diff options
author | Rohit Goswami <r95g10@gmail.com> | 2022-04-25 20:33:37 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-04-25 15:33:37 -0500 |
commit | 3ade5516a2f8b9ec998273c0f14884eb217415d0 (patch) | |
tree | 913ba24916b0bd890756c44cbf7763d9ab753332 /var | |
parent | b41de6d86bdf7bf1907bd0f8999d2ccda3003f8d (diff) | |
download | spack-3ade5516a2f8b9ec998273c0f14884eb217415d0.tar.gz spack-3ade5516a2f8b9ec998273c0f14884eb217415d0.tar.bz2 spack-3ade5516a2f8b9ec998273c0f14884eb217415d0.tar.xz spack-3ade5516a2f8b9ec998273c0f14884eb217415d0.zip |
MAINT: Add a debug variant to PyTorch (#30267)
* MAINT: Add a debug flag
* MAINT: Apply suggestions from code review
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
Diffstat (limited to 'var')
-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: |