SetImpl.h

Go to the documentation of this file.

// SetImpl.h:
// ------------
//

#ifndef __IPSDKUTIL_INSTRUCTIONSET_DETAIL_AVX512_SETIMPL_H__
#define __IPSDKUTIL_INSTRUCTIONSET_DETAIL_AVX512_SETIMPL_H__

#include <IPSDKUtil/IPSDKUtilExports.h>
#include <IPSDKUtil/Tools/ForceInline.h>
#include <IPSDKUtil/InstructionSet/unload.h>

namespace ipsdk {
namespace simd {
namespace detail {


template <typename T>
struct SetImpl<eInstructionSet::eIS_Avx512, T> {
    static IPSDK_FORCEINLINE
    void act(T* buffer, T value, ipUInt64 nbElts)
    {
        const ipUInt64 nbAlignedElts =
            nbElts - (nbElts%NbEltsPerReg<ePackType::ePT_Avx512, T>::Value);
        typedef typename Avx512Type<T>::Type RegT;
        RegT reg;
        AssignReg<eInstructionSet::eIS_Avx512, T>::act(reg, value);

        for(ipUInt64 i=0; i<nbAlignedElts; i+=NbEltsPerReg<ePackType::ePT_Avx512, T>::Value)
            UnloadReg<eInstructionSet::eIS_Avx512, T>::act(reg, buffer + i);

        for(ipUInt64 i=nbAlignedElts; i<nbElts; ++i)
            buffer[i] = value;
    }
};


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

#endif // __IPSDKUTIL_INSTRUCTIONSET_DETAIL_AVX512_SETIMPL_H__