From 7ee218b1e28bfde2f897f8c2ac31ae0d2efec9f3 Mon Sep 17 00:00:00 2001
From: Vegard Kippe <vkip@equinor.com>
Date: Fri, 13 Sep 2024 14:32:22 +0200
Subject: [PATCH] In partitioning, only account for wells that will be active
 at some point in the simulation.

---
 opm/simulators/flow/BlackoilModelNldd.hpp     | 2 +-
 opm/simulators/flow/GenericCpGridVanguard.cpp | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/opm/simulators/flow/BlackoilModelNldd.hpp b/opm/simulators/flow/BlackoilModelNldd.hpp
index 45e0f6a0b..bab7872c1 100644
--- a/opm/simulators/flow/BlackoilModelNldd.hpp
+++ b/opm/simulators/flow/BlackoilModelNldd.hpp
@@ -949,7 +949,7 @@ private:
         const auto need_wells = param.local_domain_partition_method_ == "zoltan";
 
         const auto wells = need_wells
-            ? this->model_.simulator().vanguard().schedule().getWellsatEnd()
+            ? this->model_.simulator().vanguard().schedule().getActiveWellsatEnd()
             : std::vector<Well>{};
 
         const auto& possibleFutureConnectionSet = need_wells
diff --git a/opm/simulators/flow/GenericCpGridVanguard.cpp b/opm/simulators/flow/GenericCpGridVanguard.cpp
index de6b967b4..2ed7f7784 100644
--- a/opm/simulators/flow/GenericCpGridVanguard.cpp
+++ b/opm/simulators/flow/GenericCpGridVanguard.cpp
@@ -194,7 +194,7 @@ doLoadBalance_(const Dune::EdgeWeightMethod             edgeWeightsMethod,
         }
 
         const auto wells = ((mpiSize > 1) || partitionJacobiBlocks)
-            ? schedule.getWellsatEnd()
+            ? schedule.getActiveWellsatEnd()
             : std::vector<Well>{};
         const auto& possibleFutureConnections = schedule.getPossibleFutureConnections();
         // Distribute the grid and switch to the distributed view.