#include <iostream>
#include "KokkosSparse_CrsMatrix.hpp"
#include "KokkosSparse_spmv.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_spmm (const OrdinalType ensemble_length, const OrdinalType nGrid, const OrdinalType iterCount, std::vector< double > &scalar_perf, std::vector< double > &block_left_perf, std::vector< double > &block_right_perf) |
| |
| template<typename Scalar, typename Ordinal, typename Device> |
| void | performance_test_driver (const Ordinal nGrid, const Ordinal nIter, const Ordinal ensemble_min, const Ordinal ensemble_max, const Ordinal ensemble_step) |
| |
◆ 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_spmm()
template<typename ScalarType, typename OrdinalType, typename Device>
| void test_spmm |
( |
const OrdinalType | ensemble_length, |
|
|
const OrdinalType | nGrid, |
|
|
const OrdinalType | iterCount, |
|
|
std::vector< double > & | scalar_perf, |
|
|
std::vector< double > & | block_left_perf, |
|
|
std::vector< double > & | block_right_perf ) |
◆ performance_test_driver()
template<typename
Scalar, typename Ordinal, typename Device>
| void performance_test_driver |
( |
const Ordinal | nGrid, |
|
|
const Ordinal | nIter, |
|
|
const Ordinal | ensemble_min, |
|
|
const Ordinal | ensemble_max, |
|
|
const Ordinal | ensemble_step ) |