Jos*_*uer 5 c++ x86 sse simd intrinsics
我想在 8 位灰度图像中创建垂直梯度直方图。可以指定计算梯度的垂直距离。我已经设法使用 Intrinsics 加速了代码的另一部分,但它在这里不起作用。如果 _mm_store_si128 被注释掉,则代码运行不会出现异常。当它没有被注释时,我会遇到访问冲突。
这里出了什么问题?
#define _mm_absdiff_epu8(a,b) _mm_adds_epu8(_mm_subs_epu8(a, b), _mm_subs_epu8(b, a)) //from opencv
void CreateAbsDiffHistogramUnmanaged(void* source, unsigned int sourcestride, unsigned int height, unsigned int verticalDistance, unsigned int histogram[])
{
unsigned int xcount = sourcestride / 16;
__m128i absdiffData;
unsigned char* bytes = (unsigned char*) _aligned_malloc(16, 16);
__m128i* absdiffresult = (__m128i*) bytes;
__m128i* sourceM = (__m128i*) source;
__m128i* sourceVOffset = (__m128i*)source + verticalDistance * sourcestride;
for (unsigned int y = 0; y < (height - verticalDistance); y++)
{
for (unsigned int x = 0; x < xcount; x++, ++sourceM, ++sourceVOffset)
{
absdiffData = _mm_absdiff_epu8(*sourceM, *sourceVOffset);
_mm_store_si128(absdiffresult, absdiffData);
//unroll loop
histogram[bytes[0]]++;
histogram[bytes[1]]++;
histogram[bytes[2]]++;
histogram[bytes[3]]++;
histogram[bytes[4]]++;
histogram[bytes[5]]++;
histogram[bytes[6]]++;
histogram[bytes[7]]++;
histogram[bytes[8]]++;
histogram[bytes[9]]++;
histogram[bytes[10]]++;
histogram[bytes[11]]++;
histogram[bytes[12]]++;
histogram[bytes[13]]++;
histogram[bytes[14]]++;
histogram[bytes[15]]++;
}
}
_aligned_free(bytes);
}
Run Code Online (Sandbox Code Playgroud)
由于输入数据未正确对齐,您的函数在加载时崩溃了。为了解决这个问题,您必须更改代码:
absdiffData = _mm_absdiff_epu8(*sourceM, *sourceVOffset);
Run Code Online (Sandbox Code Playgroud)
到:
absdiffData = _mm_absdiff_epu8(_mm_loadu_si128(sourceM), _mm_loadu_si128(sourceVOffset));
Run Code Online (Sandbox Code Playgroud)
这里我使用未对齐加载。
PS我已经在Simd库中实现了类似的函数( SimdAbsSecondDerivativeHistogram) 。它有SSE2、AVX2、NEON和Altivec实现。我希望它能帮助你。
PPS 另外我强烈建议检查这一行:
__m128i* sourceVOffset = (__m128i*)source + verticalDistance * sourcestride);
Run Code Online (Sandbox Code Playgroud)
它可能会导致崩溃(访问输入数组边界之外的内存)。也许你想到过这个:
__m128i* sourceVOffset = (__m128i*)((char*)source + verticalDistance * sourcestride);
Run Code Online (Sandbox Code Playgroud)