Emulating shifts on 32 bytes with AVX

后端 未结 3 428
忘了有多久
忘了有多久 2020-11-29 11:34

I am migrating vectorized code written using SSE2 intrinsics to AVX2 intrinsics.

Much to my disappointment, I discover that the shift instructions _mm256_sll

3条回答
  •  没有蜡笔的小新
    2020-11-29 11:46

    From different inputs, I gathered these solutions. The key to crossing the inter-lane barrier is the align instruction, _mm256_alignr_epi8.

    _mm256_slli_si256(A, N)

    0 < N < 16

    _mm256_alignr_epi8(A, _mm256_permute2x128_si256(A, A, _MM_SHUFFLE(0, 0, 2, 0)), 16 - N)
    

    N = 16

    _mm256_permute2x128_si256(A, A, _MM_SHUFFLE(0, 0, 2, 0))
    

    16 < N < 32

    _mm256_slli_si256(_mm256_permute2x128_si256(A, A, _MM_SHUFFLE(0, 0, 2, 0)), N - 16)
    

    _mm256_srli_si256(A, N)

    0 < N < 16

    _mm256_alignr_epi8(_mm256_permute2x128_si256(A, A, _MM_SHUFFLE(2, 0, 0, 1)), A, N)
    

    N = 16

    _mm256_permute2x128_si256(A, A, _MM_SHUFFLE(2, 0, 0, 1))
    

    16 < N < 32

    _mm256_srli_si256(_mm256_permute2x128_si256(A, A, _MM_SHUFFLE(2, 0, 0, 1)), N - 16)
    

提交回复
热议问题