Есть ли способ обеспечить valarray
использует выровненную память, чтобы она могла быть векторизована с помощью SSE и AVX? Насколько я знаю, STL не гарантирует выравнивание, и вы можете не передать распределитель valarray. Есть ли другой способ добиться этого?
Спасибо вперед!
Я обычно использую std::vector
с моим собственным распределителем, который имеет выравнивание в качестве параметра шаблона и вызывает _mm_malloc()
или же _aligned_malloc()
, Это работает очень хорошо, также с AVX (32-байтовое выравнивание). Правильно написанный шаблонный пользовательский код автоматически выбирает необходимое выравнивание.
Ниже код для AlignmentAllocator<>
и помощник. протестировано под gcc и icpc.
/// allocate and de-allocate aligned memory
template<std::size_t alignment>
struct static_allocator {
static void*allocate(std::size_t n)
{
if(n == 0) return 0;
if(n > max_size())
throw std::bad_alloc();
void*ret =
#if defined(__GNUC__) || defined (__INTEL_COMPILER)
_mm_malloc
#else
_aligned_malloc
#endif
(n,alignment);
if(!ret)
throw std::bad_alloc();
return ret;
}
static void deallocate(void*p)
{
#if defined(__GNUC__) || defined (__INTEL_COMPILER)
_mm_free
#else
_aligned_free
#endif
(p);
}
static std::size_t max_size ()
{ return std::numeric_limits<std::size_t>::max(); }
};
/// allocate and de-allocate unaligned memory
template<>
struct static_allocator<1> {
static std::size_t max_size () noexcept
{ return std::numeric_limits<std::size_t>::max(); }
static void*allocate(std::size_t n)
{
if(n == 0) return 0;
void*ret = new char[n];
return ret;
}
static void deallocate(void*p)
{ delete[] static_cast<char*>(p); }
};
template<> struct static_allocator<0>;
/// allocator with explicit alignment
template<typename _Tp, std::size_t alignment = 16>
class AlignmentAllocator
{
typedef static_allocator<alignment> static_alloc;
public:
typedef size_t size_type;
typedef ptrdiff_t difference_type;
typedef _Tp* pointer;
typedef const _Tp* const_pointer;
typedef _Tp& reference;
typedef const _Tp& const_reference;
typedef _Tp value_type;
template <typename _Tp1>
struct rebind
{ typedef AlignmentAllocator<_Tp1, alignment> other; };
AlignmentAllocator() {}
AlignmentAllocator(const AlignmentAllocator&) {}
template <typename _Tp1>
AlignmentAllocator(const AlignmentAllocator<_Tp1, alignment> &) {}
~AlignmentAllocator() {}
pointer address (reference x) const
{
#if __cplusplus >= 201103L
return std::addressof(x);
#else
return reinterpret_cast<_Tp*>(&reinterpret_cast<char&>(x));
#endif
}
const_pointer address (const_reference x) const
{
#if __cplusplus >= 201103L
return std::addressof(x);
#else
return reinterpret_cast<const _Tp*>(&reinterpret_cast<const char&>(x));
#endif
}
pointer allocate (size_type n, const void* = 0)
{ return static_cast<pointer>(static_alloc::allocate(n*sizeof(value_type))); }
void deallocate (pointer p, size_type)
{ static_alloc::deallocate(p); }
size_type max_size () const
{ return static_alloc::max_size() / sizeof (value_type); }
#if __cplusplus >= 201103L
template<typename _Up, typename... _Args>
void construct(_Up* p, _Args&&... args)
{ ::new(static_cast<void*>(p)) _Up(std::forward<_Args>(args)...); }
template<typename _Up>
void destroy(_Up* p)
{ p->~_Up(); }
#else
void construct (pointer p, const_reference val)
{ ::new(static_cast<void*>(p)) value_type(val); }
void destroy (pointer p)
{ p->~value_type (); }
#endif
bool operator!=(const AlignmentAllocator&) const
{ return false; }
// Returns true if and only if storage allocated from *this
// can be deallocated from other, and vice versa.
// Always returns true for stateless allocators.
bool operator==(const AlignmentAllocator&) const
{ return true; }
};// class AlignmentAllocator<>
/// AlignmentAllocator<void> specialization.
template<std::size_t alignment>
class AlignmentAllocator<void, alignment>
{
public:
typedef size_t size_type;
typedef ptrdiff_t difference_type;
typedef void* pointer;
typedef const void* const_pointer;
typedef void value_type;
template<typename _Tp1>
struct rebind
{ typedef AlignmentAllocator<_Tp1, alignment> other; };
};
Других решений пока нет …