Question

我试图将数组元素与SIMD并行。为了避免锁定，我使用的是可组合线程local，它不总是在16字节上对齐因为_mm_add_epi32正在抛出异常

concurrency::combinable<__m128i> sum_combine;

int length = 40; // multiple of 8
concurrency::parallel_for(0, length , 8, [&](int it)
{

    __m128i v1 = _mm_load_si128(reinterpret_cast<__m128i*>(input_arr + it));
    __m128i v2 = _mm_load_si128(reinterpret_cast<__m128i*>(input_arr + it + sizeof(uint32_t)));

    auto temp = _mm_add_epi32(v1, v2);

    auto &sum = sum_combine.local();   // here is the problem 


    TRACE(L"%d\n", it);
    TRACE(L"add %x\n", &sum);

    ASSERT(((unsigned long)&sum & 15) == 0);

    sum = _mm_add_epi32(temp, sum);
}
);

这里是ppl.h中可组合的定义

template<typename _Ty>
class combinable
{
private:

// Disable warning C4324: structure was padded due to __declspec(align())
// This padding is expected and necessary.
#pragma warning(push)
#pragma warning(disable: 4324)
    __declspec(align(64))
    struct _Node
    {
        unsigned long _M_key;
        _Ty _M_value;                   // this might not be aligned on 16 bytes
        _Node* _M_chain;

        _Node(unsigned long _Key, _Ty _InitialValue)
            : _M_key(_Key), _M_value(_InitialValue), _M_chain(NULL)
        {
        }
    };

有时对齐是正常的，代码工作正常，但大多数时候它不能正常工作

我试过使用以下内容，但这不编译

union combine 
{
        unsigned short x[sizeof(__m128i) / sizeof(unsigned int)];
        __m128i y;
};

concurrency::combinable<combine> sum_combine;
then auto &sum = sum_combine.local().y;

有关更正对齐问题的任何建议，仍然使用可组合。

在x64上它可以正常工作，因为默认的16字节对齐。在x86上，有时存在对齐问题。

Answer 1

使用未对齐的载荷加载总和

auto &sum = sum_combine.local();


#if !defined(_M_X64) 

if (((unsigned long)&sum & 15) != 0)
{
    // just for breakpoint means, sum  is unaligned.
    int a = 5;
}
auto sum_temp = _mm_loadu_si128(&sum);
sum = _mm_add_epi32(temp, sum_temp);

#else

sum = _mm_add_epi32(temp, sum);

#endif

Answer 2

由于与sum一起使用的_mm_add_epi32变量未对齐，您需要使用未对齐的加载/存储（sum / {{1}显式加载/存储_mm_loadu_si128 }）。变化：

_mm_storeu_si128

为：

sum = _mm_add_epi32(temp, sum);

PPL可组合的{SIM卡对齐问题

2 个答案: