Tpetra_leftAndOrRightScaleCrsMatrix_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_LEFTANDORRIGHTSCALECRSMATRIX_DECL_HPP
#define TPETRA_LEFTANDORRIGHTSCALECRSMATRIX_DECL_HPP


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

namespace Tpetra {

enum EScaling {
  SCALING_MULTIPLY,
  SCALING_DIVIDE
};

template <class SC, class LO, class GO, class NT>
void leftAndOrRightScaleCrsMatrix(Tpetra::CrsMatrix<SC, LO, GO, NT>& A,
                                  const Kokkos::View<
#if KOKKOS_VERSION >= 40799
                                      const typename KokkosKernels::ArithTraits<SC>::mag_type*,
#else
                                      const typename Kokkos::ArithTraits<SC>::mag_type*,
#endif
                                      typename NT::device_type>& rowScalingFactors,
                                  const Kokkos::View<
#if KOKKOS_VERSION >= 40799
                                      const typename KokkosKernels::ArithTraits<SC>::mag_type*,
#else
                                      const typename Kokkos::ArithTraits<SC>::mag_type*,
#endif
                                      typename NT::device_type>& colScalingFactors,
                                  const bool leftScale,
                                  const bool rightScale,
                                  const bool assumeSymmetric,
                                  const EScaling scaling);

template <class SC, class LO, class GO, class NT>
void leftAndOrRightScaleCrsMatrix(Tpetra::CrsMatrix<SC, LO, GO, NT>& A,
#if KOKKOS_VERSION >= 40799
                                  const Tpetra::Vector<typename KokkosKernels::ArithTraits<SC>::mag_type,
#else
                                  const Tpetra::Vector<typename Kokkos::ArithTraits<SC>::mag_type,
#endif
                                                       LO, GO, NT>& rowScalingFactors,
#if KOKKOS_VERSION >= 40799
                                  const Tpetra::Vector<typename KokkosKernels::ArithTraits<SC>::mag_type,
#else
                                  const Tpetra::Vector<typename Kokkos::ArithTraits<SC>::mag_type,
#endif
                                                       LO, GO, NT>& colScalingFactors,
                                  const bool leftScale,
                                  const bool rightScale,
                                  const bool assumeSymmetric,
                                  const EScaling scaling);

}  // namespace Tpetra

#endif  // TPETRA_LEFTANDORRIGHTSCALECRSMATRIX_DECL_HPP