From 3ade5516a2f8b9ec998273c0f14884eb217415d0 Mon Sep 17 00:00:00 2001 From: Rohit Goswami Date: Mon, 25 Apr 2022 20:33:37 +0000 Subject: 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 Co-authored-by: Adam J. Stewart --- var/spack/repos/builtin/packages/py-torch/package.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'var') 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: -- cgit v1.2.3-60-g2f50