amgcl_distributed_csr_spmm_utilities.h

Go to the documentation of this file.

 
//    |  /           |
//    ' /   __| _` | __|  _ \   __|
//    . \  |   (   | |   (   |\__ \.
//   _|\_\_|  \__,_|\__|\___/ ____/
//                   Multi-Physics
//
//  License:         BSD License
//                   Kratos default license: kratos/license.txt
//
//  Main authors:    Riccardo Rossi
//
//
 
#if !defined(KRATOS_CSR_SPMM_UTILITIES_H_INCLUDED)
#define  KRATOS_CSR_SPMM_UTILITIES_H_INCLUDED
 
// System includes
 
// External includes
#include <amgcl/backend/builtin.hpp>
#include <amgcl/adapter/zero_copy.hpp>
 
// Project includes
#include "containers/csr_matrix.h"
#include "mpi/utilities/amgcl_distributed_csr_conversion_utilities.h"
 
namespace Kratos
{
 
class DistributedAmgclCSRSpMMUtilities
{
 
public:
 
    template< class TDataType=double, class TIndexType=std::size_t >
    static typename DistributedCsrMatrix<TDataType, TIndexType>::Pointer SparseMultiply(
        const DistributedCsrMatrix<TDataType, TIndexType>& rA,
        const DistributedCsrMatrix<TDataType, TIndexType>& rB
    )
    {
        bool move_to_backend=false;
 
        auto Aoffdiag_global_index2 = rA.GetOffDiagonalIndex2DataInGlobalNumbering();
 
        auto pAamgcl = AmgclDistributedCSRConversionUtilities::ConvertToAmgcl<double,IndexType>
                       (rA,Aoffdiag_global_index2,move_to_backend);
 
        auto Boffdiag_global_index2 = rB.GetOffDiagonalIndex2DataInGlobalNumbering();
        auto pBamgcl = AmgclDistributedCSRConversionUtilities::ConvertToAmgcl<double,IndexType>
                       (rB,Boffdiag_global_index2,move_to_backend);
 
        auto Camgcl = product(*pAamgcl, *pBamgcl);
 
        sort_rows(*Camgcl);
 
        return AmgclDistributedCSRConversionUtilities::ConvertToCsrMatrix<TDataType,IndexType>(*Camgcl, rA.GetComm());
    }
 
 
 
};
 
}
 
#endif // KRATOS_CSR_SPMM_UTILITIES_H_INCLUDED