git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@13583 f3b2605a-c512-4ea7-a41b-209d697bcdaa
This commit is contained in:
sjplimp
309
lib/kokkos/core/perf_test/PerfTestBlasKernels.hpp
Executable file
309
lib/kokkos/core/perf_test/PerfTestBlasKernels.hpp
Executable file
@ -0,0 +1,309 @@
|
||||
/*
|
||||
//@HEADER
|
||||
// ************************************************************************
|
||||
//
|
||||
// Kokkos v. 2.0
|
||||
// Copyright (2014) Sandia Corporation
|
||||
//
|
||||
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
|
||||
// the U.S. Government retains certain rights in this software.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// 1. Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
//
|
||||
// 2. Redistributions in binary form must reproduce the above copyright
|
||||
// notice, this list of conditions and the following disclaimer in the
|
||||
// documentation and/or other materials provided with the distribution.
|
||||
//
|
||||
// 3. Neither the name of the Corporation nor the names of the
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
|
||||
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
|
||||
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
//
|
||||
// Questions? Contact H. Carter Edwards (hcedwar@sandia.gov)
|
||||
//
|
||||
// ************************************************************************
|
||||
//@HEADER
|
||||
*/
|
||||
|
||||
#ifndef KOKKOS_BLAS_KERNELS_HPP
|
||||
#define KOKKOS_BLAS_KERNELS_HPP
|
||||
|
||||
namespace Kokkos {
|
||||
|
||||
template< class ConstVectorType ,
|
||||
class Device = typename ConstVectorType::execution_space >
|
||||
struct Dot ;
|
||||
|
||||
template< class ConstVectorType ,
|
||||
class Device = typename ConstVectorType::execution_space >
|
||||
struct DotSingle ;
|
||||
|
||||
template< class ConstScalarType ,
|
||||
class VectorType ,
|
||||
class Device = typename VectorType::execution_space >
|
||||
struct Scale ;
|
||||
|
||||
template< class ConstScalarType ,
|
||||
class ConstVectorType ,
|
||||
class VectorType ,
|
||||
class Device = typename VectorType::execution_space >
|
||||
struct AXPBY ;
|
||||
|
||||
/** \brief Y = alpha * X + beta * Y */
|
||||
template< class ConstScalarType ,
|
||||
class ConstVectorType ,
|
||||
class VectorType >
|
||||
void axpby( const ConstScalarType & alpha ,
|
||||
const ConstVectorType & X ,
|
||||
const ConstScalarType & beta ,
|
||||
const VectorType & Y )
|
||||
{
|
||||
typedef AXPBY< ConstScalarType , ConstVectorType , VectorType > functor ;
|
||||
|
||||
parallel_for( Y.dimension_0() , functor( alpha , X , beta , Y ) );
|
||||
}
|
||||
|
||||
/** \brief Y *= alpha */
|
||||
template< class ConstScalarType ,
|
||||
class VectorType >
|
||||
void scale( const ConstScalarType & alpha , const VectorType & Y )
|
||||
{
|
||||
typedef Scale< ConstScalarType , VectorType > functor ;
|
||||
|
||||
parallel_for( Y.dimension_0() , functor( alpha , Y ) );
|
||||
}
|
||||
|
||||
template< class ConstVectorType ,
|
||||
class Finalize >
|
||||
void dot( const ConstVectorType & X ,
|
||||
const ConstVectorType & Y ,
|
||||
const Finalize & finalize )
|
||||
{
|
||||
typedef Dot< ConstVectorType > functor ;
|
||||
|
||||
parallel_reduce( X.dimension_0() , functor( X , Y ) , finalize );
|
||||
}
|
||||
|
||||
template< class ConstVectorType ,
|
||||
class Finalize >
|
||||
void dot( const ConstVectorType & X ,
|
||||
const Finalize & finalize )
|
||||
{
|
||||
typedef DotSingle< ConstVectorType > functor ;
|
||||
|
||||
parallel_reduce( X.dimension_0() , functor( X ) , finalize );
|
||||
}
|
||||
|
||||
} /* namespace Kokkos */
|
||||
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
//----------------------------------------------------------------------------
|
||||
|
||||
namespace Kokkos {
|
||||
|
||||
template< class Type , class Device >
|
||||
struct Dot
|
||||
{
|
||||
typedef typename Device::execution_space execution_space ;
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< Impl::unsigned_< 1 > ,
|
||||
Impl::unsigned_< Type::Rank > >::type ok_rank ;
|
||||
|
||||
|
||||
/* typedef typename
|
||||
Impl::StaticAssertSame< execution_space ,
|
||||
typename Type::execution_space >::type ok_device ;*/
|
||||
|
||||
typedef double value_type ;
|
||||
|
||||
#if 1
|
||||
typename Type::const_type X ;
|
||||
typename Type::const_type Y ;
|
||||
#else
|
||||
Type X ;
|
||||
Type Y ;
|
||||
#endif
|
||||
|
||||
Dot( const Type & arg_x , const Type & arg_y )
|
||||
: X(arg_x) , Y(arg_y) { }
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void operator()( int i , value_type & update ) const
|
||||
{ update += X[i] * Y[i]; }
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
static void join( volatile value_type & update ,
|
||||
const volatile value_type & source )
|
||||
{ update += source; }
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
static void init( value_type & update )
|
||||
{ update = 0 ; }
|
||||
};
|
||||
|
||||
template< class Type , class Device >
|
||||
struct DotSingle
|
||||
{
|
||||
typedef typename Device::execution_space execution_space ;
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< Impl::unsigned_< 1 > ,
|
||||
Impl::unsigned_< Type::Rank > >::type ok_rank ;
|
||||
|
||||
/* typedef typename
|
||||
Impl::StaticAssertSame< execution_space ,
|
||||
typename Type::execution_space >::type ok_device ;*/
|
||||
|
||||
typedef double value_type ;
|
||||
|
||||
#if 1
|
||||
typename Type::const_type X ;
|
||||
#else
|
||||
Type X ;
|
||||
#endif
|
||||
|
||||
DotSingle( const Type & arg_x ) : X(arg_x) {}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void operator()( int i , value_type & update ) const
|
||||
{
|
||||
const typename Type::value_type & x = X[i]; update += x * x ;
|
||||
}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
static void join( volatile value_type & update ,
|
||||
const volatile value_type & source )
|
||||
{ update += source; }
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
static void init( value_type & update )
|
||||
{ update = 0 ; }
|
||||
};
|
||||
|
||||
|
||||
template< class ScalarType , class VectorType , class Device>
|
||||
struct Scale
|
||||
{
|
||||
typedef typename Device::execution_space execution_space ;
|
||||
|
||||
/* typedef typename
|
||||
Impl::StaticAssertSame< execution_space ,
|
||||
typename ScalarType::execution_space >::type
|
||||
ok_scalar_device ;
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< execution_space ,
|
||||
typename VectorType::execution_space >::type
|
||||
ok_vector_device ;*/
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< Impl::unsigned_< 0 > ,
|
||||
Impl::unsigned_< ScalarType::Rank > >::type
|
||||
ok_scalar_rank ;
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< Impl::unsigned_< 1 > ,
|
||||
Impl::unsigned_< VectorType::Rank > >::type
|
||||
ok_vector_rank ;
|
||||
|
||||
#if 1
|
||||
typename ScalarType::const_type alpha ;
|
||||
#else
|
||||
ScalarType alpha ;
|
||||
#endif
|
||||
|
||||
VectorType Y ;
|
||||
|
||||
Scale( const ScalarType & arg_alpha , const VectorType & arg_Y )
|
||||
: alpha( arg_alpha ), Y( arg_Y ) {}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void operator()( int i ) const
|
||||
{
|
||||
Y[i] *= alpha() ;
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
template< class ScalarType ,
|
||||
class ConstVectorType ,
|
||||
class VectorType,
|
||||
class Device>
|
||||
struct AXPBY
|
||||
{
|
||||
typedef typename Device::execution_space execution_space ;
|
||||
|
||||
/* typedef typename
|
||||
Impl::StaticAssertSame< execution_space ,
|
||||
typename ScalarType::execution_space >::type
|
||||
ok_scalar_device ;
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< execution_space ,
|
||||
typename ConstVectorType::execution_space >::type
|
||||
ok_const_vector_device ;
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< execution_space ,
|
||||
typename VectorType::execution_space >::type
|
||||
ok_vector_device ;*/
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< Impl::unsigned_< 0 > ,
|
||||
Impl::unsigned_< ScalarType::Rank > >::type
|
||||
ok_scalar_rank ;
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< Impl::unsigned_< 1 > ,
|
||||
Impl::unsigned_< ConstVectorType::Rank > >::type
|
||||
ok_const_vector_rank ;
|
||||
|
||||
typedef typename
|
||||
Impl::StaticAssertSame< Impl::unsigned_< 1 > ,
|
||||
Impl::unsigned_< VectorType::Rank > >::type
|
||||
ok_vector_rank ;
|
||||
|
||||
#if 1
|
||||
typename ScalarType::const_type alpha , beta ;
|
||||
typename ConstVectorType::const_type X ;
|
||||
#else
|
||||
ScalarType alpha , beta ;
|
||||
ConstVectorType X ;
|
||||
#endif
|
||||
|
||||
VectorType Y ;
|
||||
|
||||
AXPBY( const ScalarType & arg_alpha ,
|
||||
const ConstVectorType & arg_X ,
|
||||
const ScalarType & arg_beta ,
|
||||
const VectorType & arg_Y )
|
||||
: alpha( arg_alpha ), beta( arg_beta ), X( arg_X ), Y( arg_Y ) {}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void operator()( int i ) const
|
||||
{
|
||||
Y[i] = alpha() * X[i] + beta() * Y[i] ;
|
||||
}
|
||||
};
|
||||
|
||||
} /* namespace Kokkos */
|
||||
|
||||
#endif /* #ifndef KOKKOS_BLAS_KERNELS_HPP */
|
||||
Reference in New Issue
Block a user