mirror of
https://github.com/OPM/opm-simulators.git
synced 2025-01-24 01:26:26 -06:00
Add a flow executable that only distributes cells in z-direction.
Executable is named flow_distribute_z and uses the external loadbalancing information. It can be used to test the distributed standard wells on SPE9 with 4 or more processes.
This commit is contained in:
parent
e1e1ff9a17
commit
9ccc953bf0
@ -299,6 +299,18 @@ opm_add_test(flow_poly
|
||||
$<TARGET_OBJECTS:moduleVersion>)
|
||||
target_compile_definitions(flow_poly PRIVATE USE_POLYHEDRALGRID)
|
||||
|
||||
# the production oriented general-purpose ECL simulator
|
||||
opm_add_test(flow_distribute_z
|
||||
ONLY_COMPILE
|
||||
ALWAYS_ENABLE
|
||||
DEFAULT_ENABLE_IF ${FLOW_DEFAULT_ENABLE_IF}
|
||||
DEPENDS opmsimulators
|
||||
LIBRARIES opmsimulators
|
||||
SOURCES
|
||||
flow/flow_distribute_z.cpp
|
||||
${FLOW_TGTS}
|
||||
$<TARGET_OBJECTS:moduleVersion>
|
||||
)
|
||||
|
||||
if (NOT BUILD_FLOW_VARIANTS)
|
||||
set(FLOW_VARIANTS_DEFAULT_ENABLE_IF "FALSE")
|
||||
|
67
flow/flow_distribute_z.cpp
Normal file
67
flow/flow_distribute_z.cpp
Normal file
@ -0,0 +1,67 @@
|
||||
/*
|
||||
Copyright 2013, 2014, 2015 SINTEF ICT, Applied Mathematics.
|
||||
Copyright 2014 Dr. Blatt - HPC-Simulation-Software & Services
|
||||
Copyright 2015, 2017 IRIS AS
|
||||
Copyright 2021 OPM-OP AS
|
||||
|
||||
This file is part of the Open Porous Media project (OPM).
|
||||
|
||||
OPM is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
OPM is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with OPM. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include "config.h"
|
||||
|
||||
#include <opm/simulators/flow/Main.hpp>
|
||||
#include <ebos/eclcpgridvanguard.hh>
|
||||
|
||||
std::vector<int> loadBalanceInZOnly(const Dune::CpGrid& grid)
|
||||
{
|
||||
auto cartMapper = Dune::CartesianIndexMapper<Dune::CpGrid>(grid);
|
||||
auto dims = cartMapper.cartesianDimensions();
|
||||
std::vector<int> parts(grid.leafGridView().size(0));
|
||||
auto numCellsPerProc = dims[2]/grid.comm().size();
|
||||
|
||||
if (grid.size(0)>0 && numCellsPerProc == 0)
|
||||
{
|
||||
OPM_THROW(std::logic_error,
|
||||
"cartesian grid must have more cells in z direction than number of processes.");
|
||||
}
|
||||
|
||||
using ElementMapper =
|
||||
Dune::MultipleCodimMultipleGeomTypeMapper<typename Dune::CpGrid::LeafGridView>;
|
||||
const auto& gridView = grid.leafGridView();
|
||||
const auto& idSet = grid.localIdSet();
|
||||
ElementMapper elemMapper(gridView, Dune::mcmgElementLayout());
|
||||
|
||||
for( const auto &element : elements(gridView) )
|
||||
{
|
||||
const auto& id = idSet.id(element);
|
||||
unsigned elemIdx = elemMapper.index(element);
|
||||
const auto& cartIndex = cartMapper.cartesianIndex(elemIdx);
|
||||
std::array<int,3> cartCoord;
|
||||
cartMapper.cartesianCoordinate(cartIndex, cartCoord);
|
||||
using std::min;
|
||||
auto rank = min(grid.comm().size() -1, cartCoord[2] / numCellsPerProc);
|
||||
parts[id] = rank;
|
||||
}
|
||||
return parts;
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
auto mainObject = Opm::Main(argc, argv);
|
||||
Opm::EclCpGridVanguard<Opm::Properties::TTag::EclFlowProblem>::setExternalLoadBalancer(loadBalanceInZOnly);
|
||||
return mainObject.runDynamic();
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user