Shreesh Adiga:
> Signed-off-by: Shreesh Adiga <16567adigashre...@gmail.com>
> ---
> v3: Fix build failure on older nasm by replacing "kmovw k, tmpw"
>     with "kmov k, tmpd" which matches "kmovw k, r32" syntax.
> v2: Tried to align operands and improve indentation for ASM routine.
>  libswscale/x86/rgb2rgb.c     | 21 +++++++++
>  libswscale/x86/rgb_2_rgb.asm | 90 +++++++++++++++++++++++-------------
>  2 files changed, 80 insertions(+), 31 deletions(-)
> 
> diff --git a/libswscale/x86/rgb2rgb.c b/libswscale/x86/rgb2rgb.c
> index 6790551a38..4cbed54b35 100644
> --- a/libswscale/x86/rgb2rgb.c
> +++ b/libswscale/x86/rgb2rgb.c
> @@ -2364,6 +2364,16 @@ void ff_shuffle_bytes_2013_avx2(const uint8_t *src, 
> uint8_t *dst, int src_size);
>  void ff_shuffle_bytes_2130_avx2(const uint8_t *src, uint8_t *dst, int 
> src_size);
>  void ff_shuffle_bytes_1203_avx2(const uint8_t *src, uint8_t *dst, int 
> src_size);
> 
> +void ff_shuffle_bytes_2103_avx512icl(const uint8_t *src, uint8_t *dst, int 
> src_size);
> +void ff_shuffle_bytes_0321_avx512icl(const uint8_t *src, uint8_t *dst, int 
> src_size);
> +void ff_shuffle_bytes_1230_avx512icl(const uint8_t *src, uint8_t *dst, int 
> src_size);
> +void ff_shuffle_bytes_3012_avx512icl(const uint8_t *src, uint8_t *dst, int 
> src_size);
> +void ff_shuffle_bytes_3210_avx512icl(const uint8_t *src, uint8_t *dst, int 
> src_size);
> +void ff_shuffle_bytes_3102_avx512icl(const uint8_t *src, uint8_t *dst, int 
> src_size);
> +void ff_shuffle_bytes_2013_avx512icl(const uint8_t *src, uint8_t *dst, int 
> src_size);
> +void ff_shuffle_bytes_2130_avx512icl(const uint8_t *src, uint8_t *dst, int 
> src_size);
> +void ff_shuffle_bytes_1203_avx512icl(const uint8_t *src, uint8_t *dst, int 
> src_size);
> +
>  void ff_uyvytoyuv422_sse2(uint8_t *ydst, uint8_t *udst, uint8_t *vdst,
>                            const uint8_t *src, int width, int height,
>                            int lumStride, int chromStride, int srcStride);
> @@ -2454,6 +2464,17 @@ av_cold void rgb2rgb_init_x86(void)
>          shuffle_bytes_2130 = ff_shuffle_bytes_2130_avx2;
>          shuffle_bytes_1203 = ff_shuffle_bytes_1203_avx2;
>      }
> +    if (EXTERNAL_AVX512ICL(cpu_flags)) {
> +        shuffle_bytes_0321 = ff_shuffle_bytes_0321_avx512icl;
> +        shuffle_bytes_2103 = ff_shuffle_bytes_2103_avx512icl;
> +        shuffle_bytes_1230 = ff_shuffle_bytes_1230_avx512icl;
> +        shuffle_bytes_3012 = ff_shuffle_bytes_3012_avx512icl;
> +        shuffle_bytes_3210 = ff_shuffle_bytes_3210_avx512icl;
> +        shuffle_bytes_3102 = ff_shuffle_bytes_3102_avx512icl;
> +        shuffle_bytes_2013 = ff_shuffle_bytes_2013_avx512icl;
> +        shuffle_bytes_2130 = ff_shuffle_bytes_2130_avx512icl;
> +        shuffle_bytes_1203 = ff_shuffle_bytes_1203_avx512icl;
> +    }
>      if (EXTERNAL_AVX2_FAST(cpu_flags)) {
>          uyvytoyuv422 = ff_uyvytoyuv422_avx2;
>  #endif
> diff --git a/libswscale/x86/rgb_2_rgb.asm b/libswscale/x86/rgb_2_rgb.asm
> index b468beb12d..ca7a481255 100644
> --- a/libswscale/x86/rgb_2_rgb.asm
> +++ b/libswscale/x86/rgb_2_rgb.asm
> @@ -57,40 +57,53 @@ SECTION .text
>  %macro SHUFFLE_BYTES 4
>  cglobal shuffle_bytes_%1%2%3%4, 3, 5, 2, src, dst, w, tmp, x
>      VBROADCASTI128    m0, [pb_shuffle%1%2%3%4]
> -    movsxdifnidn wq, wd
> -    mov xq, wq
> -
> -    add        srcq, wq
> -    add        dstq, wq
> -    neg          wq
> -
> -;calc scalar loop
> +    movsxdifnidn      wq, wd
> +    mov               xq, wq
> +
> +    add             srcq, wq
> +    add             dstq, wq
> +    neg               wq
> +
> +%if mmsize == 64
> +    and                    xq, mmsize - 4
> +    shr                    xq, 2
> +    mov                  tmpd, -1
> +    shlx                 tmpd, tmpd, xd
> +    not                  tmpd
> +    kmovw                  k7, tmpd
> +    vmovdqu32       m1{k7}{z}, [srcq + wq]
> +    pshufb                 m1, m0
> +    vmovdqu32 [dstq + wq]{k7}, m1
> +    lea                    wq, [wq + 4 * xq]
> +%else
> +    ;calc scalar loop
>      and xq, mmsize-4
>      je .loop_simd
> 
> -.loop_scalar:
> -   mov          tmpb, [srcq + wq + %1]
> -   mov [dstq+wq + 0], tmpb
> -   mov          tmpb, [srcq + wq + %2]
> -   mov [dstq+wq + 1], tmpb
> -   mov          tmpb, [srcq + wq + %3]
> -   mov [dstq+wq + 2], tmpb
> -   mov          tmpb, [srcq + wq + %4]
> -   mov [dstq+wq + 3], tmpb
> -   add            wq, 4
> -   sub            xq, 4
> -   jg .loop_scalar
> -
> -;check if src_size < mmsize
> -cmp wq, 0
> -jge .end
> -
> -.loop_simd:
> -    movu           m1, [srcq+wq]
> -    pshufb         m1, m0
> -    movu    [dstq+wq], m1
> -    add            wq, mmsize
> -    jl .loop_simd
> +    .loop_scalar:
> +        mov          tmpb, [srcq + wq + %1]
> +        mov [dstq+wq + 0], tmpb
> +        mov          tmpb, [srcq + wq + %2]
> +        mov [dstq+wq + 1], tmpb
> +        mov          tmpb, [srcq + wq + %3]
> +        mov [dstq+wq + 2], tmpb
> +        mov          tmpb, [srcq + wq + %4]
> +        mov [dstq+wq + 3], tmpb
> +        add            wq, 4
> +        sub            xq, 4
> +        jg .loop_scalar
> +%endif
> +
> +    ;check if src_size < mmsize
> +    cmp wq, 0
> +    jge .end
> +
> +    .loop_simd:
> +        movu            m1, [srcq + wq]
> +        pshufb          m1, m0
> +        movu   [dstq + wq], m1
> +        add             wq, mmsize
> +        jl .loop_simd
> 
>  .end:
>      RET
> @@ -122,6 +135,21 @@ SHUFFLE_BYTES 1, 2, 0, 3
>  %endif
>  %endif
> 
> +%if ARCH_X86_64
> +%if HAVE_AVX512ICL_EXTERNAL
> +INIT_ZMM avx512icl
> +SHUFFLE_BYTES 2, 1, 0, 3
> +SHUFFLE_BYTES 0, 3, 2, 1
> +SHUFFLE_BYTES 1, 2, 3, 0
> +SHUFFLE_BYTES 3, 0, 1, 2
> +SHUFFLE_BYTES 3, 2, 1, 0
> +SHUFFLE_BYTES 3, 1, 0, 2
> +SHUFFLE_BYTES 2, 0, 1, 3
> +SHUFFLE_BYTES 2, 1, 3, 0
> +SHUFFLE_BYTES 1, 2, 0, 3
> +%endif
> +%endif
> +
>  
> ;-----------------------------------------------------------------------------------------------
>  ; uyvytoyuv422(uint8_t *ydst, uint8_t *udst, uint8_t *vdst,
>  ;              const uint8_t *src, int width, int height,
> --
> 2.45.3
> 
Why does the commit message not contain speedup numbers?

- Andreas

_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".

Reply via email to