fix build command for caffe with openblas (#4299)

This commit is contained in:
Gregory Lee 2017-05-23 14:35:42 -07:00 committed by Adam J. Stewart
parent 43edcfbf13
commit 785168f5a1

View File

@ -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),