AddReg.h

Go to the documentation of this file.

// AddReg.h:
// -------------------
//

#ifndef __IPSDKUTIL_INSTRUCTIONSET_ARITHMETIC_DETAIL_AVX2_ADDREG_H__
#define __IPSDKUTIL_INSTRUCTIONSET_ARITHMETIC_DETAIL_AVX2_ADDREG_H__

#include <IPSDKUtil/IPSDKUtilExports.h>
#include <IPSDKUtil/InstructionSet/AVX/AvxTypes.h>
#include <IPSDKUtil/InstructionSet/Arithmetic/detail/AddReg.h>
#include <IPSDKUtil/Tools/ForceInline.h>

namespace ipsdk {
namespace simd {
namespace detail {


template <typename T>
struct AddReg<eInstructionSet::eIS_Avx2, T, 
    typename boost::enable_if_c<boost::is_integral<T>::value
        && sizeof(T)==1>::type>
{
    static IPSDK_FORCEINLINE
    typename AvxType<T>::Type
    act(const typename AvxType<T>::Type& in1, const typename AvxType<T>::Type& in2)
    {
        return _mm256_add_epi8(in1, in2);
    }

    static IPSDK_FORCEINLINE
    void
    act(const typename AvxType<T>::Type& in1,
        const typename AvxType<T>::Type& in2,
        typename AvxType<T>::Type& out)
    {
        out = _mm256_add_epi8(in1, in2);
    }
};

template <typename T>
struct AddReg<eInstructionSet::eIS_Avx2, T, 
    typename boost::enable_if_c<boost::is_integral<T>::value
        && sizeof(T)==2>::type>
{
    static IPSDK_FORCEINLINE
    typename AvxType<T>::Type
    act(const typename AvxType<T>::Type& in1, const typename AvxType<T>::Type& in2)
    {
        return _mm256_add_epi16(in1, in2);
    }

    static IPSDK_FORCEINLINE
    void
    act(const typename AvxType<T>::Type& in1,
        const typename AvxType<T>::Type& in2,
        typename AvxType<T>::Type& out)
    {
        out = _mm256_add_epi16(in1, in2);
    }
};

template <typename T>
struct AddReg<eInstructionSet::eIS_Avx2, T, 
    typename boost::enable_if_c<boost::is_integral<T>::value
        && sizeof(T)==4>::type>
{
    static IPSDK_FORCEINLINE
    typename AvxType<T>::Type
    act(const typename AvxType<T>::Type& in1, const typename AvxType<T>::Type& in2)
    {
        return _mm256_add_epi32(in1, in2);
    }

    static IPSDK_FORCEINLINE
    void
    act(const typename AvxType<T>::Type& in1,
        const typename AvxType<T>::Type& in2,
        typename AvxType<T>::Type& out)
    {
        out = _mm256_add_epi32(in1, in2);
    }
};

template <typename T>
struct AddReg<eInstructionSet::eIS_Avx2, T, 
    typename boost::enable_if_c<boost::is_integral<T>::value
        && sizeof(T)==8>::type>
{
    static IPSDK_FORCEINLINE
    typename AvxType<T>::Type
    act(const typename AvxType<T>::Type& in1, const typename AvxType<T>::Type& in2)
    {
        return _mm256_add_epi64(in1, in2);
    }

    static IPSDK_FORCEINLINE
    void
    act(const typename AvxType<T>::Type& in1,
        const typename AvxType<T>::Type& in2,
        typename AvxType<T>::Type& out)
    {
        out = _mm256_add_epi64(in1, in2);
    }
};


template <>
struct AddReg<eInstructionSet::eIS_Avx2, ipReal32>
{
    static IPSDK_FORCEINLINE
    AvxType<ipReal32>::Type
    act(const AvxType<ipReal32>::Type& in1, const AvxType<ipReal32>::Type& in2)
    {
        return AddReg<eInstructionSet::eIS_Avx, ipReal32>::act(in1, in2);
    }

    static IPSDK_FORCEINLINE
    void
    act(const AvxType<ipReal32>::Type& in1,
        const AvxType<ipReal32>::Type& in2,
        AvxType<ipReal32>::Type& out)
    {
        AddReg<eInstructionSet::eIS_Avx, ipReal32>::act(in1, in2, out);
    }
};

template <>
struct AddReg<eInstructionSet::eIS_Avx2, ipReal64>
{
    static IPSDK_FORCEINLINE
    AvxType<ipReal64>::Type
    act(const AvxType<ipReal64>::Type& in1, const AvxType<ipReal64>::Type& in2)
    {
        return AddReg<eInstructionSet::eIS_Avx, ipReal64>::act(in1, in2);
    }

    static IPSDK_FORCEINLINE
    void
    act(const AvxType<ipReal64>::Type& in1,
        const AvxType<ipReal64>::Type& in2,
        AvxType<ipReal64>::Type& out)
    {
        AddReg<eInstructionSet::eIS_Avx, ipReal64>::act(in1, in2, out);
    }
};


} // end of namespace detail
} // end of namespace simd
} // end of namespace ipsdk

#endif // __IPSDKUTIL_INSTRUCTIONSET_ARITHMETIC_DETAIL_AVX2_ADDREG_H__