8ee4b52560
- can be used, for example, to track global states:
// Encode as 0:empty, 1:uniform, 2:nonuniform, 3:mixed
PackedList<2> uniformity(fields.size());
forAll(fields, i)
{
uniformity.set(i, fields[i].whichUniformity());
}
reduce
(
uniformity.data(),
uniformity.size_data(),
bitOrOp<unsigned>()
);
324 lines
19 KiB
C
324 lines
19 KiB
C
/*---------------------------------------------------------------------------*\
|
|
========= |
|
|
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
|
|
\\ / O peration |
|
|
\\ / A nd | www.openfoam.com
|
|
\\/ M anipulation |
|
|
-------------------------------------------------------------------------------
|
|
Copyright (C) 2022-2023 OpenCFD Ltd.
|
|
-------------------------------------------------------------------------------
|
|
License
|
|
This file is part of OpenFOAM.
|
|
|
|
OpenFOAM 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.
|
|
|
|
OpenFOAM 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 OpenFOAM. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
\*---------------------------------------------------------------------------*/
|
|
|
|
#include "Pstream.H"
|
|
#include "PstreamReduceOps.H"
|
|
#include "UPstreamWrapping.H"
|
|
|
|
#include <cinttypes>
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
// Special reductions for bool
|
|
|
|
void Foam::UPstream::reduceAnd(bool& value, const label comm)
|
|
{
|
|
PstreamDetail::allReduce(&value, 1, MPI_C_BOOL, MPI_LAND, comm);
|
|
}
|
|
|
|
|
|
void Foam::UPstream::reduceOr(bool& value, const label comm)
|
|
{
|
|
PstreamDetail::allReduce(&value, 1, MPI_C_BOOL, MPI_LOR, comm);
|
|
}
|
|
|
|
|
|
void Foam::reduce
|
|
(
|
|
bool& value,
|
|
const andOp<bool>&,
|
|
const int tag, /* (unused) */
|
|
const label comm
|
|
)
|
|
{
|
|
PstreamDetail::allReduce(&value, 1, MPI_C_BOOL, MPI_LAND, comm);
|
|
}
|
|
|
|
|
|
void Foam::reduce
|
|
(
|
|
bool& value,
|
|
const orOp<bool>&,
|
|
const int tag, /* (unused) */
|
|
const label comm
|
|
)
|
|
{
|
|
PstreamDetail::allReduce(&value, 1, MPI_C_BOOL, MPI_LOR, comm);
|
|
}
|
|
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
// Common reductions
|
|
|
|
#undef Pstream_CommonReductions
|
|
#define Pstream_CommonReductions(Native, TaggedType) \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native values[], \
|
|
const int size, \
|
|
const minOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
values, size, TaggedType, MPI_MIN, comm \
|
|
); \
|
|
} \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native values[], \
|
|
const int size, \
|
|
const maxOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
values, size, TaggedType, MPI_MAX, comm \
|
|
); \
|
|
} \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native values[], \
|
|
const int size, \
|
|
const sumOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
values, size, TaggedType, MPI_SUM, comm \
|
|
); \
|
|
} \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native& value, \
|
|
const minOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
&value, 1, TaggedType, MPI_MIN, comm \
|
|
); \
|
|
} \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native& value, \
|
|
const maxOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
&value, 1, TaggedType, MPI_MAX, comm \
|
|
); \
|
|
} \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native& value, \
|
|
const sumOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
&value, 1, TaggedType, MPI_SUM, comm \
|
|
); \
|
|
}
|
|
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
// Floating-point reductions
|
|
|
|
#undef Pstream_FloatReductions
|
|
#define Pstream_FloatReductions(Native, TaggedType) \
|
|
\
|
|
Pstream_CommonReductions(Native, TaggedType); \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native values[], \
|
|
const int size, \
|
|
const sumOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm, \
|
|
UPstream::Request& req \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
values, size, TaggedType, MPI_SUM, comm, &req, nullptr \
|
|
); \
|
|
} \
|
|
\
|
|
/* Deprecated: prefer version with UPstream::Request */ \
|
|
void Foam::reduce \
|
|
( \
|
|
Native values[], \
|
|
const int size, \
|
|
const sumOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm, \
|
|
label& requestID \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
values, size, TaggedType, MPI_SUM, comm, nullptr, &requestID \
|
|
); \
|
|
} \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native& value, \
|
|
const sumOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm, \
|
|
UPstream::Request& req \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
&value, 1, TaggedType, MPI_SUM, comm, &req, nullptr \
|
|
); \
|
|
} \
|
|
\
|
|
/* Deprecated: prefer version with UPstream::Request */ \
|
|
void Foam::reduce \
|
|
( \
|
|
Native& value, \
|
|
const sumOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm, \
|
|
label& requestID \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
&value, 1, TaggedType, MPI_SUM, comm, nullptr, &requestID \
|
|
); \
|
|
} \
|
|
\
|
|
void Foam::sumReduce \
|
|
( \
|
|
Native& value, \
|
|
label& count, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm \
|
|
) \
|
|
{ \
|
|
if (UPstream::is_parallel(comm)) \
|
|
{ \
|
|
Native values[2]; \
|
|
values[0] = static_cast<Native>(count); \
|
|
values[1] = value; \
|
|
\
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
values, 2, TaggedType, MPI_SUM, comm \
|
|
); \
|
|
\
|
|
count = static_cast<label>(values[0]); \
|
|
value = values[1]; \
|
|
} \
|
|
}
|
|
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
// Bitwise reductions
|
|
|
|
#undef Pstream_BitwiseReductions
|
|
#define Pstream_BitwiseReductions(Native, TaggedType) \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native values[], \
|
|
const int size, \
|
|
const bitOrOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
values, size, TaggedType, MPI_BOR, comm \
|
|
); \
|
|
} \
|
|
\
|
|
void Foam::reduce \
|
|
( \
|
|
Native& value, \
|
|
const bitOrOp<Native>&, \
|
|
const int tag, /* (unused) */ \
|
|
const label comm \
|
|
) \
|
|
{ \
|
|
PstreamDetail::allReduce<Native> \
|
|
( \
|
|
&value, 1, TaggedType, MPI_BOR, comm \
|
|
); \
|
|
} \
|
|
|
|
\
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
Pstream_CommonReductions(int32_t, MPI_INT32_T);
|
|
Pstream_CommonReductions(int64_t, MPI_INT64_T);
|
|
Pstream_CommonReductions(uint32_t, MPI_UINT32_T);
|
|
Pstream_CommonReductions(uint64_t, MPI_UINT64_T);
|
|
|
|
Pstream_FloatReductions(float, MPI_FLOAT);
|
|
Pstream_FloatReductions(double, MPI_DOUBLE);
|
|
|
|
Pstream_BitwiseReductions(unsigned, MPI_UNSIGNED);
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
#undef Pstream_CommonReductions
|
|
#undef Pstream_FloatReductions
|
|
#undef Pstream_BitwiseReductions
|
|
|
|
|
|
// ************************************************************************* //
|