#include <iostream>
#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_TotalOrderBasis.hpp"
#include "Stokhos_Sparse3Tensor.hpp"
#include "Kokkos_Timer.hpp"
Go to the source code of this file.
|
| template<typename IntType> |
| IntType | map_fem_graph_coord (const IntType &N, const IntType &i, const IntType &j, const IntType &k) |
| |
| size_t | generate_fem_graph (size_t N, std::vector< std::vector< size_t > > &graph) |
| |
| template<typename ScalarType, typename OrdinalType, typename Device> |
| void | test_mean_multiply (const OrdinalType order, const OrdinalType dim, const OrdinalType nGrid, const OrdinalType iterCount, std::vector< double > &scalar_perf, std::vector< double > &block_left_perf, std::vector< double > &block_right_perf, std::vector< double > &pce_perf, std::vector< double > &block_pce_perf) |
| |
| template<typename Scalar, typename Ordinal, typename Device> |
| void | performance_test_driver (const Ordinal nGrid, const Ordinal nIter, const Ordinal order, const Ordinal min_var, const Ordinal max_var) |
| |
◆ INST_PERF_DRIVER
| #define INST_PERF_DRIVER |
( |
| SCALAR, |
|
|
| ORDINAL, |
|
|
| DEVICE ) |
Value:
const ORDINAL nGrid, const ORDINAL nIter, const ORDINAL order, \
const ORDINAL min_var, const ORDINAL max_var);
void performance_test_driver(const Teuchos::RCP< const Teuchos::Comm< int > > &comm, const int use_print, const int use_trials, const int use_nodes[], const bool check, Kokkos::Example::FENL::DeviceConfig dev_config)
Definition at line 433 of file TestMeanMultiply.hpp.
◆ map_fem_graph_coord()
template<typename IntType>
| IntType map_fem_graph_coord |
( |
const IntType & | N, |
|
|
const IntType & | i, |
|
|
const IntType & | j, |
|
|
const IntType & | k ) |
|
inline |
◆ generate_fem_graph()
| size_t generate_fem_graph |
( |
size_t | N, |
|
|
std::vector< std::vector< size_t > > & | graph ) |
|
inline |
◆ test_mean_multiply()
template<typename ScalarType, typename OrdinalType, typename Device>
| void test_mean_multiply |
( |
const OrdinalType | order, |
|
|
const OrdinalType | dim, |
|
|
const OrdinalType | nGrid, |
|
|
const OrdinalType | iterCount, |
|
|
std::vector< double > & | scalar_perf, |
|
|
std::vector< double > & | block_left_perf, |
|
|
std::vector< double > & | block_right_perf, |
|
|
std::vector< double > & | pce_perf, |
|
|
std::vector< double > & | block_pce_perf ) |
◆ performance_test_driver()
template<typename
Scalar, typename Ordinal, typename Device>
| void performance_test_driver |
( |
const Ordinal | nGrid, |
|
|
const Ordinal | nIter, |
|
|
const Ordinal | order, |
|
|
const Ordinal | min_var, |
|
|
const Ordinal | max_var ) |