zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

MAINT: Add a debug variant to PyTorch (#30267)

Browse Source 
* 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>
This commit is contained in:
Rohit Goswami 2022-04-25 20:33:37 +00:00 committed by GitHub
parent b41de6d86b
commit 3ade5516a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
var/spack/repos/builtin/packages/py-torch/package.py
View File

@ -49,6 +49,7 @@ class PyTorch(PythonPackage, CudaPackage):
# All options are defined in CMakeLists.txt. # All options are defined in CMakeLists.txt.
# Some are listed in setup.py, but not all. # 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('caffe2', default=True, description='Build Caffe2', when='@1.7:')
variant('test', default=False, description='Build C++ test binaries') variant('test', default=False, description='Build C++ test binaries')
variant('cuda', default=not is_darwin, description='Use CUDA') variant('cuda', default=not is_darwin, description='Use CUDA')
@ -343,6 +344,11 @@ def enable_or_disable(variant, keyword='USE', var=None, newer=False):
enable_or_disable('gloo', newer=True) enable_or_disable('gloo', newer=True)
enable_or_disable('tensorpipe') enable_or_disable('tensorpipe')
if '+debug' in self.spec:
env.set('DEBUG', 'ON')
else:
env.set('DEBUG', 'OFF')
if '+onnx_ml' in self.spec: if '+onnx_ml' in self.spec:
env.set('ONNX_ML', 'ON') env.set('ONNX_ML', 'ON')
elif '~onnx_ml' in self.spec: elif '~onnx_ml' in self.spec: