diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/caffe/package.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/caffe/package.py b/var/spack/repos/builtin/packages/caffe/package.py index 5d83b8e1a7..ba01c25a6e 100644 --- a/var/spack/repos/builtin/packages/caffe/package.py +++ b/var/spack/repos/builtin/packages/caffe/package.py @@ -73,7 +73,8 @@ class Caffe(CMakePackage): def cmake_args(self): spec = self.spec - args = ['-DBLAS={0}'.format(spec['blas'].name), + 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), '-DBUILD_python=%s' % ('+python' in spec), |