From d7f054e261711f84391e58fa129aa1f6f9c24423 Mon Sep 17 00:00:00 2001
From: Marcin Kirsz <mkirsz@ed.ac.uk>
Date: Mon, 7 Oct 2024 13:02:12 +0100
Subject: [PATCH] added mpi include

---
 trainer.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/trainer.h b/trainer.h
index 3e6b542..1b26782 100644
--- a/trainer.h
+++ b/trainer.h
@@ -52,7 +52,9 @@ class Trainer {
     }
 
 };
+
 #ifdef TADAH_BUILD_MPI
+#include <mpi.h>
 extern "C" void blacs_get_(int*, int*, int*);
 extern "C" void blacs_pinfo_(int*, int*);
 extern "C" void blacs_gridinit_(int*, char*, int*, int*);
-- 
GitLab