fix for sundials compiled with mpi

seamplex · Dec 27, 2024 · e368573 · e368573
1 parent 6201f62
commit e368573
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/math/dae.c b/src/math/dae.c
@@ -345,7 +345,13 @@ int feenox_dae_init(void) {
 
   int err = 0; // this is used inside the ida_call() macro
 #if SUNDIALS_VERSION_MAJOR >= 6
+ #ifdef SUNDIALS_MPI_ENABLED
+  MPI_Init(NULL, NULL);
+  ida_call(SUNContext_Create(MPI_COMM_WORLD, &feenox.dae.ctx));
+ #else
   ida_call(SUNContext_Create(NULL, &feenox.dae.ctx));
+ #endif
+
   feenox_check_alloc(feenox.dae.x = N_VNew_Serial(feenox.dae.dimension, feenox.dae.ctx));
   feenox_check_alloc(feenox.dae.dxdt = N_VNew_Serial(feenox.dae.dimension, feenox.dae.ctx));
   feenox_check_alloc(feenox.dae.id = N_VNew_Serial(feenox.dae.dimension, feenox.dae.ctx));