-
Notifications
You must be signed in to change notification settings - Fork 99
BLAS 1::dot
Christian Trott edited this page Jan 22, 2018
·
4 revisions
Header File: KokkosBlas1_dot.hpp
Usage: result = KokkosBlas::dot(x,y);
KokkosBlas::dot(r,x,y);
Multiplies each value of x(i)
[x(i,j)
] with y(i)
or [y(i,j)
] and computes the total.
template<class VectorX, class VectorY>
InnerProductSpaceTraits<VectorX::non_const_value_type>::dot_type
dot (const VectorX& X, const VectorY& Y);
- VectorX: A rank-1
Kokkos::View
- VectorY: A rank-1
Kokkos::View
-
Y.rank == 1
orX.rank == 1
Y.extent(0) == X.extent(0)
template<class ReturnVector, class VectorX, class VectorY>
void dot (const ReturnVector& R, const VectorX& X, const VectorY& Y);
- ReturnVector: A rank-0 or rank-1
Kokkos::View
- VectorX: A rank-1 or rank-2
Kokkos::View
- VectorY: A rank-1 or rank-2
Kokkos::View
X.rank == Y.rank
X.rank == R.rank + 1
Y.extent(0) == X.extent(0)
Y.extent(1) == X.extent(1)
R.extent(0) == X.extent(1)
ReturnVector::non_const_value_type == ReturnVector::value_type
#include<Kokkos_Core.hpp>
#include<KokkosBlas1_dot.hpp>
int main(int argc, char* argv[]) {
Kokkos::initialize();
int N = atoi(argv[1]);
Kokkos::View<double*> x("X",N);
Kokkos::View<double*> y("Y",N);
Kokkos::deep_copy(x,3.0);
Kokkos::deep_copy(y,2.0);
double x_y = KokkosBlas::dot(x,y);
printf("X_dot_Y: %lf Expected: %lf Diff: %e\n",x_y,1.0*N*(3.0 * 2.0),x_y-1.0*N(3.0 * 2.0));
Kokkos::finalize();
}