[x265] [PATCH 195 of 307] x86: AVX512 interp_8tap_vert_sp_24x32 and interp_8tap_vert_ss_24x32

mythreyi at multicorewareinc.com mythreyi at multicorewareinc.com
Sat Apr 7 04:33:13 CEST 2018


# HG changeset patch
# User Vignesh Vijayakumar<vignesh at multicorewareinc.com>
# Date 1511505545 -19800
#      Fri Nov 24 12:09:05 2017 +0530
# Node ID 47b99c09008b1921881b0dfa00d80cce1f8d15eb
# Parent  ecaf36f641dd1428d556f172e83cf7078f0287fb
x86: AVX512 interp_8tap_vert_sp_24x32 and interp_8tap_vert_ss_24x32

luma_vss
AVX2 performance   : 10.98x
AVX512 performance : 16.36x

luma_vsp
AVX2 performance   : 12.19x
AVX512 performance : 17.20x

diff -r ecaf36f641dd -r 47b99c09008b source/common/x86/asm-primitives.cpp
--- a/source/common/x86/asm-primitives.cpp	Fri Nov 24 11:34:33 2017 +0530
+++ b/source/common/x86/asm-primitives.cpp	Fri Nov 24 12:09:05 2017 +0530
@@ -2844,6 +2844,7 @@
         p.pu[LUMA_16x16].luma_vss = PFX(interp_8tap_vert_ss_16x16_avx512);
         p.pu[LUMA_16x32].luma_vss = PFX(interp_8tap_vert_ss_16x32_avx512);
         p.pu[LUMA_16x64].luma_vss = PFX(interp_8tap_vert_ss_16x64_avx512);
+        p.pu[LUMA_24x32].luma_vss = PFX(interp_8tap_vert_ss_24x32_avx512);
         p.pu[LUMA_32x8].luma_vss = PFX(interp_8tap_vert_ss_32x8_avx512);
         p.pu[LUMA_32x16].luma_vss = PFX(interp_8tap_vert_ss_32x16_avx512);
         p.pu[LUMA_32x32].luma_vss = PFX(interp_8tap_vert_ss_32x32_avx512);
@@ -2864,6 +2865,7 @@
         p.pu[LUMA_16x16].luma_vsp = PFX(interp_8tap_vert_sp_16x16_avx512);
         p.pu[LUMA_16x32].luma_vsp = PFX(interp_8tap_vert_sp_16x32_avx512);
         p.pu[LUMA_16x64].luma_vsp = PFX(interp_8tap_vert_sp_16x64_avx512);
+        p.pu[LUMA_24x32].luma_vsp = PFX(interp_8tap_vert_sp_24x32_avx512);
         p.pu[LUMA_32x8].luma_vsp = PFX(interp_8tap_vert_sp_32x8_avx512);
         p.pu[LUMA_32x16].luma_vsp = PFX(interp_8tap_vert_sp_32x16_avx512);
         p.pu[LUMA_32x32].luma_vsp = PFX(interp_8tap_vert_sp_32x32_avx512);
diff -r ecaf36f641dd -r 47b99c09008b source/common/x86/ipfilter16.asm
--- a/source/common/x86/ipfilter16.asm	Fri Nov 24 11:34:33 2017 +0530
+++ b/source/common/x86/ipfilter16.asm	Fri Nov 24 12:09:05 2017 +0530
@@ -11134,6 +11134,284 @@
     FILTER_VER_S_LUMA_16xN_AVX512 sp, 64
 %endif
 
+%macro PROCESS_LUMA_VERT_S_24x8_AVX512 1
+    PROCESS_LUMA_VERT_S_16x4_AVX512 %1
+    lea                  r4,                  [r6 + 4 * r1]
+    lea                  r8,                  [r4 + 4 * r1]
+    movu                 ym1,                 [r6]
+    movu                 ym3,                 [r6 + r1]
+    vinserti32x8         m1,                  [r6 + 2 * r1],          1
+    vinserti32x8         m3,                  [r6 + r7],              1
+    punpcklwd            m0,                  m1,                     m3
+    pmaddwd              m0,                  m15
+    punpckhwd            m1,                  m3
+    pmaddwd              m1,                  m15
+
+    movu                 ym4,                 [r6 + 2 * r1]
+    vinserti32x8         m4,                  [r4],                   1
+    punpcklwd            m2,                  m3,                     m4
+    pmaddwd              m2,                  m15
+    punpckhwd            m3,                  m4
+    pmaddwd              m3,                  m15
+
+    movu                 ym5,                 [r6 + r7]
+    vinserti32x8         m5,                  [r4 + r1],              1
+    punpcklwd            m6,                  m4,                     m5
+    pmaddwd              m6,                  m16
+    punpckhwd            m4,                  m5
+    pmaddwd              m4,                  m16
+
+    paddd                m0,                  m6
+    paddd                m1,                  m4
+
+    movu                 ym4,                 [r4]
+    vinserti32x8         m4,                  [r4 + 2 * r1],          1
+    punpcklwd            m6,                  m5,                     m4
+    pmaddwd              m6,                  m16
+    punpckhwd            m5,                  m4
+    pmaddwd              m5,                  m16
+
+    paddd                m2,                  m6
+    paddd                m3,                  m5
+
+    movu                 ym11,                [r4 + r1]
+    vinserti32x8         m11,                 [r4 + r7],              1
+    punpcklwd            m8,                  m4,                     m11
+    pmaddwd              m8,                  m17
+    punpckhwd            m4,                  m11
+    pmaddwd              m4,                  m17
+
+    movu                 ym12,                [r4 + 2 * r1]
+    vinserti32x8         m12,                 [r4 + 4 * r1],          1
+    punpcklwd            m10,                 m11,                    m12
+    pmaddwd              m10,                 m17
+    punpckhwd            m11,                 m12
+    pmaddwd              m11,                 m17
+
+    movu                 ym13,                [r4 + r7]
+    vinserti32x8         m13,                 [r8 + r1],              1
+    punpcklwd            m14,                 m12,                    m13
+    pmaddwd              m14,                 m18
+    punpckhwd            m12,                 m13
+    pmaddwd              m12,                 m18
+
+    paddd                m8,                  m14
+    paddd                m4,                  m12
+    paddd                m0,                  m8
+    paddd                m1,                  m4
+
+    movu                 ym12,                [r4 + 4 * r1]
+    vinserti32x8         m12,                 [r8 + 2 * r1],          1
+    punpcklwd            m14,                 m13,                    m12
+    pmaddwd              m14,                 m18
+    punpckhwd            m13,                 m12
+    pmaddwd              m13,                 m18
+
+    paddd                m10,                 m14
+    paddd                m11,                 m13
+    paddd                m2,                  m10
+    paddd                m3,                  m11
+
+%ifidn %1, sp
+    paddd                m0,                  m19
+    paddd                m1,                  m19
+    paddd                m2,                  m19
+    paddd                m3,                  m19
+
+    psrad                m0,                  INTERP_SHIFT_SP
+    psrad                m1,                  INTERP_SHIFT_SP
+    psrad                m2,                  INTERP_SHIFT_SP
+    psrad                m3,                  INTERP_SHIFT_SP
+
+    packssdw             m0,                  m1
+    packssdw             m2,                  m3
+    CLIPW2               m0,                  m2,                   m20,                 m21
+%else
+    psrad                m0,                  6
+    psrad                m1,                  6
+    psrad                m2,                  6
+    psrad                m3,                  6
+
+    packssdw             m0,                  m1
+    packssdw             m2,                  m3
+%endif
+    lea                  r9,                  [r2 + 4 * r3]
+    movu                 [r9],                ym0
+    movu                 [r9 + r3],           ym2
+    vextracti32x8        [r9 + 2 * r3],       m0,                1
+    vextracti32x8        [r9 + r5],           m2,                1
+
+    movu                 xm1,                 [r0 + mmsize/2]
+    vinserti32x4         m1,                  [r0 + 2 * r1 + mmsize/2],          1
+    vinserti32x4         m1,                  [r0 + 4 * r1 + mmsize/2],          2
+    vinserti32x4         m1,                  [r6 + 2 * r1 + mmsize/2],          3
+    movu                 xm3,                 [r0 + r1 + mmsize/2]
+    vinserti32x4         m3,                  [r0 + r7 + mmsize/2],              1
+    vinserti32x4         m3,                  [r6 + r1 + mmsize/2],              2
+    vinserti32x4         m3,                  [r6 + r7 + mmsize/2],              3
+    punpcklwd            m0,                  m1,                     m3
+    pmaddwd              m0,                  m15
+    punpckhwd            m1,                  m3
+    pmaddwd              m1,                  m15
+
+    movu                 xm4,                 [r0 + 2 * r1 + mmsize/2]
+    vinserti32x4         m4,                  [r0 + 4 * r1 + mmsize/2],          1
+    vinserti32x4         m4,                  [r6 + 2 * r1 + mmsize/2],          2
+    vinserti32x4         m4,                  [r6 + 4 * r1 + mmsize/2],          3
+    punpcklwd            m2,                  m3,                     m4
+    pmaddwd              m2,                  m15
+    punpckhwd            m3,                  m4
+    pmaddwd              m3,                  m15
+
+    movu                 xm5,                 [r0 + r7 + mmsize/2]
+    vinserti32x4         m5,                  [r6 + r1 + mmsize/2],              1
+    vinserti32x4         m5,                  [r6 + r7 + mmsize/2],              2
+    vinserti32x4         m5,                  [r4 + r1 + mmsize/2],              3
+    punpcklwd            m6,                  m4,                     m5
+    pmaddwd              m6,                  m16
+    punpckhwd            m4,                  m5
+    pmaddwd              m4,                  m16
+
+    paddd                m0,                  m6
+    paddd                m1,                  m4
+
+    movu                 xm4,                 [r0 + 4 * r1 + mmsize/2]
+    vinserti32x4         m4,                  [r6 + 2 * r1 + mmsize/2],              1
+    vinserti32x4         m4,                  [r6 + 4 * r1 + mmsize/2],              2
+    vinserti32x4         m4,                  [r4 + 2 * r1 + mmsize/2],              3
+    punpcklwd            m6,                  m5,                     m4
+    pmaddwd              m6,                  m16
+    punpckhwd            m5,                  m4
+    pmaddwd              m5,                  m16
+
+    paddd                m2,                  m6
+    paddd                m3,                  m5
+
+    movu                 xm11,                [r6 + r1 + mmsize/2]
+    vinserti32x4         m11,                 [r6 + r7 + mmsize/2],              1
+    vinserti32x4         m11,                 [r4 + r1 + mmsize/2],              2
+    vinserti32x4         m11,                 [r4 + r7 + mmsize/2],              3
+    punpcklwd            m8,                  m4,                     m11
+    pmaddwd              m8,                  m17
+    punpckhwd            m4,                  m11
+    pmaddwd              m4,                  m17
+
+    movu                 xm12,                [r6 + 2 * r1 + mmsize/2]
+    vinserti32x4         m12,                 [r6 + 4 * r1 + mmsize/2],          1
+    vinserti32x4         m12,                 [r4 + 2 * r1 + mmsize/2],          2
+    vinserti32x4         m12,                 [r4 + 4 * r1 + mmsize/2],          3
+    punpcklwd            m10,                 m11,                    m12
+    pmaddwd              m10,                 m17
+    punpckhwd            m11,                 m12
+    pmaddwd              m11,                 m17
+
+    movu                 xm13,                [r6 + r7 + mmsize/2]
+    vinserti32x4         m13,                 [r4 + r1 + mmsize/2],              1
+    vinserti32x4         m13,                 [r4 + r7 + mmsize/2],              2
+    vinserti32x4         m13,                 [r8 + r1 + mmsize/2],              3
+    punpcklwd            m14,                 m12,                    m13
+    pmaddwd              m14,                 m18
+    punpckhwd            m12,                 m13
+    pmaddwd              m12,                 m18
+
+    paddd                m8,                  m14
+    paddd                m4,                  m12
+    paddd                m0,                  m8
+    paddd                m1,                  m4
+
+    movu                 xm12,                [r6 + 4 * r1 + mmsize/2]
+    vinserti32x4         m12,                 [r4 + 2 * r1 + mmsize/2],          1
+    vinserti32x4         m12,                 [r4 + 4 * r1 + mmsize/2],          2
+    vinserti32x4         m12,                 [r8 + 2 * r1 + mmsize/2],          3
+    punpcklwd            m14,                 m13,                    m12
+    pmaddwd              m14,                 m18
+    punpckhwd            m13,                 m12
+    pmaddwd              m13,                 m18
+
+    paddd                m10,                 m14
+    paddd                m11,                 m13
+    paddd                m2,                  m10
+    paddd                m3,                  m11
+
+%ifidn %1, sp
+    paddd                m0,                  m19
+    paddd                m1,                  m19
+    paddd                m2,                  m19
+    paddd                m3,                  m19
+
+    psrad                m0,                  INTERP_SHIFT_SP
+    psrad                m1,                  INTERP_SHIFT_SP
+    psrad                m2,                  INTERP_SHIFT_SP
+    psrad                m3,                  INTERP_SHIFT_SP
+
+    packssdw             m0,                  m1
+    packssdw             m2,                  m3
+    CLIPW2               m0,                  m2,                   m20,                 m21
+%else
+    psrad                m0,                  6
+    psrad                m1,                  6
+    psrad                m2,                  6
+    psrad                m3,                  6
+
+    packssdw             m0,                  m1
+    packssdw             m2,                  m3
+%endif
+
+    movu                 [r2 + mmsize/2],                xm0
+    movu                 [r2 + r3 + mmsize/2],           xm2
+    vextracti32x4        [r2 + 2 * r3 + mmsize/2],       m0,                  1
+    vextracti32x4        [r2 + r5 + mmsize/2],           m2,                  1
+    lea                  r2,                             [r2 + 4 * r3]
+    vextracti32x4        [r2 + mmsize/2],                m0,                  2
+    vextracti32x4        [r2 + r3 + mmsize/2],           m2,                  2
+    vextracti32x4        [r2 + 2 * r3 + mmsize/2],       m0,                  3
+    vextracti32x4        [r2 + r5 + mmsize/2],           m2,                  3
+%endmacro
+;-----------------------------------------------------------------------------------------------------------------
+; void interp_8tap_vert(int16_t *src, intptr_t srcStride, int16_t *dst, intptr_t dstStride, int coeffIdx)
+;-----------------------------------------------------------------------------------------------------------------
+%macro FILTER_VER_S_LUMA_24x32_AVX512 1
+INIT_ZMM avx512
+cglobal interp_8tap_vert_%1_24x32, 5, 10, 22
+    add                   r1d,                r1d
+    add                   r3d,                r3d
+    lea                   r7,                 [3 * r1]
+    sub                   r0,                 r7
+    shl                   r4d,                8
+%ifdef PIC
+    lea                   r5,                 [tab_LumaCoeffVer_avx512]
+    mova                  m15,                [r5 + r4]
+    mova                  m16,                [r5 + r4 + 1 * mmsize]
+    mova                  m17,                [r5 + r4 + 2 * mmsize]
+    mova                  m18,                [r5 + r4 + 3 * mmsize]
+%else
+    lea                   r5,                 [tab_LumaCoeffVer_avx512 + r4]
+    mova                  m15,                [r5]
+    mova                  m16,                [r5 + 1 * mmsize]
+    mova                  m17,                [r5 + 2 * mmsize]
+    mova                  m18,                [r5 + 3 * mmsize]
+%endif
+%ifidn %1, sp
+    vbroadcasti32x4       m19,                [INTERP_OFFSET_SP]
+    pxor                  m20,                m20
+    vbroadcasti32x8       m21,                [pw_pixel_max]
+%endif
+    lea                   r5,                 [3 * r3]
+
+%rep 3
+    PROCESS_LUMA_VERT_S_24x8_AVX512 %1
+    lea                   r0,                 [r4]
+    lea                   r2,                 [r2 + 4 * r3]
+%endrep
+    PROCESS_LUMA_VERT_S_24x8_AVX512 %1
+    RET
+%endmacro
+
+%if ARCH_X86_64
+    FILTER_VER_S_LUMA_24x32_AVX512 ss
+    FILTER_VER_S_LUMA_24x32_AVX512 sp
+%endif
+
 %macro PROCESS_LUMA_VERT_S_32x2_AVX512 1
     movu                 m1,                  [r0]                           ;0 row
     movu                 m3,                  [r0 + r1]                      ;1 row


More information about the x265-devel mailing list