How to convert 32-bit float to 8-bit signed char?

后端 未结 2 1099
旧巷少年郎
旧巷少年郎 2021-01-12 16:35

What I want to do is:

  1. Multiply the input floating point number by a fixed factor.
  2. Convert them to 8-bit signed char.

Note that most of

相关标签:
2条回答
  • 2021-01-12 16:44

    For good throughput with multiple source vectors, it's a good thing that _mm256_packs_epi16 has 2 input vectors instead of producing a narrower output. (AVX512 _mm256_cvtepi32_epi8 isn't necessarily the most efficient way to do things, because the version with a memory destination decodes to multiple uops, or the regular version gives you multiple small outputs that need to be stored separately.)

    Or are you complaining about how it operates in-lane? Yes that's annoying, but _mm256_packs_epi32 does the same thing. If it's ok for your outputs to have interleaved groups of data there, do the same thing for this, too.

    Your best bet is to combine 4 vectors down to 1, in 2 steps of in-lane packing (because there's no lane-crossing pack). Then use one lane-crossing shuffle to fix it up.

    #include <immintrin.h>
    // loads 128 bytes = 32 floats
    // converts and packs with signed saturation to 32 int8_t
    __m256i pack_float_int8(const float*p) {
        __m256i a = _mm256_cvtps_epi32(_mm256_loadu_ps(p));
        __m256i b = _mm256_cvtps_epi32(_mm256_loadu_ps(p+8));
        __m256i c = _mm256_cvtps_epi32(_mm256_loadu_ps(p+16));
        __m256i d = _mm256_cvtps_epi32(_mm256_loadu_ps(p+24));
        __m256i ab = _mm256_packs_epi32(a,b);        // 16x int16_t
        __m256i cd = _mm256_packs_epi32(c,d);
        __m256i abcd = _mm256_packs_epi16(ab, cd);   // 32x int8_t
        // packed to one vector, but in [ a_lo, b_lo, c_lo, d_lo | a_hi, b_hi, c_hi, d_hi ] order
        // if you can deal with that in-memory format (e.g. for later in-lane unpack), great, you're done
    
        // but if you need sequential order, then vpermd:
        __m256i lanefix = _mm256_permutevar8x32_epi32(abcd, _mm256_setr_epi32(0,4, 1,5, 2,6, 3,7));
        return lanefix;
    }
    

    (Compiles nicely on the Godbolt compiler explorer).

    Call this in a loop and _mm256_store_si256 the resulting vector.


    (For uint8_t unsigned destination, use _mm256_packus_epi16 for the 16->8 step and keep everything else the same. We still use signed 32->16 packing, because 16 -> u8 vpackuswb packing still takes its epi16 input as signed. You need -1 to be treated as -1, not +0xFFFF, for unsigned saturation to clamp it to 0.)


    With 4 total shuffles per 256-bit store, 1 shuffle per clock throughput will be the bottleneck on Intel CPUs. You should get a throughput of one float vector per clock, bottlenecked on port 5. (https://agner.org/optimize/). Or maybe bottlenecked on memory bandwidth if data isn't hot in L2.


    If you only have a single vector to do, you could consider using _mm256_shuffle_epi8 to put the low byte of each epi32 element into the low 32 bits of each lane, then _mm256_permutevar8x32_epi32 for lane-crossing.

    Another single-vector alternative (good on Ryzen) is extracti128 + 128-bit packssdw + packsswb. But that's still only good if you're just doing a single vector. (Still on Ryzen, you'll want to work in 128-bit vectors to avoid extra lane-crossing shuffles, because Ryzen splits every 256-bit instruction into (at least) 2 128-bit uops.)

    Related:

    • SSE - AVX conversion from double to char
    • How can I convert a vector of float to short int using avx instructions?
    0 讨论(0)
  • 2021-01-12 17:05

    Please check the IEEE754 standard format to store float values, first understand how this float and double get store in memory ,then you only came to know how to convert float or double to the char , it is quite simple .

    0 讨论(0)
提交回复
热议问题