caffe: use MPI compiler when using hdf5+mpi (#17125)

2020-06-23 15:54:54 +09:00 · 2020-06-23 15:54:54 +09:00 · 1d10c6c730
commit 1d10c6c730
parent 55f26b0ccb
1 changed files with 6 additions and 0 deletions
--- a/var/spack/repos/builtin/packages/caffe/package.py
+++ b/var/spack/repos/builtin/packages/caffe/package.py
@ -81,4 +81,10 @@ def cmake_args(self):
            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