Stokhos Package Browser (Single Doxygen Collection) Version of the Day
Loading...
Searching...
No Matches
Functions
TestSpMM.hpp File Reference
#include <iostream>
#include "KokkosSparse_CrsMatrix.hpp"
#include "KokkosSparse_spmv.hpp"
#include "Kokkos_Timer.hpp"
Include dependency graph for TestSpMM.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions

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)
 

Function Documentation

◆ 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

Definition at line 53 of file TestSpMM.hpp.

◆ generate_fem_graph()

size_t generate_fem_graph ( size_t N,
std::vector< std::vector< size_t > > & graph )
inline

Definition at line 62 of file TestSpMM.hpp.

◆ 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 )

Definition at line 96 of file TestSpMM.hpp.

◆ 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 )

Definition at line 242 of file TestSpMM.hpp.