From 785168f5a1b613ab76f0ed65896e5bf0b0b6af24 Mon Sep 17 00:00:00 2001 From: Gregory Lee Date: Tue, 23 May 2017 14:35:42 -0700 Subject: fix build command for caffe with openblas (#4299) --- var/spack/repos/builtin/packages/caffe/package.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'var') 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), -- cgit v1.2.3-70-g09d2