diff options
author | Gregory Lee <lee218@llnl.gov> | 2017-05-23 14:35:42 -0700 |
---|---|---|
committer | Adam J. Stewart <ajstewart426@gmail.com> | 2017-05-23 16:35:42 -0500 |
commit | 785168f5a1b613ab76f0ed65896e5bf0b0b6af24 (patch) | |
tree | 9c496e418b029e26a5c0ce21e35fb4fd81b74e76 | |
parent | 43edcfbf13d8811347750a653ee8a7b1b314c228 (diff) | |
download | spack-785168f5a1b613ab76f0ed65896e5bf0b0b6af24.tar.gz spack-785168f5a1b613ab76f0ed65896e5bf0b0b6af24.tar.bz2 spack-785168f5a1b613ab76f0ed65896e5bf0b0b6af24.tar.xz spack-785168f5a1b613ab76f0ed65896e5bf0b0b6af24.zip |
fix build command for caffe with openblas (#4299)
-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), |