zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix build command for caffe with openblas (#4299)

Browse Source
This commit is contained in:
Gregory Lee 2017-05-23 14:35:42 -07:00 committed by Adam J. Stewart
parent 43edcfbf13
commit 785168f5a1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
var/spack/repos/builtin/packages/caffe/package.py
View File

@ -73,7 +73,8 @@ class Caffe(CMakePackage):
def cmake_args(self): def cmake_args(self):
spec = self.spec 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), '-DCPU_ONLY=%s' % ('~gpu' in spec),
'-DUSE_CUDNN=%s' % ('+gpu' in spec), '-DUSE_CUDNN=%s' % ('+gpu' in spec),
'-DBUILD_python=%s' % ('+python' in spec), '-DBUILD_python=%s' % ('+python' in spec),