Stokhos Package Browser (Single Doxygen Collection)
Version of the Day
Toggle main menu visibility
Loading...
Searching...
No Matches
src
kokkos
Cuda
Stokhos_Cuda_FlatSparse3Tensor.hpp
Go to the documentation of this file.
1
// @HEADER
2
// ***********************************************************************
3
//
4
// Stokhos Package
5
// Copyright (2009) Sandia Corporation
6
//
7
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
8
// license for use of this work by or on behalf of the U.S. Government.
9
//
10
// Redistribution and use in source and binary forms, with or without
11
// modification, are permitted provided that the following conditions are
12
// met:
13
//
14
// 1. Redistributions of source code must retain the above copyright
15
// notice, this list of conditions and the following disclaimer.
16
//
17
// 2. Redistributions in binary form must reproduce the above copyright
18
// notice, this list of conditions and the following disclaimer in the
19
// documentation and/or other materials provided with the distribution.
20
//
21
// 3. Neither the name of the Corporation nor the names of the
22
// contributors may be used to endorse or promote products derived from
23
// this software without specific prior written permission.
24
//
25
// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36
//
37
// Questions? Contact Eric T. Phipps (etphipp@sandia.gov).
38
//
39
// ***********************************************************************
40
// @HEADER
41
42
#ifndef STOKHOS_CUDA_FLAT_SPARSE_3_TENSOR_HPP
43
#define STOKHOS_CUDA_FLAT_SPARSE_3_TENSOR_HPP
44
45
#include <iostream>
46
47
#include "Kokkos_Core.hpp"
48
49
#include "
Stokhos_Multiply.hpp
"
50
#include "
Stokhos_BlockCrsMatrix.hpp
"
51
#include "
Stokhos_FlatSparse3Tensor.hpp
"
52
53
#include "cuda_profiler_api.h"
54
55
namespace
Stokhos
{
56
57
//----------------------------------------------------------------------------
58
59
template
<
typename
TensorScalar ,
60
typename
MatrixScalar ,
61
typename
VectorScalar >
62
class
Multiply
<
63
BlockCrsMatrix
<
FlatSparse3Tensor
< TensorScalar,
Kokkos
::Cuda >,
64
MatrixScalar, Kokkos::Cuda >,
65
Kokkos::View<VectorScalar**, Kokkos::LayoutLeft, Kokkos::Cuda>,
66
Kokkos::View<VectorScalar**, Kokkos::LayoutLeft, Kokkos::Cuda> >
67
{
68
public
:
69
70
typedef
Kokkos::Cuda
execution_space
;
71
typedef
execution_space::size_type
size_type
;
72
73
typedef
FlatSparse3Tensor< TensorScalar , execution_space >
tensor_type
;
74
typedef
BlockCrsMatrix< tensor_type, MatrixScalar, execution_space >
matrix_type
;
75
typedef
Kokkos::View< VectorScalar** ,
76
Kokkos::LayoutLeft ,
77
Kokkos::Cuda >
vector_type
;
78
79
80
81
//------------------------------------
82
83
static
void
apply
(
const
matrix_type
& A ,
84
const
vector_type
& x ,
85
const
vector_type
& y )
86
{
87
}
88
};
89
90
//----------------------------------------------------------------------------
91
//----------------------------------------------------------------------------
92
93
}
// namespace Stokhos
94
95
#endif
/* #ifndef STOKHOS_CUDA_FLAT_SPARSE_3_TENSOR_HPP */
Stokhos_BlockCrsMatrix.hpp
Stokhos_FlatSparse3Tensor.hpp
Stokhos_Multiply.hpp
Stokhos::BlockCrsMatrix
CRS matrix of dense blocks.
Definition
Stokhos_BlockCrsMatrix.hpp:61
Stokhos::FlatSparse3Tensor
Sparse product tensor with replicated entries to provide subsets with a given coordinate.
Definition
Stokhos_FlatSparse3Tensor.hpp:62
Stokhos::Multiply< BlockCrsMatrix< FlatSparse3Tensor< TensorScalar, Kokkos::Cuda >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::matrix_type
BlockCrsMatrix< tensor_type, MatrixScalar, execution_space > matrix_type
Definition
Stokhos_Cuda_FlatSparse3Tensor.hpp:74
Stokhos::Multiply< BlockCrsMatrix< FlatSparse3Tensor< TensorScalar, Kokkos::Cuda >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::apply
static void apply(const matrix_type &A, const vector_type &x, const vector_type &y)
Definition
Stokhos_Cuda_FlatSparse3Tensor.hpp:83
Stokhos::Multiply< BlockCrsMatrix< FlatSparse3Tensor< TensorScalar, Kokkos::Cuda >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::tensor_type
FlatSparse3Tensor< TensorScalar, execution_space > tensor_type
Definition
Stokhos_Cuda_FlatSparse3Tensor.hpp:73
Stokhos::Multiply< BlockCrsMatrix< FlatSparse3Tensor< TensorScalar, Kokkos::Cuda >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::size_type
execution_space::size_type size_type
Definition
Stokhos_Cuda_FlatSparse3Tensor.hpp:71
Stokhos::Multiply< BlockCrsMatrix< FlatSparse3Tensor< TensorScalar, Kokkos::Cuda >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::execution_space
Kokkos::Cuda execution_space
Definition
Stokhos_Cuda_FlatSparse3Tensor.hpp:70
Stokhos::Multiply< BlockCrsMatrix< FlatSparse3Tensor< TensorScalar, Kokkos::Cuda >, MatrixScalar, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda >, Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > >::vector_type
Kokkos::View< VectorScalar **, Kokkos::LayoutLeft, Kokkos::Cuda > vector_type
Definition
Stokhos_Cuda_FlatSparse3Tensor.hpp:77
Stokhos::Multiply
Definition
Stokhos_Multiply.hpp:96
Kokkos
Definition
Stokhos_CrsMatrix.hpp:663
Stokhos
Top-level namespace for Stokhos classes and functions.
Definition
Stokhos_AbstractPreconditionerFactory.hpp:48
Generated by
1.17.0