From 1d10c6c73091c6130b3af696d4a9309497dcdd54 Mon Sep 17 00:00:00 2001
From: "Tomoki, Karatsu" <49965247+t-karatsu@users.noreply.github.com>
Date: Tue, 23 Jun 2020 15:54:54 +0900
Subject: [PATCH] caffe: use MPI compiler when using hdf5+mpi (#17125)

---
 var/spack/repos/builtin/packages/caffe/package.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/var/spack/repos/builtin/packages/caffe/package.py b/var/spack/repos/builtin/packages/caffe/package.py
index 7df83b1d537..187bdc5764e 100644
--- 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