Stokhos Package Browser (Single Doxygen Collection) Version of the Day
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
Stokhos::Multiply< BlockCrsMatrix< CooProductTensor< TensorScalar, Kokkos::Cuda, Pack >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::CooKernel Class Reference

#include <Stokhos_Cuda_CooProductTensor.hpp>

Public Member Functions

 CooKernel (const matrix_type &A, const vector_type &x, const vector_type &y, const size_type entries_per_thread, const size_type block_size)
 
__device__ void operator() (void) const
 

Public Attributes

const matrix_type m_A
 
const vector_type m_x
 
const vector_type m_y
 
const size_type m_block_size
 
const size_type m_entries_per_thread
 

Detailed Description

template<typename TensorScalar, typename MatrixScalar, typename VectorScalar, bool Pack>
class Stokhos::Multiply< BlockCrsMatrix< CooProductTensor< TensorScalar, Kokkos::Cuda, Pack >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::CooKernel

Definition at line 81 of file Stokhos_Cuda_CooProductTensor.hpp.

Constructor & Destructor Documentation

◆ CooKernel()

template<typename TensorScalar , typename MatrixScalar , typename VectorScalar , bool Pack>
Stokhos::Multiply< BlockCrsMatrix< CooProductTensor< TensorScalar, Kokkos::Cuda, Pack >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::CooKernel::CooKernel ( const matrix_type A,
const vector_type x,
const vector_type y,
const size_type  entries_per_thread,
const size_type  block_size 
)
inline

Definition at line 90 of file Stokhos_Cuda_CooProductTensor.hpp.

Member Function Documentation

◆ operator()()

template<typename TensorScalar , typename MatrixScalar , typename VectorScalar , bool Pack>
__device__ void Stokhos::Multiply< BlockCrsMatrix< CooProductTensor< TensorScalar, Kokkos::Cuda, Pack >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::CooKernel::operator() ( void  ) const
inline

Definition at line 102 of file Stokhos_Cuda_CooProductTensor.hpp.

Member Data Documentation

◆ m_A

template<typename TensorScalar , typename MatrixScalar , typename VectorScalar , bool Pack>
const matrix_type Stokhos::Multiply< BlockCrsMatrix< CooProductTensor< TensorScalar, Kokkos::Cuda, Pack >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::CooKernel::m_A

Definition at line 84 of file Stokhos_Cuda_CooProductTensor.hpp.

◆ m_x

template<typename TensorScalar , typename MatrixScalar , typename VectorScalar , bool Pack>
const vector_type Stokhos::Multiply< BlockCrsMatrix< CooProductTensor< TensorScalar, Kokkos::Cuda, Pack >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::CooKernel::m_x

Definition at line 85 of file Stokhos_Cuda_CooProductTensor.hpp.

◆ m_y

template<typename TensorScalar , typename MatrixScalar , typename VectorScalar , bool Pack>
const vector_type Stokhos::Multiply< BlockCrsMatrix< CooProductTensor< TensorScalar, Kokkos::Cuda, Pack >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::CooKernel::m_y

Definition at line 86 of file Stokhos_Cuda_CooProductTensor.hpp.

◆ m_block_size

template<typename TensorScalar , typename MatrixScalar , typename VectorScalar , bool Pack>
const size_type Stokhos::Multiply< BlockCrsMatrix< CooProductTensor< TensorScalar, Kokkos::Cuda, Pack >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::CooKernel::m_block_size

Definition at line 87 of file Stokhos_Cuda_CooProductTensor.hpp.

◆ m_entries_per_thread

template<typename TensorScalar , typename MatrixScalar , typename VectorScalar , bool Pack>
const size_type Stokhos::Multiply< BlockCrsMatrix< CooProductTensor< TensorScalar, Kokkos::Cuda, Pack >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::CooKernel::m_entries_per_thread

Definition at line 88 of file Stokhos_Cuda_CooProductTensor.hpp.


The documentation for this class was generated from the following file: