Tpetra_computeRowAndColumnOneNorms_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//          Tpetra: Templated Linear Algebra Services Package
//
// Copyright 2008 NTESS and the Tpetra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef TPETRA_COMPUTEROWANDCOLUMNONENORMS_DECL_HPP
#define TPETRA_COMPUTEROWANDCOLUMNONENORMS_DECL_HPP


#include "TpetraCore_config.h"
#include "Kokkos_Core.hpp"
#if KOKKOS_VERSION >= 40799
#include "KokkosKernels_ArithTraits.hpp"
#else
#include "Kokkos_ArithTraits.hpp"
#endif
#include "Tpetra_Details_EquilibrationInfo.hpp"
#include "Tpetra_RowMatrix_fwd.hpp"

namespace Tpetra {

template <class SC, class LO, class GO, class NT>
#if KOKKOS_VERSION >= 40799
Details::EquilibrationInfo<typename KokkosKernels::ArithTraits<SC>::val_type,
#else
Details::EquilibrationInfo<typename Kokkos::ArithTraits<SC>::val_type,
#endif
                           typename NT::device_type>
computeRowOneNorms(const Tpetra::RowMatrix<SC, LO, GO, NT>& A);

template <class SC, class LO, class GO, class NT>
#if KOKKOS_VERSION >= 40799
Details::EquilibrationInfo<typename KokkosKernels::ArithTraits<SC>::val_type,
#else
Details::EquilibrationInfo<typename Kokkos::ArithTraits<SC>::val_type,
#endif
                           typename NT::device_type>
computeRowAndColumnOneNorms(const Tpetra::RowMatrix<SC, LO, GO, NT>& A,
                            const bool assumeSymmetric);

}  // namespace Tpetra

#endif  // TPETRA_COMPUTEROWANDCOLUMNONENORMS_DECL_HPP