Branch data     Line data    Source code

       1                 :            : // *****************************************************************************
       2                 :            : /*!
       3                 :            :   \file      src/Statistics/PDFReducer.cpp
       4                 :            :   \copyright 2012-2015 J. Bakosi,
       5                 :            :              2016-2018 Los Alamos National Security, LLC.,
       6                 :            :              2019-2021 Triad National Security, LLC.,
       7                 :            :              2022-2024 J. Bakosi
       8                 :            :              All rights reserved. See the LICENSE file for details.
       9                 :            :   \brief     Custom Charm++ reducer for merging PDFs across PEs
      10                 :            :   \details   Custom Charm++ reducer for merging PDFs across PEs.
      11                 :            : */
      12                 :            : // *****************************************************************************
      13                 :            : 
      14                 :            : #include "PDFReducer.hpp"
      15                 :            : 
      16                 :            : namespace tk {
      17                 :            : 
      18                 :            : std::pair< int, std::unique_ptr<char[]> >
      19                 :       2922 : serialize( std::size_t meshid, const std::vector< tk::UniPDF >& u )
      20                 :            : // *****************************************************************************
      21                 :            : // Serialize univariate PDFs to raw memory stream
      22                 :            : //! \param[in] meshid Mesh ID
      23                 :            : //! \param[in] u Univariate PDFs
      24                 :            : //! \return Pair of the length and the raw stream containing the serialized PDFs
      25                 :            : // *****************************************************************************
      26                 :            : {
      27                 :            :   // Prepare for serializing PDF to a raw binary stream, compute size
      28                 :       2922 :   PUP::sizer sizer;
      29                 :            :   // cppcheck-suppress uninitvar
      30         [ +  - ]:       2922 :   sizer | meshid;
      31         [ +  - ]:       2922 :   sizer | const_cast< std::vector< tk::UniPDF >& >( u );
      32                 :            : 
      33                 :            :   // Create raw character stream to store the serialized PDF
      34         [ +  - ]:       2922 :   std::unique_ptr<char[]> flatData = std::make_unique<char[]>( sizer.size() );
      35                 :            : 
      36                 :            :   // Serialize PDF, the message will contain a univariate PDF
      37                 :       2922 :   PUP::toMem packer( flatData.get() );
      38                 :            :   // cppcheck-suppress uninitvar
      39         [ +  - ]:       2922 :   packer | meshid;
      40         [ +  - ]:       2922 :   packer | const_cast< std::vector< tk::UniPDF >& >( u );
      41                 :            : 
      42                 :            :   // Return size of and raw stream
      43                 :       5844 :   return { sizer.size(), std::move(flatData) };
      44                 :       2922 : }
      45                 :            : 
      46                 :            : CkReductionMsg*
      47                 :        463 : mergeUniPDFs( int nmsg, CkReductionMsg **msgs )
      48                 :            : // *****************************************************************************
      49                 :            : // Charm++ custom reducer for merging a univariate PDFs during reduction across
      50                 :            : // PEs
      51                 :            : //! \param[in] nmsg Number of messages in msgs
      52                 :            : //! \param[in] msgs Charm++ reduction message containing the serialized PDF
      53                 :            : //! \return Aggregated PDF built for further aggregation if needed
      54                 :            : // *****************************************************************************
      55                 :            : {
      56                 :            :   // Will store deserialized univariate PDFs
      57                 :            :   std::size_t meshid;
      58                 :        463 :   std::vector< tk::UniPDF > updf;
      59                 :            : 
      60                 :            :   // Create PUP deserializer based on message passed in
      61                 :        463 :   PUP::fromMem creator( msgs[0]->getData() );
      62                 :            : 
      63                 :            :   // Deserialize PDFs from raw stream
      64                 :            :   // cppcheck-suppress uninitvar
      65         [ +  - ]:        463 :   creator | meshid;
      66         [ +  - ]:        463 :   creator | updf;
      67                 :            : 
      68         [ +  + ]:       2680 :   for (int m=1; m<nmsg; ++m) {
      69                 :            :     // Unpack PDF
      70                 :            :     std::size_t mid;
      71                 :       2217 :     std::vector< tk::UniPDF > u;
      72                 :       2217 :     PUP::fromMem curCreator( msgs[m]->getData() );
      73                 :            :     // cppcheck-suppress uninitvar
      74         [ +  - ]:       2217 :     curCreator | mid;
      75         [ +  - ]:       2217 :     curCreator | u;
      76                 :            :     // Merge PDFs
      77                 :            :     // cppcheck-suppress uninitvar
      78                 :       2217 :     meshid = mid;
      79                 :       2217 :     std::size_t i = 0;
      80                 :            :     // cppcheck-suppress knownEmptyContainer
      81                 :            :     // cppcheck-suppress containerOutOfBounds
      82 [ +  - ][ +  + ]:       8868 :     for (const auto& p : u) updf[i++].addPDF( p );
      83                 :       2217 :   }
      84                 :            : 
      85                 :            :   // Serialize vector of merged PDF to raw stream
      86                 :            :   // cppcheck-suppress uninitvar
      87         [ +  - ]:        463 :   auto stream = tk::serialize( meshid, updf );
      88                 :            : 
      89                 :            :   // Forward serialized PDFs
      90         [ +  - ]:        926 :   return CkReductionMsg::buildNew( stream.first, stream.second.get() );
      91                 :        463 : }
      92                 :            : 
      93                 :            : } // tk::