有没有办法在AVX2中重建_mm_slli_si128指令,以将__mm256i寄存器移位x个字节？
_mm256_slli_si256似乎只是在[127：0]和[255：128]上执行两个_mm_slli_si128.
左移应该在__m256i上工作,如下所示：
[1,2,3,4,5,6,7,8,9,10,11,12,...,32] -> [2,0]
我在thread年看到,可以用_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)),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,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,0) );
return _mm256_alignr_epi8(a,mask,16-N);
}
