Stokhos Package Browser (Single Doxygen Collection)  Version of the Day
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Classes | Macros | Typedefs | Functions
Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp File Reference
#include "Teuchos_UnitTestHelpers.hpp"
#include "Stokhos_UnitTestHelpers.hpp"
#include "Stokhos_Sacado_Kokkos_UQ_PCE.hpp"
#include "KokkosSparse_CrsMatrix.hpp"
#include "KokkosSparse_spmv.hpp"
#include "Kokkos_CrsMatrix_UQ_PCE.hpp"
#include "Kokkos_CrsMatrix_UQ_PCE_Cuda.hpp"
#include "Stokhos_LegendreBasis.hpp"
#include "Stokhos_CompletePolynomialBasis.hpp"
#include "Stokhos_Sparse3Tensor.hpp"
#include "Kokkos_Core.hpp"
Include dependency graph for Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  ScalarTol< Scalar >
 
struct  ScalarTol< float >
 
struct  ScalarTol< double >
 
struct  ReplaceDiagonalValuesKernel< MatrixType >
 
struct  AddDiagonalValuesKernel< MatrixType >
 
struct  AddDiagonalValuesAtomicKernel< MatrixType >
 
struct  Kokkos_MV_Multiply_Op
 
struct  Stokhos_MV_Multiply_Op< Tag >
 

Macros

#define CRSMATRIX_UQ_PCE_TESTS_MATRIXSCALAR(SCALAR)
 
#define CRSMATRIX_UQ_PCE_MEAN_MULTIPLY_TESTS(SCALAR)
 
#define CRS_MATRIX_UQ_PCE_MULTIPLY_TESTS_SCALAR_OP(SCALAR, OP)
 
#define CRS_MATRIX_UQ_PCE_MULTIPLY_TESTS_SCALAR(SCALAR)   CRS_MATRIX_UQ_PCE_MULTIPLY_TESTS_SCALAR_OP( SCALAR, KokkosMultiply )
 
#define CRSMATRIX_UQ_PCE_TESTS_STORAGE(STORAGE)
 
#define CRSMATRIX_UQ_PCE_TESTS_ORDINAL_SCALAR_DEVICE(ORDINAL, SCALAR, DEVICE)
 
#define CRSMATRIX_UQ_PCE_TESTS_DEVICE(DEVICE)   CRSMATRIX_UQ_PCE_TESTS_ORDINAL_SCALAR_DEVICE( int, double, DEVICE )
 

Typedefs

typedef Kokkos_MV_Multiply_Op KokkosMultiply
 

Functions

template<typename IntType >
IntType map_fem_graph_coord (const IntType &N, const IntType &i, const IntType &j, const IntType &k)
 
template<typename ordinal >
ordinal generate_fem_graph (ordinal N, std::vector< std::vector< ordinal > > &graph)
 
template<typename scalar , typename ordinal >
scalar generate_matrix_coefficient (const ordinal nFEM, const ordinal nStoch, const ordinal iRowFEM, const ordinal iColFEM, const ordinal iStoch)
 
template<typename scalar , typename ordinal >
scalar generate_vector_coefficient (const ordinal nFEM, const ordinal nStoch, const ordinal iColFEM, const ordinal iStoch)
 
template<typename kokkos_cijk_type , typename ordinal_type >
kokkos_cijk_type build_cijk (ordinal_type stoch_dim, ordinal_type poly_ord)
 
template<typename array_type , typename scalar_type >
bool compare_rank_2_views (const array_type &y, const array_type &y_exp, const scalar_type rel_tol, const scalar_type abs_tol, Teuchos::FancyOStream &out)
 
template<typename vector_type , typename scalar_type >
bool compareRank1 (const vector_type &y, const vector_type &y_exp, const scalar_type rel_tol, const scalar_type abs_tol, Teuchos::FancyOStream &out)
 
template<typename vector_type , typename scalar_type >
bool compareRank2 (const vector_type &y, const vector_type &y_exp, const scalar_type rel_tol, const scalar_type abs_tol, Teuchos::FancyOStream &out)
 
template<typename MatrixType , typename CijkType >
MatrixType buildDiagonalMatrix (typename MatrixType::ordinal_type nrow, typename MatrixType::ordinal_type pce_size, const CijkType &cijk)
 
 TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL (Kokkos_CrsMatrix_PCE, ReplaceValues, MatrixScalar)
 
 TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL (Kokkos_CrsMatrix_PCE, SumIntoValues, MatrixScalar)
 
 TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL (Kokkos_CrsMatrix_PCE, SumIntoValuesAtomic, MatrixScalar)
 
template<typename PCEType , typename Multiply >
bool test_embedded_pce (const typename PCEType::ordinal_type nGrid, const typename PCEType::ordinal_type stoch_dim, const typename PCEType::ordinal_type poly_ord, KokkosSparse::DeviceConfig dev_config, Multiply multiply_op, Teuchos::FancyOStream &out)
 
 TEUCHOS_UNIT_TEST_TEMPLATE_2_DECL (Kokkos_CrsMatrix_PCE, Multiply, Scalar, MultiplyOp)
 
 TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL (Kokkos_CrsMatrix_PCE, MeanMultiplyRank1, Scalar)
 
 TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL (Kokkos_CrsMatrix_PCE, MeanMultiplyRank2, Scalar)
 

Macro Definition Documentation

#define CRSMATRIX_UQ_PCE_TESTS_MATRIXSCALAR (   SCALAR)
Value:
TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT( \
Kokkos_CrsMatrix_PCE, ReplaceValues, SCALAR ) \
TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT( \
Kokkos_CrsMatrix_PCE, SumIntoValues, SCALAR ) \
TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT( \
Kokkos_CrsMatrix_PCE, SumIntoValuesAtomic, SCALAR )
TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT
#define TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT(TEST_GROUP, TEST_NAME, TYPE)

Definition at line 1009 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

#define CRSMATRIX_UQ_PCE_MEAN_MULTIPLY_TESTS (   SCALAR)
Value:
TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT( \
Kokkos_CrsMatrix_PCE, MeanMultiplyRank1, SCALAR ) \
TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT( \
Kokkos_CrsMatrix_PCE, MeanMultiplyRank2, SCALAR )
TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT
#define TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT(TEST_GROUP, TEST_NAME, TYPE)

Definition at line 1016 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

#define CRS_MATRIX_UQ_PCE_MULTIPLY_TESTS_SCALAR_OP (   SCALAR,
  OP 
)
Value:
TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( \
Kokkos_CrsMatrix_PCE, Multiply, SCALAR, OP )
TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT
TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT(Tpetra_CrsMatrix, MatVec, double, Node) TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT(Tpetra_CrsMatrix

Definition at line 1021 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

#define CRS_MATRIX_UQ_PCE_MULTIPLY_TESTS_SCALAR (   SCALAR)    CRS_MATRIX_UQ_PCE_MULTIPLY_TESTS_SCALAR_OP( SCALAR, KokkosMultiply )

Definition at line 1025 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

#define CRSMATRIX_UQ_PCE_TESTS_STORAGE (   STORAGE)
Value:
typedef Sacado::UQ::PCE<STORAGE> UQ_PCE_ ## STORAGE; \
CRSMATRIX_UQ_PCE_TESTS_MATRIXSCALAR( UQ_PCE_ ## STORAGE ) \
CRS_MATRIX_UQ_PCE_MULTIPLY_TESTS_SCALAR( UQ_PCE_ ## STORAGE ) \
CRSMATRIX_UQ_PCE_MEAN_MULTIPLY_TESTS( UQ_PCE_ ## STORAGE )
CRS_MATRIX_UQ_PCE_MULTIPLY_TESTS_SCALAR
#define CRS_MATRIX_UQ_PCE_MULTIPLY_TESTS_SCALAR(SCALAR)
Definition: Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp:1025
CRSMATRIX_UQ_PCE_TESTS_MATRIXSCALAR
#define CRSMATRIX_UQ_PCE_TESTS_MATRIXSCALAR(SCALAR)
Definition: Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp:1009
CRSMATRIX_UQ_PCE_MEAN_MULTIPLY_TESTS
#define CRSMATRIX_UQ_PCE_MEAN_MULTIPLY_TESTS(SCALAR)
Definition: Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp:1016

Definition at line 1028 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

#define CRSMATRIX_UQ_PCE_TESTS_ORDINAL_SCALAR_DEVICE (   ORDINAL,
  SCALAR,
  DEVICE 
)
Value:
typedef Stokhos::DynamicStorage<ORDINAL,SCALAR,DEVICE> DS; \
CRSMATRIX_UQ_PCE_TESTS_STORAGE( DS )
CRSMATRIX_UQ_PCE_TESTS_STORAGE
#define CRSMATRIX_UQ_PCE_TESTS_STORAGE(STORAGE)
Definition: Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp:1028

Definition at line 1034 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

#define CRSMATRIX_UQ_PCE_TESTS_DEVICE (   DEVICE)    CRSMATRIX_UQ_PCE_TESTS_ORDINAL_SCALAR_DEVICE( int, double, DEVICE )

Definition at line 1038 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

Typedef Documentation

typedef Kokkos_MV_Multiply_Op KokkosMultiply

Definition at line 1007 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

Function Documentation

template<typename IntType >
IntType map_fem_graph_coord ( const IntType &  N,
const IntType &  i,
const IntType &  j,
const IntType &  k 
)
inline

Definition at line 28 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

template<typename ordinal >
ordinal generate_fem_graph ( ordinal  N,
std::vector< std::vector< ordinal > > &  graph 
)
inline

Definition at line 38 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

template<typename scalar , typename ordinal >
scalar generate_matrix_coefficient ( const ordinal  nFEM,
const ordinal  nStoch,
const ordinal  iRowFEM,
const ordinal  iColFEM,
const ordinal  iStoch 
)
inline

Definition at line 72 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

template<typename scalar , typename ordinal >
scalar generate_vector_coefficient ( const ordinal  nFEM,
const ordinal  nStoch,
const ordinal  iColFEM,
const ordinal  iStoch 
)
inline

Definition at line 89 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

template<typename kokkos_cijk_type , typename ordinal_type >
kokkos_cijk_type build_cijk ( ordinal_type  stoch_dim,
ordinal_type  poly_ord 
)

Definition at line 101 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

template<typename array_type , typename scalar_type >
bool compare_rank_2_views ( const array_type &  y,
const array_type &  y_exp,
const scalar_type  rel_tol,
const scalar_type  abs_tol,
Teuchos::FancyOStream &  out 
)

Definition at line 138 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

template<typename vector_type , typename scalar_type >
bool compareRank1 ( const vector_type &  y,
const vector_type &  y_exp,
const scalar_type  rel_tol,
const scalar_type  abs_tol,
Teuchos::FancyOStream &  out 
)

Definition at line 175 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

template<typename vector_type , typename scalar_type >
bool compareRank2 ( const vector_type &  y,
const vector_type &  y_exp,
const scalar_type  rel_tol,
const scalar_type  abs_tol,
Teuchos::FancyOStream &  out 
)

Definition at line 210 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

template<typename MatrixType , typename CijkType >
MatrixType buildDiagonalMatrix ( typename MatrixType::ordinal_type  nrow,
typename MatrixType::ordinal_type  pce_size,
const CijkType &  cijk 
)

Definition at line 254 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL ( Kokkos_CrsMatrix_PCE  ,
ReplaceValues  ,
MatrixScalar   
)

Definition at line 431 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL ( Kokkos_CrsMatrix_PCE  ,
SumIntoValues  ,
MatrixScalar   
)

Definition at line 457 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL ( Kokkos_CrsMatrix_PCE  ,
SumIntoValuesAtomic  ,
MatrixScalar   
)

Definition at line 483 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

template<typename PCEType , typename Multiply >
bool test_embedded_pce ( const typename PCEType::ordinal_type  nGrid,
const typename PCEType::ordinal_type  stoch_dim,
const typename PCEType::ordinal_type  poly_ord,
KokkosSparse::DeviceConfig  dev_config,
Multiply  multiply_op,
Teuchos::FancyOStream &  out 
)

Definition at line 510 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

TEUCHOS_UNIT_TEST_TEMPLATE_2_DECL ( Kokkos_CrsMatrix_PCE  ,
Multiply  ,
Scalar  ,
MultiplyOp   
)

Definition at line 668 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL ( Kokkos_CrsMatrix_PCE  ,
MeanMultiplyRank1  ,
Scalar   
)

Definition at line 704 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

TEUCHOS_UNIT_TEST_TEMPLATE_1_DECL ( Kokkos_CrsMatrix_PCE  ,
MeanMultiplyRank2  ,
Scalar   
)

Definition at line 855 of file Stokhos_KokkosCrsMatrixUQPCEUnitTest.hpp.

Generated on Fri Nov 22 2024 09:18:17 for Stokhos Package Browser (Single Doxygen Collection) by   doxygen 1.8.5