diff options
author | Tomoki, Karatsu <49965247+t-karatsu@users.noreply.github.com> | 2020-06-23 15:54:54 +0900 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-06-23 08:54:54 +0200 |
commit | 1d10c6c73091c6130b3af696d4a9309497dcdd54 (patch) | |
tree | bd3548ff8bb6720d2e26acd21f5321bd781b8e8b | |
parent | 55f26b0ccb6f53d2a554730c343c6d24a340cc29 (diff) | |
download | spack-1d10c6c73091c6130b3af696d4a9309497dcdd54.tar.gz spack-1d10c6c73091c6130b3af696d4a9309497dcdd54.tar.bz2 spack-1d10c6c73091c6130b3af696d4a9309497dcdd54.tar.xz spack-1d10c6c73091c6130b3af696d4a9309497dcdd54.zip |
caffe: use MPI compiler when using hdf5+mpi (#17125)
-rw-r--r-- | var/spack/repos/builtin/packages/caffe/package.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/caffe/package.py b/var/spack/repos/builtin/packages/caffe/package.py index 7df83b1d53..187bdc5764 100644 --- a/var/spack/repos/builtin/packages/caffe/package.py +++ b/var/spack/repos/builtin/packages/caffe/package.py @@ -81,4 +81,10 @@ class Caffe(CMakePackage): version = spec['python'].version.up_to(1) args.append('-Dpython_version=%s' % version) + if spec['hdf5'].satisfies('+mpi'): + args.extend([ + '-DCMAKE_C_COMPILER={0}'.format(self.spec['mpi'].mpicc), + '-DCMAKE_CXX_COMPILER={0}'.format(self.spec['mpi'].mpicxx) + ]) + return args |