summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/caffe/package.py3
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),