GPUFunctionTable.h

Go to the documentation of this file.

#ifndef included_AMP_GPUFunctionTable
#define included_AMP_GPUFunctionTable
 
 
#include "AMP/utils/ArraySize.h"
 
#include <memory>
 
 
namespace AMP {
 
 
template<class TYPE>
class GPUFunctionTable final
{
public:
    typedef TYPE value_type;
    template<class TYPE2>
    using cloneTo = GPUFunctionTable<TYPE2>;
 
public:
    static void rand( size_t N, TYPE *x );
 
    template<typename LAMBDA>
    static TYPE reduce( LAMBDA &op, size_t N, const TYPE *A, TYPE initialValue )
    {
        AMP_ERROR( "Not implemented for GPU" );
    }
 
    template<typename LAMBDA>
    static TYPE reduce( LAMBDA &op, size_t N, const TYPE *A, const TYPE *B, TYPE initialValue )
    {
        AMP_ERROR( "Not implemented for GPU" );
    }
 
    template<typename LAMBDA>
    static void transform( LAMBDA &fun, size_t N, const TYPE *x, TYPE *y )
    {
        AMP_ERROR( "Not implemented for GPU" );
    }
 
    template<typename LAMBDA>
    static void transform( LAMBDA &fun, size_t N, const TYPE *x, const TYPE *y, TYPE *z )
    {
        AMP_ERROR( "Not implemented for GPU" );
    }
 
    static TYPE min( size_t N, const TYPE *x );
 
    static TYPE max( size_t N, const TYPE *x );
 
    static TYPE sum( size_t N, const TYPE *x );
 
    static void multiply( const ArraySize &sa,
                          const TYPE *a,
                          const ArraySize &sb,
                          const TYPE *b,
                          const ArraySize &sc,
                          TYPE *c );
 
    static void scale( size_t N, TYPE x, TYPE *y );
 
    static void px( size_t N, TYPE x, TYPE *y );
 
    static void px( size_t N, const TYPE *x, TYPE *y );
 
    static void mx( size_t N, TYPE x, TYPE *y );
 
    static void mx( size_t N, const TYPE *x, TYPE *y );
 
    static void axpy( TYPE alpha, size_t N, const TYPE *x, TYPE *y );
 
    static void axpby( TYPE alpha, size_t N, const TYPE *x, TYPE beta, TYPE *y );
 
    static bool equals( size_t N, const TYPE *A, const TYPE *B, TYPE tol );
 
 
    /* Specialized Functions */
 
    static void transformReLU( size_t N, const TYPE *A, TYPE *B );
 
    static void transformAbs( size_t N, const TYPE *A, TYPE *B );
 
    static void transformTanh( size_t N, const TYPE *A, TYPE *B );
 
    static void transformHardTanh( size_t N, const TYPE *A, TYPE *B );
 
    static void transformSigmoid( size_t N, const TYPE *A, TYPE *B );
 
    static void transformSoftPlus( size_t N, const TYPE *A, TYPE *B );
 
private:
    GPUFunctionTable() = delete;
};
} // namespace AMP
 
 
#endif