From 2b792f943f25a827257933d3a9f1b84e48f88c1a Mon Sep 17 00:00:00 2001 From: Jimmy Tang Date: Tue, 25 Jul 2017 13:53:33 +0100 Subject: Rename the gpu variant to cuda, this is to be consistent with other (#4890) packages. --- var/spack/repos/builtin/packages/caffe/package.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/var/spack/repos/builtin/packages/caffe/package.py b/var/spack/repos/builtin/packages/caffe/package.py index 4e2ef26dbb..021814c7fd 100644 --- a/var/spack/repos/builtin/packages/caffe/package.py +++ b/var/spack/repos/builtin/packages/caffe/package.py @@ -39,7 +39,7 @@ class Caffe(CMakePackage): version('rc3', '84e39223115753b48312a8bf48c31f59') version('rc2', 'c331932e34b5e2f5022fcc34c419080f') - variant('gpu', default=False, + variant('cuda', default=False, description='Builds with support for GPUs via CUDA and cuDNN') variant('opencv', default=True, description='Build with OpenCV support') @@ -54,7 +54,7 @@ class Caffe(CMakePackage): depends_on('boost') depends_on('boost +python', when='+python') - depends_on('cuda', when='+gpu') + depends_on('cuda', when='+cuda') depends_on('blas') depends_on('protobuf') depends_on('glog') @@ -75,8 +75,8 @@ class Caffe(CMakePackage): spec = self.spec args = ['-DBLAS={0}'.format('open' if spec['blas'].name == 'openblas' else spec['blas'].name), - '-DCPU_ONLY=%s' % ('~gpu' in spec), - '-DUSE_CUDNN=%s' % ('+gpu' in spec), + '-DCPU_ONLY=%s' % ('~cuda' in spec), + '-DUSE_CUDNN=%s' % ('+cuda' in spec), '-DBUILD_python=%s' % ('+python' in spec), '-DBUILD_python_layer=%s' % ('+python' in spec), '-DBUILD_matlab=%s' % ('+matlab' in spec), -- cgit v1.2.3-60-g2f50