[英]8 bit shift operation in AVX2 with shifting in zeros
有什么方法可以重建 AVX2 中的_mm_slli_si128
指令以將__mm256i
寄存器移位 x 字節?
_mm256_slli_si256
似乎只是在 a[127:0] 和 a[255:128] 上執行兩個_mm_slli_si128
。
左移應該像這樣在__m256i
上工作:
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, ..., 32] -> [2, 3, 4, 5, 6, 7, 8, 9, ..., 0]
我在線程中看到可以使用_mm256_permutevar8x32_ps
為 32 位創建移位。 但我需要一個更通用的解決方案來移動 x 字節。 有沒有人已經解決了這個問題?
好的,我實現了一個可以左移最多 16 個字節的函數。
template <unsigned int N> __m256i _mm256_shift_left(__m256i a)
{
__m256i mask = _mm256_srli_si256(
_mm256_permute2x128_si256(a, a, _MM_SHUFFLE(0,0,3,0))
, 16-N);
return _mm256_or_si256(_mm256_slli_si256(a,N),mask);
}
例子:
int main(int argc, char* argv[]) {
__m256i reg = _mm256_set_epi8(32,31,30,29,28,27,26,25,24,23,22,21,20,19,18,17,16,15,
14,13,12,11,10,9,8,7,6,5,4,3,2,1);
__m256i result = _mm256_shift_left<1>(reg);
for(int i = 0; i < 32; i++)
printf("%2d ",((unsigned char *)&result)[i]);
printf("\n");
}
輸出是0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 0 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
編輯:帶有新對齊器指令的新版本。 感謝您的提示@Evgney Kluev
template <unsigned int N> __m256i _mm256_shift_left(__m256i a)
{
__m256i mask = _mm256_permute2x128_si256(a, a, _MM_SHUFFLE(0,0,3,0) );
return _mm256_alignr_epi8(a,mask,16-N);
}
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.