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

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

Browse Source
This commit is contained in:
Tomoki, Karatsu 2020-06-23 15:54:54 +09:00 committed by GitHub
parent 55f26b0ccb
commit 1d10c6c730
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -81,4 +81,10 @@ def cmake_args(self):
version = spec['python'].version.up_to(1) version = spec['python'].version.up_to(1)
args.append('-Dpython_version=%s' % version) 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 return args