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

template<typename IntType >

IntType map_fem_graph_coord ( const IntType &  N, const IntType &  i, const IntType &  j, const IntType &  k  )

inline

Definition at line 22 of file TestSpMM.hpp.

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

inline

Definition at line 31 of file TestSpMM.hpp.

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 65 of file TestSpMM.hpp.

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 211 of file TestSpMM.hpp.