| /* | |
| * Copyright (c) 2016, Alliance for Open Media. All rights reserved | |
| * | |
| * This source code is subject to the terms of the BSD 2 Clause License and | |
| * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License | |
| * was not distributed with this source code in the LICENSE file, you can | |
| * obtain it at www.aomedia.org/license/software. If the Alliance for Open | |
| * Media Patent License 1.0 was not distributed with this source code in the | |
| * PATENTS file, you can obtain it at www.aomedia.org/license/patent. | |
| */ | |
| #ifndef AOM_DSP_MIPS_MACROS_MSA_H_ | |
| #define AOM_DSP_MIPS_MACROS_MSA_H_ | |
| #include <msa.h> | |
| #include "./aom_config.h" | |
| #include "aom/aom_integer.h" | |
| #define LD_B(RTYPE, psrc) *((const RTYPE *)(psrc)) | |
| #define LD_UB(...) LD_B(v16u8, __VA_ARGS__) | |
| #define LD_SB(...) LD_B(v16i8, __VA_ARGS__) | |
| #define LD_H(RTYPE, psrc) *((const RTYPE *)(psrc)) | |
| #define LD_UH(...) LD_H(v8u16, __VA_ARGS__) | |
| #define LD_SH(...) LD_H(v8i16, __VA_ARGS__) | |
| #define LD_W(RTYPE, psrc) *((const RTYPE *)(psrc)) | |
| #define LD_SW(...) LD_W(v4i32, __VA_ARGS__) | |
| #define ST_B(RTYPE, in, pdst) *((RTYPE *)(pdst)) = (in) | |
| #define ST_UB(...) ST_B(v16u8, __VA_ARGS__) | |
| #define ST_SB(...) ST_B(v16i8, __VA_ARGS__) | |
| #define ST_H(RTYPE, in, pdst) *((RTYPE *)(pdst)) = (in) | |
| #define ST_SH(...) ST_H(v8i16, __VA_ARGS__) | |
| #define ST_W(RTYPE, in, pdst) *((RTYPE *)(pdst)) = (in) | |
| #define ST_SW(...) ST_W(v4i32, __VA_ARGS__) | |
| #if (__mips_isa_rev >= 6) | |
| #define LH(psrc) \ | |
| ({ \ | |
| const uint8_t *psrc_m = (const uint8_t *)(psrc); \ | |
| uint16_t val_m; \ | |
| \ | |
| __asm__ __volatile__("lh %[val_m], %[psrc_m] \n\t" \ | |
| \ | |
| : [val_m] "=r"(val_m) \ | |
| : [psrc_m] "m"(*psrc_m)); \ | |
| \ | |
| val_m; \ | |
| }) | |
| #define LW(psrc) \ | |
| ({ \ | |
| const uint8_t *psrc_m = (const uint8_t *)(psrc); \ | |
| uint32_t val_m; \ | |
| \ | |
| __asm__ __volatile__("lw %[val_m], %[psrc_m] \n\t" \ | |
| \ | |
| : [val_m] "=r"(val_m) \ | |
| : [psrc_m] "m"(*psrc_m)); \ | |
| \ | |
| val_m; \ | |
| }) | |
| #if (__mips == 64) | |
| #define LD(psrc) \ | |
| ({ \ | |
| const uint8_t *psrc_m = (const uint8_t *)(psrc); \ | |
| uint64_t val_m = 0; \ | |
| \ | |
| __asm__ __volatile__("ld %[val_m], %[psrc_m] \n\t" \ | |
| \ | |
| : [val_m] "=r"(val_m) \ | |
| : [psrc_m] "m"(*psrc_m)); \ | |
| \ | |
| val_m; \ | |
| }) | |
| #else // !(__mips == 64) | |
| #define LD(psrc) \ | |
| ({ \ | |
| const uint8_t *psrc_m = (const uint8_t *)(psrc); \ | |
| uint32_t val0_m, val1_m; \ | |
| uint64_t val_m = 0; \ | |
| \ | |
| val0_m = LW(psrc_m); \ | |
| val1_m = LW(psrc_m + 4); \ | |
| \ | |
| val_m = (uint64_t)(val1_m); \ | |
| val_m = (uint64_t)((val_m << 32) & 0xFFFFFFFF00000000); \ | |
| val_m = (uint64_t)(val_m | (uint64_t)val0_m); \ | |
| \ | |
| val_m; \ | |
| }) | |
| #endif // (__mips == 64) | |
| #define SH(val, pdst) \ | |
| { \ | |
| uint8_t *pdst_m = (uint8_t *)(pdst); \ | |
| const uint16_t val_m = (val); \ | |
| \ | |
| __asm__ __volatile__("sh %[val_m], %[pdst_m] \n\t" \ | |
| \ | |
| : [pdst_m] "=m"(*pdst_m) \ | |
| : [val_m] "r"(val_m)); \ | |
| } | |
| #define SW(val, pdst) \ | |
| { \ | |
| uint8_t *pdst_m = (uint8_t *)(pdst); \ | |
| const uint32_t val_m = (val); \ | |
| \ | |
| __asm__ __volatile__("sw %[val_m], %[pdst_m] \n\t" \ | |
| \ | |
| : [pdst_m] "=m"(*pdst_m) \ | |
| : [val_m] "r"(val_m)); \ | |
| } | |
| #define SD(val, pdst) \ | |
| { \ | |
| uint8_t *pdst_m = (uint8_t *)(pdst); \ | |
| const uint64_t val_m = (val); \ | |
| \ | |
| __asm__ __volatile__("sd %[val_m], %[pdst_m] \n\t" \ | |
| \ | |
| : [pdst_m] "=m"(*pdst_m) \ | |
| : [val_m] "r"(val_m)); \ | |
| } | |
| #else // !(__mips_isa_rev >= 6) | |
| #define LH(psrc) \ | |
| ({ \ | |
| const uint8_t *psrc_m = (const uint8_t *)(psrc); \ | |
| uint16_t val_m; \ | |
| \ | |
| __asm__ __volatile__("ulh %[val_m], %[psrc_m] \n\t" \ | |
| \ | |
| : [val_m] "=r"(val_m) \ | |
| : [psrc_m] "m"(*psrc_m)); \ | |
| \ | |
| val_m; \ | |
| }) | |
| #define LW(psrc) \ | |
| ({ \ | |
| const uint8_t *psrc_m = (const uint8_t *)(psrc); \ | |
| uint32_t val_m; \ | |
| \ | |
| __asm__ __volatile__("ulw %[val_m], %[psrc_m] \n\t" \ | |
| \ | |
| : [val_m] "=r"(val_m) \ | |
| : [psrc_m] "m"(*psrc_m)); \ | |
| \ | |
| val_m; \ | |
| }) | |
| #if (__mips == 64) | |
| #define LD(psrc) \ | |
| ({ \ | |
| const uint8_t *psrc_m = (const uint8_t *)(psrc); \ | |
| uint64_t val_m = 0; \ | |
| \ | |
| __asm__ __volatile__("uld %[val_m], %[psrc_m] \n\t" \ | |
| \ | |
| : [val_m] "=r"(val_m) \ | |
| : [psrc_m] "m"(*psrc_m)); \ | |
| \ | |
| val_m; \ | |
| }) | |
| #else // !(__mips == 64) | |
| #define LD(psrc) \ | |
| ({ \ | |
| const uint8_t *psrc_m1 = (const uint8_t *)(psrc); \ | |
| uint32_t val0_m, val1_m; \ | |
| uint64_t val_m_combined = 0; \ | |
| \ | |
| val0_m = LW(psrc_m1); \ | |
| val1_m = LW(psrc_m1 + 4); \ | |
| \ | |
| val_m_combined = (uint64_t)(val1_m); \ | |
| val_m_combined = (uint64_t)((val_m_combined << 32) & 0xFFFFFFFF00000000); \ | |
| val_m_combined = (uint64_t)(val_m_combined | (uint64_t)val0_m); \ | |
| \ | |
| val_m_combined; \ | |
| }) | |
| #endif // (__mips == 64) | |
| #define SH(val, pdst) \ | |
| { \ | |
| uint8_t *pdst_m = (uint8_t *)(pdst); \ | |
| const uint16_t val_m = (val); \ | |
| \ | |
| __asm__ __volatile__("ush %[val_m], %[pdst_m] \n\t" \ | |
| \ | |
| : [pdst_m] "=m"(*pdst_m) \ | |
| : [val_m] "r"(val_m)); \ | |
| } | |
| #define SW(val, pdst) \ | |
| { \ | |
| uint8_t *pdst_m = (uint8_t *)(pdst); \ | |
| const uint32_t val_m = (val); \ | |
| \ | |
| __asm__ __volatile__("usw %[val_m], %[pdst_m] \n\t" \ | |
| \ | |
| : [pdst_m] "=m"(*pdst_m) \ | |
| : [val_m] "r"(val_m)); \ | |
| } | |
| #define SD(val, pdst) \ | |
| { \ | |
| uint8_t *pdst_m1 = (uint8_t *)(pdst); \ | |
| uint32_t val0_m, val1_m; \ | |
| \ | |
| val0_m = (uint32_t)((val)&0x00000000FFFFFFFF); \ | |
| val1_m = (uint32_t)(((val) >> 32) & 0x00000000FFFFFFFF); \ | |
| \ | |
| SW(val0_m, pdst_m1); \ | |
| SW(val1_m, pdst_m1 + 4); \ | |
| } | |
| #endif // (__mips_isa_rev >= 6) | |
| /* Description : Load 4 words with stride | |
| Arguments : Inputs - psrc, stride | |
| Outputs - out0, out1, out2, out3 | |
| Details : Load word in 'out0' from (psrc) | |
| Load word in 'out1' from (psrc + stride) | |
| Load word in 'out2' from (psrc + 2 * stride) | |
| Load word in 'out3' from (psrc + 3 * stride) | |
| */ | |
| #define LW4(psrc, stride, out0, out1, out2, out3) \ | |
| { \ | |
| out0 = LW((psrc)); \ | |
| out1 = LW((psrc) + stride); \ | |
| out2 = LW((psrc) + 2 * stride); \ | |
| out3 = LW((psrc) + 3 * stride); \ | |
| } | |
| /* Description : Load double words with stride | |
| Arguments : Inputs - psrc, stride | |
| Outputs - out0, out1 | |
| Details : Load double word in 'out0' from (psrc) | |
| Load double word in 'out1' from (psrc + stride) | |
| */ | |
| #define LD2(psrc, stride, out0, out1) \ | |
| { \ | |
| out0 = LD((psrc)); \ | |
| out1 = LD((psrc) + stride); \ | |
| } | |
| #define LD4(psrc, stride, out0, out1, out2, out3) \ | |
| { \ | |
| LD2((psrc), stride, out0, out1); \ | |
| LD2((psrc) + 2 * stride, stride, out2, out3); \ | |
| } | |
| /* Description : Store 4 words with stride | |
| Arguments : Inputs - in0, in1, in2, in3, pdst, stride | |
| Details : Store word from 'in0' to (pdst) | |
| Store word from 'in1' to (pdst + stride) | |
| Store word from 'in2' to (pdst + 2 * stride) | |
| Store word from 'in3' to (pdst + 3 * stride) | |
| */ | |
| #define SW4(in0, in1, in2, in3, pdst, stride) \ | |
| { \ | |
| SW(in0, (pdst)) \ | |
| SW(in1, (pdst) + stride); \ | |
| SW(in2, (pdst) + 2 * stride); \ | |
| SW(in3, (pdst) + 3 * stride); \ | |
| } | |
| /* Description : Store 4 double words with stride | |
| Arguments : Inputs - in0, in1, in2, in3, pdst, stride | |
| Details : Store double word from 'in0' to (pdst) | |
| Store double word from 'in1' to (pdst + stride) | |
| Store double word from 'in2' to (pdst + 2 * stride) | |
| Store double word from 'in3' to (pdst + 3 * stride) | |
| */ | |
| #define SD4(in0, in1, in2, in3, pdst, stride) \ | |
| { \ | |
| SD(in0, (pdst)) \ | |
| SD(in1, (pdst) + stride); \ | |
| SD(in2, (pdst) + 2 * stride); \ | |
| SD(in3, (pdst) + 3 * stride); \ | |
| } | |
| /* Description : Load vectors with 16 byte elements with stride | |
| Arguments : Inputs - psrc, stride | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Load 16 byte elements in 'out0' from (psrc) | |
| Load 16 byte elements in 'out1' from (psrc + stride) | |
| */ | |
| #define LD_B2(RTYPE, psrc, stride, out0, out1) \ | |
| { \ | |
| out0 = LD_B(RTYPE, (psrc)); \ | |
| out1 = LD_B(RTYPE, (psrc) + stride); \ | |
| } | |
| #define LD_UB2(...) LD_B2(v16u8, __VA_ARGS__) | |
| #define LD_SB2(...) LD_B2(v16i8, __VA_ARGS__) | |
| #define LD_B3(RTYPE, psrc, stride, out0, out1, out2) \ | |
| { \ | |
| LD_B2(RTYPE, (psrc), stride, out0, out1); \ | |
| out2 = LD_B(RTYPE, (psrc) + 2 * stride); \ | |
| } | |
| #define LD_UB3(...) LD_B3(v16u8, __VA_ARGS__) | |
| #define LD_B4(RTYPE, psrc, stride, out0, out1, out2, out3) \ | |
| { \ | |
| LD_B2(RTYPE, (psrc), stride, out0, out1); \ | |
| LD_B2(RTYPE, (psrc) + 2 * stride, stride, out2, out3); \ | |
| } | |
| #define LD_UB4(...) LD_B4(v16u8, __VA_ARGS__) | |
| #define LD_SB4(...) LD_B4(v16i8, __VA_ARGS__) | |
| #define LD_B5(RTYPE, psrc, stride, out0, out1, out2, out3, out4) \ | |
| { \ | |
| LD_B4(RTYPE, (psrc), stride, out0, out1, out2, out3); \ | |
| out4 = LD_B(RTYPE, (psrc) + 4 * stride); \ | |
| } | |
| #define LD_UB5(...) LD_B5(v16u8, __VA_ARGS__) | |
| #define LD_SB5(...) LD_B5(v16i8, __VA_ARGS__) | |
| #define LD_B7(RTYPE, psrc, stride, out0, out1, out2, out3, out4, out5, out6) \ | |
| { \ | |
| LD_B5(RTYPE, (psrc), stride, out0, out1, out2, out3, out4); \ | |
| LD_B2(RTYPE, (psrc) + 5 * stride, stride, out5, out6); \ | |
| } | |
| #define LD_SB7(...) LD_B7(v16i8, __VA_ARGS__) | |
| #define LD_B8(RTYPE, psrc, stride, out0, out1, out2, out3, out4, out5, out6, \ | |
| out7) \ | |
| { \ | |
| LD_B4(RTYPE, (psrc), stride, out0, out1, out2, out3); \ | |
| LD_B4(RTYPE, (psrc) + 4 * stride, stride, out4, out5, out6, out7); \ | |
| } | |
| #define LD_UB8(...) LD_B8(v16u8, __VA_ARGS__) | |
| #define LD_SB8(...) LD_B8(v16i8, __VA_ARGS__) | |
| /* Description : Load vectors with 8 halfword elements with stride | |
| Arguments : Inputs - psrc, stride | |
| Outputs - out0, out1 | |
| Details : Load 8 halfword elements in 'out0' from (psrc) | |
| Load 8 halfword elements in 'out1' from (psrc + stride) | |
| */ | |
| #define LD_H2(RTYPE, psrc, stride, out0, out1) \ | |
| { \ | |
| out0 = LD_H(RTYPE, (psrc)); \ | |
| out1 = LD_H(RTYPE, (psrc) + (stride)); \ | |
| } | |
| #define LD_SH2(...) LD_H2(v8i16, __VA_ARGS__) | |
| #define LD_H4(RTYPE, psrc, stride, out0, out1, out2, out3) \ | |
| { \ | |
| LD_H2(RTYPE, (psrc), stride, out0, out1); \ | |
| LD_H2(RTYPE, (psrc) + 2 * stride, stride, out2, out3); \ | |
| } | |
| #define LD_SH4(...) LD_H4(v8i16, __VA_ARGS__) | |
| #define LD_H8(RTYPE, psrc, stride, out0, out1, out2, out3, out4, out5, out6, \ | |
| out7) \ | |
| { \ | |
| LD_H4(RTYPE, (psrc), stride, out0, out1, out2, out3); \ | |
| LD_H4(RTYPE, (psrc) + 4 * stride, stride, out4, out5, out6, out7); \ | |
| } | |
| #define LD_SH8(...) LD_H8(v8i16, __VA_ARGS__) | |
| #define LD_H16(RTYPE, psrc, stride, out0, out1, out2, out3, out4, out5, out6, \ | |
| out7, out8, out9, out10, out11, out12, out13, out14, out15) \ | |
| { \ | |
| LD_H8(RTYPE, (psrc), stride, out0, out1, out2, out3, out4, out5, out6, \ | |
| out7); \ | |
| LD_H8(RTYPE, (psrc) + 8 * stride, stride, out8, out9, out10, out11, out12, \ | |
| out13, out14, out15); \ | |
| } | |
| #define LD_SH16(...) LD_H16(v8i16, __VA_ARGS__) | |
| /* Description : Load 4x4 block of signed halfword elements from 1D source | |
| data into 4 vectors (Each vector with 4 signed halfwords) | |
| Arguments : Input - psrc | |
| Outputs - out0, out1, out2, out3 | |
| */ | |
| #define LD4x4_SH(psrc, out0, out1, out2, out3) \ | |
| { \ | |
| out0 = LD_SH(psrc); \ | |
| out2 = LD_SH(psrc + 8); \ | |
| out1 = (v8i16)__msa_ilvl_d((v2i64)out0, (v2i64)out0); \ | |
| out3 = (v8i16)__msa_ilvl_d((v2i64)out2, (v2i64)out2); \ | |
| } | |
| /* Description : Load 2 vectors of signed word elements with stride | |
| Arguments : Inputs - psrc, stride | |
| Outputs - out0, out1 | |
| Return Type - signed word | |
| */ | |
| #define LD_SW2(psrc, stride, out0, out1) \ | |
| { \ | |
| out0 = LD_SW((psrc)); \ | |
| out1 = LD_SW((psrc) + stride); \ | |
| } | |
| /* Description : Store vectors of 16 byte elements with stride | |
| Arguments : Inputs - in0, in1, pdst, stride | |
| Details : Store 16 byte elements from 'in0' to (pdst) | |
| Store 16 byte elements from 'in1' to (pdst + stride) | |
| */ | |
| #define ST_B2(RTYPE, in0, in1, pdst, stride) \ | |
| { \ | |
| ST_B(RTYPE, in0, (pdst)); \ | |
| ST_B(RTYPE, in1, (pdst) + stride); \ | |
| } | |
| #define ST_UB2(...) ST_B2(v16u8, __VA_ARGS__) | |
| #define ST_B4(RTYPE, in0, in1, in2, in3, pdst, stride) \ | |
| { \ | |
| ST_B2(RTYPE, in0, in1, (pdst), stride); \ | |
| ST_B2(RTYPE, in2, in3, (pdst) + 2 * stride, stride); \ | |
| } | |
| #define ST_UB4(...) ST_B4(v16u8, __VA_ARGS__) | |
| #define ST_B8(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, pdst, stride) \ | |
| { \ | |
| ST_B4(RTYPE, in0, in1, in2, in3, pdst, stride); \ | |
| ST_B4(RTYPE, in4, in5, in6, in7, (pdst) + 4 * stride, stride); \ | |
| } | |
| #define ST_UB8(...) ST_B8(v16u8, __VA_ARGS__) | |
| /* Description : Store vectors of 8 halfword elements with stride | |
| Arguments : Inputs - in0, in1, pdst, stride | |
| Details : Store 8 halfword elements from 'in0' to (pdst) | |
| Store 8 halfword elements from 'in1' to (pdst + stride) | |
| */ | |
| #define ST_H2(RTYPE, in0, in1, pdst, stride) \ | |
| { \ | |
| ST_H(RTYPE, in0, (pdst)); \ | |
| ST_H(RTYPE, in1, (pdst) + stride); \ | |
| } | |
| #define ST_SH2(...) ST_H2(v8i16, __VA_ARGS__) | |
| #define ST_H4(RTYPE, in0, in1, in2, in3, pdst, stride) \ | |
| { \ | |
| ST_H2(RTYPE, in0, in1, (pdst), stride); \ | |
| ST_H2(RTYPE, in2, in3, (pdst) + 2 * stride, stride); \ | |
| } | |
| #define ST_SH4(...) ST_H4(v8i16, __VA_ARGS__) | |
| #define ST_H8(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, pdst, stride) \ | |
| { \ | |
| ST_H4(RTYPE, in0, in1, in2, in3, (pdst), stride); \ | |
| ST_H4(RTYPE, in4, in5, in6, in7, (pdst) + 4 * stride, stride); \ | |
| } | |
| #define ST_SH8(...) ST_H8(v8i16, __VA_ARGS__) | |
| /* Description : Store vectors of word elements with stride | |
| Arguments : Inputs - in0, in1, pdst, stride | |
| Details : Store 4 word elements from 'in0' to (pdst) | |
| Store 4 word elements from 'in1' to (pdst + stride) | |
| */ | |
| #define ST_SW2(in0, in1, pdst, stride) \ | |
| { \ | |
| ST_SW(in0, (pdst)); \ | |
| ST_SW(in1, (pdst) + stride); \ | |
| } | |
| /* Description : Store 2x4 byte block to destination memory from input vector | |
| Arguments : Inputs - in, stidx, pdst, stride | |
| Details : Index 'stidx' halfword element from 'in' vector is copied to | |
| the GP register and stored to (pdst) | |
| Index 'stidx+1' halfword element from 'in' vector is copied to | |
| the GP register and stored to (pdst + stride) | |
| Index 'stidx+2' halfword element from 'in' vector is copied to | |
| the GP register and stored to (pdst + 2 * stride) | |
| Index 'stidx+3' halfword element from 'in' vector is copied to | |
| the GP register and stored to (pdst + 3 * stride) | |
| */ | |
| #define ST2x4_UB(in, stidx, pdst, stride) \ | |
| { \ | |
| uint16_t out0_m, out1_m, out2_m, out3_m; \ | |
| uint8_t *pblk_2x4_m = (uint8_t *)(pdst); \ | |
| \ | |
| out0_m = __msa_copy_u_h((v8i16)in, (stidx)); \ | |
| out1_m = __msa_copy_u_h((v8i16)in, (stidx + 1)); \ | |
| out2_m = __msa_copy_u_h((v8i16)in, (stidx + 2)); \ | |
| out3_m = __msa_copy_u_h((v8i16)in, (stidx + 3)); \ | |
| \ | |
| SH(out0_m, pblk_2x4_m); \ | |
| SH(out1_m, pblk_2x4_m + stride); \ | |
| SH(out2_m, pblk_2x4_m + 2 * stride); \ | |
| SH(out3_m, pblk_2x4_m + 3 * stride); \ | |
| } | |
| /* Description : Store 4x2 byte block to destination memory from input vector | |
| Arguments : Inputs - in, pdst, stride | |
| Details : Index 0 word element from 'in' vector is copied to the GP | |
| register and stored to (pdst) | |
| Index 1 word element from 'in' vector is copied to the GP | |
| register and stored to (pdst + stride) | |
| */ | |
| #define ST4x2_UB(in, pdst, stride) \ | |
| { \ | |
| uint32_t out0_m, out1_m; \ | |
| uint8_t *pblk_4x2_m = (uint8_t *)(pdst); \ | |
| \ | |
| out0_m = __msa_copy_u_w((v4i32)in, 0); \ | |
| out1_m = __msa_copy_u_w((v4i32)in, 1); \ | |
| \ | |
| SW(out0_m, pblk_4x2_m); \ | |
| SW(out1_m, pblk_4x2_m + stride); \ | |
| } | |
| /* Description : Store 4x4 byte block to destination memory from input vector | |
| Arguments : Inputs - in0, in1, pdst, stride | |
| Details : 'Idx0' word element from input vector 'in0' is copied to the | |
| GP register and stored to (pdst) | |
| 'Idx1' word element from input vector 'in0' is copied to the | |
| GP register and stored to (pdst + stride) | |
| 'Idx2' word element from input vector 'in0' is copied to the | |
| GP register and stored to (pdst + 2 * stride) | |
| 'Idx3' word element from input vector 'in0' is copied to the | |
| GP register and stored to (pdst + 3 * stride) | |
| */ | |
| #define ST4x4_UB(in0, in1, idx0, idx1, idx2, idx3, pdst, stride) \ | |
| { \ | |
| uint32_t out0_m, out1_m, out2_m, out3_m; \ | |
| uint8_t *pblk_4x4_m = (uint8_t *)(pdst); \ | |
| \ | |
| out0_m = __msa_copy_u_w((v4i32)in0, idx0); \ | |
| out1_m = __msa_copy_u_w((v4i32)in0, idx1); \ | |
| out2_m = __msa_copy_u_w((v4i32)in1, idx2); \ | |
| out3_m = __msa_copy_u_w((v4i32)in1, idx3); \ | |
| \ | |
| SW4(out0_m, out1_m, out2_m, out3_m, pblk_4x4_m, stride); \ | |
| } | |
| #define ST4x8_UB(in0, in1, pdst, stride) \ | |
| { \ | |
| uint8_t *pblk_4x8 = (uint8_t *)(pdst); \ | |
| \ | |
| ST4x4_UB(in0, in0, 0, 1, 2, 3, pblk_4x8, stride); \ | |
| ST4x4_UB(in1, in1, 0, 1, 2, 3, pblk_4x8 + 4 * stride, stride); \ | |
| } | |
| /* Description : Store 8x1 byte block to destination memory from input vector | |
| Arguments : Inputs - in, pdst | |
| Details : Index 0 double word element from 'in' vector is copied to the | |
| GP register and stored to (pdst) | |
| */ | |
| #define ST8x1_UB(in, pdst) \ | |
| { \ | |
| uint64_t out0_m; \ | |
| \ | |
| out0_m = __msa_copy_u_d((v2i64)in, 0); \ | |
| SD(out0_m, pdst); \ | |
| } | |
| /* Description : Store 8x2 byte block to destination memory from input vector | |
| Arguments : Inputs - in, pdst, stride | |
| Details : Index 0 double word element from 'in' vector is copied to the | |
| GP register and stored to (pdst) | |
| Index 1 double word element from 'in' vector is copied to the | |
| GP register and stored to (pdst + stride) | |
| */ | |
| #define ST8x2_UB(in, pdst, stride) \ | |
| { \ | |
| uint64_t out0_m, out1_m; \ | |
| uint8_t *pblk_8x2_m = (uint8_t *)(pdst); \ | |
| \ | |
| out0_m = __msa_copy_u_d((v2i64)in, 0); \ | |
| out1_m = __msa_copy_u_d((v2i64)in, 1); \ | |
| \ | |
| SD(out0_m, pblk_8x2_m); \ | |
| SD(out1_m, pblk_8x2_m + stride); \ | |
| } | |
| /* Description : Store 8x4 byte block to destination memory from input | |
| vectors | |
| Arguments : Inputs - in0, in1, pdst, stride | |
| Details : Index 0 double word element from 'in0' vector is copied to the | |
| GP register and stored to (pdst) | |
| Index 1 double word element from 'in0' vector is copied to the | |
| GP register and stored to (pdst + stride) | |
| Index 0 double word element from 'in1' vector is copied to the | |
| GP register and stored to (pdst + 2 * stride) | |
| Index 1 double word element from 'in1' vector is copied to the | |
| GP register and stored to (pdst + 3 * stride) | |
| */ | |
| #define ST8x4_UB(in0, in1, pdst, stride) \ | |
| { \ | |
| uint64_t out0_m, out1_m, out2_m, out3_m; \ | |
| uint8_t *pblk_8x4_m = (uint8_t *)(pdst); \ | |
| \ | |
| out0_m = __msa_copy_u_d((v2i64)in0, 0); \ | |
| out1_m = __msa_copy_u_d((v2i64)in0, 1); \ | |
| out2_m = __msa_copy_u_d((v2i64)in1, 0); \ | |
| out3_m = __msa_copy_u_d((v2i64)in1, 1); \ | |
| \ | |
| SD4(out0_m, out1_m, out2_m, out3_m, pblk_8x4_m, stride); \ | |
| } | |
| /* Description : average with rounding (in0 + in1 + 1) / 2. | |
| Arguments : Inputs - in0, in1, in2, in3, | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Each unsigned byte element from 'in0' vector is added with | |
| each unsigned byte element from 'in1' vector. Then the average | |
| with rounding is calculated and written to 'out0' | |
| */ | |
| #define AVER_UB2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_aver_u_b((v16u8)in0, (v16u8)in1); \ | |
| out1 = (RTYPE)__msa_aver_u_b((v16u8)in2, (v16u8)in3); \ | |
| } | |
| #define AVER_UB2_UB(...) AVER_UB2(v16u8, __VA_ARGS__) | |
| #define AVER_UB4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| AVER_UB2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| AVER_UB2(RTYPE, in4, in5, in6, in7, out2, out3) \ | |
| } | |
| #define AVER_UB4_UB(...) AVER_UB4(v16u8, __VA_ARGS__) | |
| /* Description : Immediate number of elements to slide with zero | |
| Arguments : Inputs - in0, in1, slide_val | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Byte elements from 'zero_m' vector are slid into 'in0' by | |
| value specified in the 'slide_val' | |
| */ | |
| #define SLDI_B2_0(RTYPE, in0, in1, out0, out1, slide_val) \ | |
| { \ | |
| v16i8 zero_m = { 0 }; \ | |
| out0 = (RTYPE)__msa_sldi_b((v16i8)zero_m, (v16i8)in0, slide_val); \ | |
| out1 = (RTYPE)__msa_sldi_b((v16i8)zero_m, (v16i8)in1, slide_val); \ | |
| } | |
| #define SLDI_B2_0_SW(...) SLDI_B2_0(v4i32, __VA_ARGS__) | |
| #define SLDI_B4_0(RTYPE, in0, in1, in2, in3, out0, out1, out2, out3, \ | |
| slide_val) \ | |
| { \ | |
| SLDI_B2_0(RTYPE, in0, in1, out0, out1, slide_val); \ | |
| SLDI_B2_0(RTYPE, in2, in3, out2, out3, slide_val); \ | |
| } | |
| #define SLDI_B4_0_UB(...) SLDI_B4_0(v16u8, __VA_ARGS__) | |
| /* Description : Immediate number of elements to slide | |
| Arguments : Inputs - in0_0, in0_1, in1_0, in1_1, slide_val | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Byte elements from 'in0_0' vector are slid into 'in1_0' by | |
| value specified in the 'slide_val' | |
| */ | |
| #define SLDI_B2(RTYPE, in0_0, in0_1, in1_0, in1_1, out0, out1, slide_val) \ | |
| { \ | |
| out0 = (RTYPE)__msa_sldi_b((v16i8)in0_0, (v16i8)in1_0, slide_val); \ | |
| out1 = (RTYPE)__msa_sldi_b((v16i8)in0_1, (v16i8)in1_1, slide_val); \ | |
| } | |
| #define SLDI_B2_UB(...) SLDI_B2(v16u8, __VA_ARGS__) | |
| #define SLDI_B2_SH(...) SLDI_B2(v8i16, __VA_ARGS__) | |
| #define SLDI_B3(RTYPE, in0_0, in0_1, in0_2, in1_0, in1_1, in1_2, out0, out1, \ | |
| out2, slide_val) \ | |
| { \ | |
| SLDI_B2(RTYPE, in0_0, in0_1, in1_0, in1_1, out0, out1, slide_val) \ | |
| out2 = (RTYPE)__msa_sldi_b((v16i8)in0_2, (v16i8)in1_2, slide_val); \ | |
| } | |
| #define SLDI_B3_SB(...) SLDI_B3(v16i8, __VA_ARGS__) | |
| #define SLDI_B3_UH(...) SLDI_B3(v8u16, __VA_ARGS__) | |
| /* Description : Shuffle byte vector elements as per mask vector | |
| Arguments : Inputs - in0, in1, in2, in3, mask0, mask1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Byte elements from 'in0' & 'in1' are copied selectively to | |
| 'out0' as per control vector 'mask0' | |
| */ | |
| #define VSHF_B2(RTYPE, in0, in1, in2, in3, mask0, mask1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_vshf_b((v16i8)mask0, (v16i8)in1, (v16i8)in0); \ | |
| out1 = (RTYPE)__msa_vshf_b((v16i8)mask1, (v16i8)in3, (v16i8)in2); \ | |
| } | |
| #define VSHF_B2_UB(...) VSHF_B2(v16u8, __VA_ARGS__) | |
| #define VSHF_B2_SB(...) VSHF_B2(v16i8, __VA_ARGS__) | |
| #define VSHF_B2_UH(...) VSHF_B2(v8u16, __VA_ARGS__) | |
| #define VSHF_B4(RTYPE, in0, in1, mask0, mask1, mask2, mask3, out0, out1, out2, \ | |
| out3) \ | |
| { \ | |
| VSHF_B2(RTYPE, in0, in1, in0, in1, mask0, mask1, out0, out1); \ | |
| VSHF_B2(RTYPE, in0, in1, in0, in1, mask2, mask3, out2, out3); \ | |
| } | |
| #define VSHF_B4_SB(...) VSHF_B4(v16i8, __VA_ARGS__) | |
| #define VSHF_B4_SH(...) VSHF_B4(v8i16, __VA_ARGS__) | |
| /* Description : Dot product of byte vector elements | |
| Arguments : Inputs - mult0, mult1, cnst0, cnst1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Unsigned byte elements from 'mult0' are multiplied with | |
| unsigned byte elements from 'cnst0' producing a result | |
| twice the size of input i.e. unsigned halfword. | |
| The multiplication result of adjacent odd-even elements | |
| are added together and written to the 'out0' vector | |
| */ | |
| #define DOTP_UB2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_dotp_u_h((v16u8)mult0, (v16u8)cnst0); \ | |
| out1 = (RTYPE)__msa_dotp_u_h((v16u8)mult1, (v16u8)cnst1); \ | |
| } | |
| #define DOTP_UB2_UH(...) DOTP_UB2(v8u16, __VA_ARGS__) | |
| #define DOTP_UB4(RTYPE, mult0, mult1, mult2, mult3, cnst0, cnst1, cnst2, \ | |
| cnst3, out0, out1, out2, out3) \ | |
| { \ | |
| DOTP_UB2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1); \ | |
| DOTP_UB2(RTYPE, mult2, mult3, cnst2, cnst3, out2, out3); \ | |
| } | |
| #define DOTP_UB4_UH(...) DOTP_UB4(v8u16, __VA_ARGS__) | |
| /* Description : Dot product of byte vector elements | |
| Arguments : Inputs - mult0, mult1, cnst0, cnst1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Signed byte elements from 'mult0' are multiplied with | |
| signed byte elements from 'cnst0' producing a result | |
| twice the size of input i.e. signed halfword. | |
| The multiplication result of adjacent odd-even elements | |
| are added together and written to the 'out0' vector | |
| */ | |
| #define DOTP_SB2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_dotp_s_h((v16i8)mult0, (v16i8)cnst0); \ | |
| out1 = (RTYPE)__msa_dotp_s_h((v16i8)mult1, (v16i8)cnst1); \ | |
| } | |
| #define DOTP_SB2_SH(...) DOTP_SB2(v8i16, __VA_ARGS__) | |
| #define DOTP_SB4(RTYPE, mult0, mult1, mult2, mult3, cnst0, cnst1, cnst2, \ | |
| cnst3, out0, out1, out2, out3) \ | |
| { \ | |
| DOTP_SB2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1); \ | |
| DOTP_SB2(RTYPE, mult2, mult3, cnst2, cnst3, out2, out3); \ | |
| } | |
| #define DOTP_SB4_SH(...) DOTP_SB4(v8i16, __VA_ARGS__) | |
| /* Description : Dot product of halfword vector elements | |
| Arguments : Inputs - mult0, mult1, cnst0, cnst1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Signed halfword elements from 'mult0' are multiplied with | |
| signed halfword elements from 'cnst0' producing a result | |
| twice the size of input i.e. signed word. | |
| The multiplication result of adjacent odd-even elements | |
| are added together and written to the 'out0' vector | |
| */ | |
| #define DOTP_SH2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_dotp_s_w((v8i16)mult0, (v8i16)cnst0); \ | |
| out1 = (RTYPE)__msa_dotp_s_w((v8i16)mult1, (v8i16)cnst1); \ | |
| } | |
| #define DOTP_SH2_SW(...) DOTP_SH2(v4i32, __VA_ARGS__) | |
| #define DOTP_SH4(RTYPE, mult0, mult1, mult2, mult3, cnst0, cnst1, cnst2, \ | |
| cnst3, out0, out1, out2, out3) \ | |
| { \ | |
| DOTP_SH2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1); \ | |
| DOTP_SH2(RTYPE, mult2, mult3, cnst2, cnst3, out2, out3); \ | |
| } | |
| #define DOTP_SH4_SW(...) DOTP_SH4(v4i32, __VA_ARGS__) | |
| /* Description : Dot product of word vector elements | |
| Arguments : Inputs - mult0, mult1, cnst0, cnst1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Signed word elements from 'mult0' are multiplied with | |
| signed word elements from 'cnst0' producing a result | |
| twice the size of input i.e. signed double word. | |
| The multiplication result of adjacent odd-even elements | |
| are added together and written to the 'out0' vector | |
| */ | |
| #define DOTP_SW2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_dotp_s_d((v4i32)mult0, (v4i32)cnst0); \ | |
| out1 = (RTYPE)__msa_dotp_s_d((v4i32)mult1, (v4i32)cnst1); \ | |
| } | |
| #define DOTP_SW2_SD(...) DOTP_SW2(v2i64, __VA_ARGS__) | |
| /* Description : Dot product & addition of byte vector elements | |
| Arguments : Inputs - mult0, mult1, cnst0, cnst1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Signed byte elements from 'mult0' are multiplied with | |
| signed byte elements from 'cnst0' producing a result | |
| twice the size of input i.e. signed halfword. | |
| The multiplication result of adjacent odd-even elements | |
| are added to the 'out0' vector | |
| */ | |
| #define DPADD_SB2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_dpadd_s_h((v8i16)out0, (v16i8)mult0, (v16i8)cnst0); \ | |
| out1 = (RTYPE)__msa_dpadd_s_h((v8i16)out1, (v16i8)mult1, (v16i8)cnst1); \ | |
| } | |
| #define DPADD_SB2_SH(...) DPADD_SB2(v8i16, __VA_ARGS__) | |
| #define DPADD_SB4(RTYPE, mult0, mult1, mult2, mult3, cnst0, cnst1, cnst2, \ | |
| cnst3, out0, out1, out2, out3) \ | |
| { \ | |
| DPADD_SB2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1); \ | |
| DPADD_SB2(RTYPE, mult2, mult3, cnst2, cnst3, out2, out3); \ | |
| } | |
| #define DPADD_SB4_SH(...) DPADD_SB4(v8i16, __VA_ARGS__) | |
| /* Description : Dot product & addition of halfword vector elements | |
| Arguments : Inputs - mult0, mult1, cnst0, cnst1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Signed halfword elements from 'mult0' are multiplied with | |
| signed halfword elements from 'cnst0' producing a result | |
| twice the size of input i.e. signed word. | |
| The multiplication result of adjacent odd-even elements | |
| are added to the 'out0' vector | |
| */ | |
| #define DPADD_SH2(RTYPE, mult0, mult1, cnst0, cnst1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_dpadd_s_w((v4i32)out0, (v8i16)mult0, (v8i16)cnst0); \ | |
| out1 = (RTYPE)__msa_dpadd_s_w((v4i32)out1, (v8i16)mult1, (v8i16)cnst1); \ | |
| } | |
| #define DPADD_SH2_SW(...) DPADD_SH2(v4i32, __VA_ARGS__) | |
| /* Description : Dot product & addition of double word vector elements | |
| Arguments : Inputs - mult0, mult1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Each signed word element from 'mult0' is multiplied with itself | |
| producing an intermediate result twice the size of input | |
| i.e. signed double word | |
| The multiplication result of adjacent odd-even elements | |
| are added to the 'out0' vector | |
| */ | |
| #define DPADD_SD2(RTYPE, mult0, mult1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_dpadd_s_d((v2i64)out0, (v4i32)mult0, (v4i32)mult0); \ | |
| out1 = (RTYPE)__msa_dpadd_s_d((v2i64)out1, (v4i32)mult1, (v4i32)mult1); \ | |
| } | |
| #define DPADD_SD2_SD(...) DPADD_SD2(v2i64, __VA_ARGS__) | |
| /* Description : Minimum values between unsigned elements of | |
| either vector are copied to the output vector | |
| Arguments : Inputs - in0, in1, min_vec | |
| Outputs - in place operation | |
| Return Type - as per RTYPE | |
| Details : Minimum of unsigned halfword element values from 'in0' and | |
| 'min_vec' are written to output vector 'in0' | |
| */ | |
| #define MIN_UH2(RTYPE, in0, in1, min_vec) \ | |
| { \ | |
| in0 = (RTYPE)__msa_min_u_h((v8u16)in0, min_vec); \ | |
| in1 = (RTYPE)__msa_min_u_h((v8u16)in1, min_vec); \ | |
| } | |
| #define MIN_UH2_UH(...) MIN_UH2(v8u16, __VA_ARGS__) | |
| #define MIN_UH4(RTYPE, in0, in1, in2, in3, min_vec) \ | |
| { \ | |
| MIN_UH2(RTYPE, in0, in1, min_vec); \ | |
| MIN_UH2(RTYPE, in2, in3, min_vec); \ | |
| } | |
| #define MIN_UH4_UH(...) MIN_UH4(v8u16, __VA_ARGS__) | |
| /* Description : Clips all signed halfword elements of input vector | |
| between 0 & 255 | |
| Arguments : Input - in | |
| Output - out_m | |
| Return Type - signed halfword | |
| */ | |
| #define CLIP_SH_0_255(in) \ | |
| ({ \ | |
| v8i16 max_m = __msa_ldi_h(255); \ | |
| v8i16 out_m; \ | |
| \ | |
| out_m = __msa_maxi_s_h((v8i16)in, 0); \ | |
| out_m = __msa_min_s_h((v8i16)max_m, (v8i16)out_m); \ | |
| out_m; \ | |
| }) | |
| #define CLIP_SH2_0_255(in0, in1) \ | |
| { \ | |
| in0 = CLIP_SH_0_255(in0); \ | |
| in1 = CLIP_SH_0_255(in1); \ | |
| } | |
| #define CLIP_SH4_0_255(in0, in1, in2, in3) \ | |
| { \ | |
| CLIP_SH2_0_255(in0, in1); \ | |
| CLIP_SH2_0_255(in2, in3); \ | |
| } | |
| /* Description : Horizontal addition of 4 signed word elements of input vector | |
| Arguments : Input - in (signed word vector) | |
| Output - sum_m (i32 sum) | |
| Return Type - signed word (GP) | |
| Details : 4 signed word elements of 'in' vector are added together and | |
| the resulting integer sum is returned | |
| */ | |
| #define HADD_SW_S32(in) \ | |
| ({ \ | |
| v2i64 res0_m, res1_m; \ | |
| int32_t sum_m; \ | |
| \ | |
| res0_m = __msa_hadd_s_d((v4i32)in, (v4i32)in); \ | |
| res1_m = __msa_splati_d(res0_m, 1); \ | |
| res0_m = res0_m + res1_m; \ | |
| sum_m = __msa_copy_s_w((v4i32)res0_m, 0); \ | |
| sum_m; \ | |
| }) | |
| /* Description : Horizontal addition of 8 unsigned halfword elements | |
| Arguments : Inputs - in (unsigned halfword vector) | |
| Outputs - sum_m (u32 sum) | |
| Return Type - unsigned word | |
| Details : 8 unsigned halfword elements of input vector are added | |
| together and the resulting integer sum is returned | |
| */ | |
| #define HADD_UH_U32(in) \ | |
| ({ \ | |
| v4u32 res_m; \ | |
| v2u64 res0_m, res1_m; \ | |
| uint32_t sum_m; \ | |
| \ | |
| res_m = __msa_hadd_u_w((v8u16)in, (v8u16)in); \ | |
| res0_m = __msa_hadd_u_d(res_m, res_m); \ | |
| res1_m = (v2u64)__msa_splati_d((v2i64)res0_m, 1); \ | |
| res0_m = res0_m + res1_m; \ | |
| sum_m = __msa_copy_u_w((v4i32)res0_m, 0); \ | |
| sum_m; \ | |
| }) | |
| /* Description : Horizontal addition of unsigned byte vector elements | |
| Arguments : Inputs - in0, in1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Each unsigned odd byte element from 'in0' is added to | |
| even unsigned byte element from 'in0' (pairwise) and the | |
| halfword result is written to 'out0' | |
| */ | |
| #define HADD_UB2(RTYPE, in0, in1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_hadd_u_h((v16u8)in0, (v16u8)in0); \ | |
| out1 = (RTYPE)__msa_hadd_u_h((v16u8)in1, (v16u8)in1); \ | |
| } | |
| #define HADD_UB2_UH(...) HADD_UB2(v8u16, __VA_ARGS__) | |
| #define HADD_UB4(RTYPE, in0, in1, in2, in3, out0, out1, out2, out3) \ | |
| { \ | |
| HADD_UB2(RTYPE, in0, in1, out0, out1); \ | |
| HADD_UB2(RTYPE, in2, in3, out2, out3); \ | |
| } | |
| #define HADD_UB4_UH(...) HADD_UB4(v8u16, __VA_ARGS__) | |
| /* Description : Horizontal subtraction of unsigned byte vector elements | |
| Arguments : Inputs - in0, in1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Each unsigned odd byte element from 'in0' is subtracted from | |
| even unsigned byte element from 'in0' (pairwise) and the | |
| halfword result is written to 'out0' | |
| */ | |
| #define HSUB_UB2(RTYPE, in0, in1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_hsub_u_h((v16u8)in0, (v16u8)in0); \ | |
| out1 = (RTYPE)__msa_hsub_u_h((v16u8)in1, (v16u8)in1); \ | |
| } | |
| #define HSUB_UB2_SH(...) HSUB_UB2(v8i16, __VA_ARGS__) | |
| /* Description : SAD (Sum of Absolute Difference) | |
| Arguments : Inputs - in0, in1, ref0, ref1 | |
| Outputs - sad_m (halfword vector) | |
| Return Type - unsigned halfword | |
| Details : Absolute difference of all the byte elements from 'in0' with | |
| 'ref0' is calculated and preserved in 'diff0'. Then even-odd | |
| pairs are added together to generate 8 halfword results. | |
| */ | |
| #define SAD_UB2_UH(in0, in1, ref0, ref1) \ | |
| ({ \ | |
| v16u8 diff0_m, diff1_m; \ | |
| v8u16 sad_m = { 0 }; \ | |
| \ | |
| diff0_m = __msa_asub_u_b((v16u8)in0, (v16u8)ref0); \ | |
| diff1_m = __msa_asub_u_b((v16u8)in1, (v16u8)ref1); \ | |
| \ | |
| sad_m += __msa_hadd_u_h((v16u8)diff0_m, (v16u8)diff0_m); \ | |
| sad_m += __msa_hadd_u_h((v16u8)diff1_m, (v16u8)diff1_m); \ | |
| \ | |
| sad_m; \ | |
| }) | |
| /* Description : Horizontal subtraction of signed halfword vector elements | |
| Arguments : Inputs - in0, in1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Each signed odd halfword element from 'in0' is subtracted from | |
| even signed halfword element from 'in0' (pairwise) and the | |
| word result is written to 'out0' | |
| */ | |
| #define HSUB_UH2(RTYPE, in0, in1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_hsub_s_w((v8i16)in0, (v8i16)in0); \ | |
| out1 = (RTYPE)__msa_hsub_s_w((v8i16)in1, (v8i16)in1); \ | |
| } | |
| #define HSUB_UH2_SW(...) HSUB_UH2(v4i32, __VA_ARGS__) | |
| /* Description : Set element n input vector to GPR value | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Output - out | |
| Return Type - as per RTYPE | |
| Details : Set element 0 in vector 'out' to value specified in 'in0' | |
| */ | |
| #define INSERT_W2(RTYPE, in0, in1, out) \ | |
| { \ | |
| out = (RTYPE)__msa_insert_w((v4i32)out, 0, in0); \ | |
| out = (RTYPE)__msa_insert_w((v4i32)out, 1, in1); \ | |
| } | |
| #define INSERT_W2_SB(...) INSERT_W2(v16i8, __VA_ARGS__) | |
| #define INSERT_W4(RTYPE, in0, in1, in2, in3, out) \ | |
| { \ | |
| out = (RTYPE)__msa_insert_w((v4i32)out, 0, in0); \ | |
| out = (RTYPE)__msa_insert_w((v4i32)out, 1, in1); \ | |
| out = (RTYPE)__msa_insert_w((v4i32)out, 2, in2); \ | |
| out = (RTYPE)__msa_insert_w((v4i32)out, 3, in3); \ | |
| } | |
| #define INSERT_W4_UB(...) INSERT_W4(v16u8, __VA_ARGS__) | |
| #define INSERT_W4_SB(...) INSERT_W4(v16i8, __VA_ARGS__) | |
| #define INSERT_D2(RTYPE, in0, in1, out) \ | |
| { \ | |
| out = (RTYPE)__msa_insert_d((v2i64)out, 0, in0); \ | |
| out = (RTYPE)__msa_insert_d((v2i64)out, 1, in1); \ | |
| } | |
| #define INSERT_D2_UB(...) INSERT_D2(v16u8, __VA_ARGS__) | |
| #define INSERT_D2_SB(...) INSERT_D2(v16i8, __VA_ARGS__) | |
| /* Description : Interleave even byte elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Even byte elements of 'in0' and 'in1' are interleaved | |
| and written to 'out0' | |
| */ | |
| #define ILVEV_B2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvev_b((v16i8)in1, (v16i8)in0); \ | |
| out1 = (RTYPE)__msa_ilvev_b((v16i8)in3, (v16i8)in2); \ | |
| } | |
| #define ILVEV_B2_UB(...) ILVEV_B2(v16u8, __VA_ARGS__) | |
| #define ILVEV_B2_SH(...) ILVEV_B2(v8i16, __VA_ARGS__) | |
| /* Description : Interleave even halfword elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Even halfword elements of 'in0' and 'in1' are interleaved | |
| and written to 'out0' | |
| */ | |
| #define ILVEV_H2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvev_h((v8i16)in1, (v8i16)in0); \ | |
| out1 = (RTYPE)__msa_ilvev_h((v8i16)in3, (v8i16)in2); \ | |
| } | |
| #define ILVEV_H2_UB(...) ILVEV_H2(v16u8, __VA_ARGS__) | |
| #define ILVEV_H2_SH(...) ILVEV_H2(v8i16, __VA_ARGS__) | |
| #define ILVEV_H2_SW(...) ILVEV_H2(v4i32, __VA_ARGS__) | |
| /* Description : Interleave even word elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Even word elements of 'in0' and 'in1' are interleaved | |
| and written to 'out0' | |
| */ | |
| #define ILVEV_W2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvev_w((v4i32)in1, (v4i32)in0); \ | |
| out1 = (RTYPE)__msa_ilvev_w((v4i32)in3, (v4i32)in2); \ | |
| } | |
| #define ILVEV_W2_SB(...) ILVEV_W2(v16i8, __VA_ARGS__) | |
| /* Description : Interleave even double word elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Even double word elements of 'in0' and 'in1' are interleaved | |
| and written to 'out0' | |
| */ | |
| #define ILVEV_D2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvev_d((v2i64)in1, (v2i64)in0); \ | |
| out1 = (RTYPE)__msa_ilvev_d((v2i64)in3, (v2i64)in2); \ | |
| } | |
| #define ILVEV_D2_UB(...) ILVEV_D2(v16u8, __VA_ARGS__) | |
| /* Description : Interleave left half of byte elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Left half of byte elements of 'in0' and 'in1' are interleaved | |
| and written to 'out0'. | |
| */ | |
| #define ILVL_B2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvl_b((v16i8)in0, (v16i8)in1); \ | |
| out1 = (RTYPE)__msa_ilvl_b((v16i8)in2, (v16i8)in3); \ | |
| } | |
| #define ILVL_B2_UB(...) ILVL_B2(v16u8, __VA_ARGS__) | |
| #define ILVL_B2_SB(...) ILVL_B2(v16i8, __VA_ARGS__) | |
| #define ILVL_B2_UH(...) ILVL_B2(v8u16, __VA_ARGS__) | |
| #define ILVL_B2_SH(...) ILVL_B2(v8i16, __VA_ARGS__) | |
| #define ILVL_B4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| ILVL_B2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| ILVL_B2(RTYPE, in4, in5, in6, in7, out2, out3); \ | |
| } | |
| #define ILVL_B4_SB(...) ILVL_B4(v16i8, __VA_ARGS__) | |
| #define ILVL_B4_SH(...) ILVL_B4(v8i16, __VA_ARGS__) | |
| #define ILVL_B4_UH(...) ILVL_B4(v8u16, __VA_ARGS__) | |
| /* Description : Interleave left half of halfword elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Left half of halfword elements of 'in0' and 'in1' are | |
| interleaved and written to 'out0'. | |
| */ | |
| #define ILVL_H2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvl_h((v8i16)in0, (v8i16)in1); \ | |
| out1 = (RTYPE)__msa_ilvl_h((v8i16)in2, (v8i16)in3); \ | |
| } | |
| #define ILVL_H2_SH(...) ILVL_H2(v8i16, __VA_ARGS__) | |
| #define ILVL_H2_SW(...) ILVL_H2(v4i32, __VA_ARGS__) | |
| /* Description : Interleave left half of word elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Left half of word elements of 'in0' and 'in1' are interleaved | |
| and written to 'out0'. | |
| */ | |
| #define ILVL_W2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvl_w((v4i32)in0, (v4i32)in1); \ | |
| out1 = (RTYPE)__msa_ilvl_w((v4i32)in2, (v4i32)in3); \ | |
| } | |
| #define ILVL_W2_UB(...) ILVL_W2(v16u8, __VA_ARGS__) | |
| #define ILVL_W2_SH(...) ILVL_W2(v8i16, __VA_ARGS__) | |
| /* Description : Interleave right half of byte elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Right half of byte elements of 'in0' and 'in1' are interleaved | |
| and written to out0. | |
| */ | |
| #define ILVR_B2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvr_b((v16i8)in0, (v16i8)in1); \ | |
| out1 = (RTYPE)__msa_ilvr_b((v16i8)in2, (v16i8)in3); \ | |
| } | |
| #define ILVR_B2_UB(...) ILVR_B2(v16u8, __VA_ARGS__) | |
| #define ILVR_B2_SB(...) ILVR_B2(v16i8, __VA_ARGS__) | |
| #define ILVR_B2_UH(...) ILVR_B2(v8u16, __VA_ARGS__) | |
| #define ILVR_B2_SH(...) ILVR_B2(v8i16, __VA_ARGS__) | |
| #define ILVR_B4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| ILVR_B2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| ILVR_B2(RTYPE, in4, in5, in6, in7, out2, out3); \ | |
| } | |
| #define ILVR_B4_UB(...) ILVR_B4(v16u8, __VA_ARGS__) | |
| #define ILVR_B4_SB(...) ILVR_B4(v16i8, __VA_ARGS__) | |
| #define ILVR_B4_UH(...) ILVR_B4(v8u16, __VA_ARGS__) | |
| #define ILVR_B4_SH(...) ILVR_B4(v8i16, __VA_ARGS__) | |
| #define ILVR_B8(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, in8, in9, in10, \ | |
| in11, in12, in13, in14, in15, out0, out1, out2, out3, out4, \ | |
| out5, out6, out7) \ | |
| { \ | |
| ILVR_B4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, out2, \ | |
| out3); \ | |
| ILVR_B4(RTYPE, in8, in9, in10, in11, in12, in13, in14, in15, out4, out5, \ | |
| out6, out7); \ | |
| } | |
| #define ILVR_B8_UH(...) ILVR_B8(v8u16, __VA_ARGS__) | |
| /* Description : Interleave right half of halfword elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Right half of halfword elements of 'in0' and 'in1' are | |
| interleaved and written to 'out0'. | |
| */ | |
| #define ILVR_H2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvr_h((v8i16)in0, (v8i16)in1); \ | |
| out1 = (RTYPE)__msa_ilvr_h((v8i16)in2, (v8i16)in3); \ | |
| } | |
| #define ILVR_H2_SH(...) ILVR_H2(v8i16, __VA_ARGS__) | |
| #define ILVR_H2_SW(...) ILVR_H2(v4i32, __VA_ARGS__) | |
| #define ILVR_H4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| ILVR_H2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| ILVR_H2(RTYPE, in4, in5, in6, in7, out2, out3); \ | |
| } | |
| #define ILVR_H4_SH(...) ILVR_H4(v8i16, __VA_ARGS__) | |
| #define ILVR_W2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvr_w((v4i32)in0, (v4i32)in1); \ | |
| out1 = (RTYPE)__msa_ilvr_w((v4i32)in2, (v4i32)in3); \ | |
| } | |
| #define ILVR_W2_UB(...) ILVR_W2(v16u8, __VA_ARGS__) | |
| #define ILVR_W2_SH(...) ILVR_W2(v8i16, __VA_ARGS__) | |
| #define ILVR_W4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| ILVR_W2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| ILVR_W2(RTYPE, in4, in5, in6, in7, out2, out3); \ | |
| } | |
| #define ILVR_W4_UB(...) ILVR_W4(v16u8, __VA_ARGS__) | |
| /* Description : Interleave right half of double word elements from vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Right half of double word elements of 'in0' and 'in1' are | |
| interleaved and written to 'out0'. | |
| */ | |
| #define ILVR_D2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvr_d((v2i64)(in0), (v2i64)(in1)); \ | |
| out1 = (RTYPE)__msa_ilvr_d((v2i64)(in2), (v2i64)(in3)); \ | |
| } | |
| #define ILVR_D2_UB(...) ILVR_D2(v16u8, __VA_ARGS__) | |
| #define ILVR_D2_SB(...) ILVR_D2(v16i8, __VA_ARGS__) | |
| #define ILVR_D2_SH(...) ILVR_D2(v8i16, __VA_ARGS__) | |
| #define ILVR_D3(RTYPE, in0, in1, in2, in3, in4, in5, out0, out1, out2) \ | |
| { \ | |
| ILVR_D2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| out2 = (RTYPE)__msa_ilvr_d((v2i64)(in4), (v2i64)(in5)); \ | |
| } | |
| #define ILVR_D3_SB(...) ILVR_D3(v16i8, __VA_ARGS__) | |
| #define ILVR_D4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| ILVR_D2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| ILVR_D2(RTYPE, in4, in5, in6, in7, out2, out3); \ | |
| } | |
| #define ILVR_D4_SB(...) ILVR_D4(v16i8, __VA_ARGS__) | |
| #define ILVR_D4_UB(...) ILVR_D4(v16u8, __VA_ARGS__) | |
| /* Description : Interleave both left and right half of input vectors | |
| Arguments : Inputs - in0, in1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Right half of byte elements from 'in0' and 'in1' are | |
| interleaved and written to 'out0' | |
| */ | |
| #define ILVRL_B2(RTYPE, in0, in1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvr_b((v16i8)in0, (v16i8)in1); \ | |
| out1 = (RTYPE)__msa_ilvl_b((v16i8)in0, (v16i8)in1); \ | |
| } | |
| #define ILVRL_B2_UB(...) ILVRL_B2(v16u8, __VA_ARGS__) | |
| #define ILVRL_B2_SB(...) ILVRL_B2(v16i8, __VA_ARGS__) | |
| #define ILVRL_B2_UH(...) ILVRL_B2(v8u16, __VA_ARGS__) | |
| #define ILVRL_B2_SH(...) ILVRL_B2(v8i16, __VA_ARGS__) | |
| #define ILVRL_H2(RTYPE, in0, in1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvr_h((v8i16)in0, (v8i16)in1); \ | |
| out1 = (RTYPE)__msa_ilvl_h((v8i16)in0, (v8i16)in1); \ | |
| } | |
| #define ILVRL_H2_SH(...) ILVRL_H2(v8i16, __VA_ARGS__) | |
| #define ILVRL_H2_SW(...) ILVRL_H2(v4i32, __VA_ARGS__) | |
| #define ILVRL_W2(RTYPE, in0, in1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ilvr_w((v4i32)in0, (v4i32)in1); \ | |
| out1 = (RTYPE)__msa_ilvl_w((v4i32)in0, (v4i32)in1); \ | |
| } | |
| #define ILVRL_W2_UB(...) ILVRL_W2(v16u8, __VA_ARGS__) | |
| #define ILVRL_W2_SH(...) ILVRL_W2(v8i16, __VA_ARGS__) | |
| #define ILVRL_W2_SW(...) ILVRL_W2(v4i32, __VA_ARGS__) | |
| /* Description : Saturate the halfword element values to the max | |
| unsigned value of (sat_val + 1) bits | |
| The element data width remains unchanged | |
| Arguments : Inputs - in0, in1, sat_val | |
| Outputs - in place operation | |
| Return Type - as per RTYPE | |
| Details : Each unsigned halfword element from 'in0' is saturated to the | |
| value generated with (sat_val + 1) bit range. | |
| The results are written in place | |
| */ | |
| #define SAT_UH2(RTYPE, in0, in1, sat_val) \ | |
| { \ | |
| in0 = (RTYPE)__msa_sat_u_h((v8u16)in0, sat_val); \ | |
| in1 = (RTYPE)__msa_sat_u_h((v8u16)in1, sat_val); \ | |
| } | |
| #define SAT_UH2_UH(...) SAT_UH2(v8u16, __VA_ARGS__) | |
| #define SAT_UH4(RTYPE, in0, in1, in2, in3, sat_val) \ | |
| { \ | |
| SAT_UH2(RTYPE, in0, in1, sat_val); \ | |
| SAT_UH2(RTYPE, in2, in3, sat_val) \ | |
| } | |
| #define SAT_UH4_UH(...) SAT_UH4(v8u16, __VA_ARGS__) | |
| /* Description : Saturate the halfword element values to the max | |
| unsigned value of (sat_val + 1) bits | |
| The element data width remains unchanged | |
| Arguments : Inputs - in0, in1, sat_val | |
| Outputs - in place operation | |
| Return Type - as per RTYPE | |
| Details : Each unsigned halfword element from 'in0' is saturated to the | |
| value generated with (sat_val + 1) bit range | |
| The results are written in place | |
| */ | |
| #define SAT_SH2(RTYPE, in0, in1, sat_val) \ | |
| { \ | |
| in0 = (RTYPE)__msa_sat_s_h((v8i16)in0, sat_val); \ | |
| in1 = (RTYPE)__msa_sat_s_h((v8i16)in1, sat_val); \ | |
| } | |
| #define SAT_SH2_SH(...) SAT_SH2(v8i16, __VA_ARGS__) | |
| #define SAT_SH4(RTYPE, in0, in1, in2, in3, sat_val) \ | |
| { \ | |
| SAT_SH2(RTYPE, in0, in1, sat_val); \ | |
| SAT_SH2(RTYPE, in2, in3, sat_val); \ | |
| } | |
| #define SAT_SH4_SH(...) SAT_SH4(v8i16, __VA_ARGS__) | |
| /* Description : Indexed halfword element values are replicated to all | |
| elements in output vector | |
| Arguments : Inputs - in, idx0, idx1 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : 'idx0' element value from 'in' vector is replicated to all | |
| elements in 'out0' vector | |
| Valid index range for halfword operation is 0-7 | |
| */ | |
| #define SPLATI_H2(RTYPE, in, idx0, idx1, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_splati_h((v8i16)in, idx0); \ | |
| out1 = (RTYPE)__msa_splati_h((v8i16)in, idx1); \ | |
| } | |
| #define SPLATI_H2_SH(...) SPLATI_H2(v8i16, __VA_ARGS__) | |
| #define SPLATI_H4(RTYPE, in, idx0, idx1, idx2, idx3, out0, out1, out2, out3) \ | |
| { \ | |
| SPLATI_H2(RTYPE, in, idx0, idx1, out0, out1); \ | |
| SPLATI_H2(RTYPE, in, idx2, idx3, out2, out3); \ | |
| } | |
| #define SPLATI_H4_SB(...) SPLATI_H4(v16i8, __VA_ARGS__) | |
| #define SPLATI_H4_SH(...) SPLATI_H4(v8i16, __VA_ARGS__) | |
| /* Description : Pack even byte elements of vector pairs | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Even byte elements of 'in0' are copied to the left half of | |
| 'out0' & even byte elements of 'in1' are copied to the right | |
| half of 'out0'. | |
| */ | |
| #define PCKEV_B2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_pckev_b((v16i8)in0, (v16i8)in1); \ | |
| out1 = (RTYPE)__msa_pckev_b((v16i8)in2, (v16i8)in3); \ | |
| } | |
| #define PCKEV_B2_SB(...) PCKEV_B2(v16i8, __VA_ARGS__) | |
| #define PCKEV_B2_UB(...) PCKEV_B2(v16u8, __VA_ARGS__) | |
| #define PCKEV_B2_SH(...) PCKEV_B2(v8i16, __VA_ARGS__) | |
| #define PCKEV_B4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| PCKEV_B2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| PCKEV_B2(RTYPE, in4, in5, in6, in7, out2, out3); \ | |
| } | |
| #define PCKEV_B4_SB(...) PCKEV_B4(v16i8, __VA_ARGS__) | |
| #define PCKEV_B4_UB(...) PCKEV_B4(v16u8, __VA_ARGS__) | |
| #define PCKEV_B4_SH(...) PCKEV_B4(v8i16, __VA_ARGS__) | |
| /* Description : Pack even halfword elements of vector pairs | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Even halfword elements of 'in0' are copied to the left half of | |
| 'out0' & even halfword elements of 'in1' are copied to the | |
| right half of 'out0'. | |
| */ | |
| #define PCKEV_H2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_pckev_h((v8i16)in0, (v8i16)in1); \ | |
| out1 = (RTYPE)__msa_pckev_h((v8i16)in2, (v8i16)in3); \ | |
| } | |
| #define PCKEV_H2_SH(...) PCKEV_H2(v8i16, __VA_ARGS__) | |
| #define PCKEV_H2_SW(...) PCKEV_H2(v4i32, __VA_ARGS__) | |
| #define PCKEV_H4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| PCKEV_H2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| PCKEV_H2(RTYPE, in4, in5, in6, in7, out2, out3); \ | |
| } | |
| #define PCKEV_H4_SH(...) PCKEV_H4(v8i16, __VA_ARGS__) | |
| /* Description : Pack even double word elements of vector pairs | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Even double elements of 'in0' are copied to the left half of | |
| 'out0' & even double elements of 'in1' are copied to the right | |
| half of 'out0'. | |
| */ | |
| #define PCKEV_D2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_pckev_d((v2i64)in0, (v2i64)in1); \ | |
| out1 = (RTYPE)__msa_pckev_d((v2i64)in2, (v2i64)in3); \ | |
| } | |
| #define PCKEV_D2_UB(...) PCKEV_D2(v16u8, __VA_ARGS__) | |
| #define PCKEV_D2_SH(...) PCKEV_D2(v8i16, __VA_ARGS__) | |
| #define PCKEV_D4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| PCKEV_D2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| PCKEV_D2(RTYPE, in4, in5, in6, in7, out2, out3); \ | |
| } | |
| #define PCKEV_D4_UB(...) PCKEV_D4(v16u8, __VA_ARGS__) | |
| /* Description : Each byte element is logically xor'ed with immediate 128 | |
| Arguments : Inputs - in0, in1 | |
| Outputs - in place operation | |
| Return Type - as per RTYPE | |
| Details : Each unsigned byte element from input vector 'in0' is | |
| logically xor'ed with 128 and the result is stored in-place. | |
| */ | |
| #define XORI_B2_128(RTYPE, in0, in1) \ | |
| { \ | |
| in0 = (RTYPE)__msa_xori_b((v16u8)in0, 128); \ | |
| in1 = (RTYPE)__msa_xori_b((v16u8)in1, 128); \ | |
| } | |
| #define XORI_B2_128_UB(...) XORI_B2_128(v16u8, __VA_ARGS__) | |
| #define XORI_B2_128_SB(...) XORI_B2_128(v16i8, __VA_ARGS__) | |
| #define XORI_B3_128(RTYPE, in0, in1, in2) \ | |
| { \ | |
| XORI_B2_128(RTYPE, in0, in1); \ | |
| in2 = (RTYPE)__msa_xori_b((v16u8)in2, 128); \ | |
| } | |
| #define XORI_B3_128_SB(...) XORI_B3_128(v16i8, __VA_ARGS__) | |
| #define XORI_B4_128(RTYPE, in0, in1, in2, in3) \ | |
| { \ | |
| XORI_B2_128(RTYPE, in0, in1); \ | |
| XORI_B2_128(RTYPE, in2, in3); \ | |
| } | |
| #define XORI_B4_128_UB(...) XORI_B4_128(v16u8, __VA_ARGS__) | |
| #define XORI_B4_128_SB(...) XORI_B4_128(v16i8, __VA_ARGS__) | |
| #define XORI_B7_128(RTYPE, in0, in1, in2, in3, in4, in5, in6) \ | |
| { \ | |
| XORI_B4_128(RTYPE, in0, in1, in2, in3); \ | |
| XORI_B3_128(RTYPE, in4, in5, in6); \ | |
| } | |
| #define XORI_B7_128_SB(...) XORI_B7_128(v16i8, __VA_ARGS__) | |
| /* Description : Average of signed halfword elements -> (a + b) / 2 | |
| Arguments : Inputs - in0, in1, in2, in3, in4, in5, in6, in7 | |
| Outputs - out0, out1, out2, out3 | |
| Return Type - as per RTYPE | |
| Details : Each signed halfword element from 'in0' is added to each | |
| signed halfword element of 'in1' with full precision resulting | |
| in one extra bit in the result. The result is then divided by | |
| 2 and written to 'out0' | |
| */ | |
| #define AVE_SH4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| out0 = (RTYPE)__msa_ave_s_h((v8i16)in0, (v8i16)in1); \ | |
| out1 = (RTYPE)__msa_ave_s_h((v8i16)in2, (v8i16)in3); \ | |
| out2 = (RTYPE)__msa_ave_s_h((v8i16)in4, (v8i16)in5); \ | |
| out3 = (RTYPE)__msa_ave_s_h((v8i16)in6, (v8i16)in7); \ | |
| } | |
| #define AVE_SH4_SH(...) AVE_SH4(v8i16, __VA_ARGS__) | |
| /* Description : Addition of signed halfword elements and signed saturation | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Return Type - as per RTYPE | |
| Details : Signed halfword elements from 'in0' are added to signed | |
| halfword elements of 'in1'. The result is then signed saturated | |
| between halfword data type range | |
| */ | |
| #define ADDS_SH2(RTYPE, in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = (RTYPE)__msa_adds_s_h((v8i16)in0, (v8i16)in1); \ | |
| out1 = (RTYPE)__msa_adds_s_h((v8i16)in2, (v8i16)in3); \ | |
| } | |
| #define ADDS_SH2_SH(...) ADDS_SH2(v8i16, __VA_ARGS__) | |
| #define ADDS_SH4(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3) \ | |
| { \ | |
| ADDS_SH2(RTYPE, in0, in1, in2, in3, out0, out1); \ | |
| ADDS_SH2(RTYPE, in4, in5, in6, in7, out2, out3); \ | |
| } | |
| #define ADDS_SH4_SH(...) ADDS_SH4(v8i16, __VA_ARGS__) | |
| /* Description : Shift left all elements of vector (generic for all data types) | |
| Arguments : Inputs - in0, in1, in2, in3, shift | |
| Outputs - in place operation | |
| Return Type - as per input vector RTYPE | |
| Details : Each element of vector 'in0' is left shifted by 'shift' and | |
| the result is written in-place. | |
| */ | |
| #define SLLI_4V(in0, in1, in2, in3, shift) \ | |
| { \ | |
| in0 = in0 << shift; \ | |
| in1 = in1 << shift; \ | |
| in2 = in2 << shift; \ | |
| in3 = in3 << shift; \ | |
| } | |
| /* Description : Arithmetic shift right all elements of vector | |
| (generic for all data types) | |
| Arguments : Inputs - in0, in1, in2, in3, shift | |
| Outputs - in place operation | |
| Return Type - as per input vector RTYPE | |
| Details : Each element of vector 'in0' is right shifted by 'shift' and | |
| the result is written in-place. 'shift' is a GP variable. | |
| */ | |
| #define SRA_4V(in0, in1, in2, in3, shift) \ | |
| { \ | |
| in0 = in0 >> shift; \ | |
| in1 = in1 >> shift; \ | |
| in2 = in2 >> shift; \ | |
| in3 = in3 >> shift; \ | |
| } | |
| /* Description : Shift right arithmetic rounded words | |
| Arguments : Inputs - in0, in1, shift | |
| Outputs - in place operation | |
| Return Type - as per RTYPE | |
| Details : Each element of vector 'in0' is shifted right arithmetically by | |
| the number of bits in the corresponding element in the vector | |
| 'shift'. The last discarded bit is added to shifted value for | |
| rounding and the result is written in-place. | |
| 'shift' is a vector. | |
| */ | |
| #define SRAR_W2(RTYPE, in0, in1, shift) \ | |
| { \ | |
| in0 = (RTYPE)__msa_srar_w((v4i32)in0, (v4i32)shift); \ | |
| in1 = (RTYPE)__msa_srar_w((v4i32)in1, (v4i32)shift); \ | |
| } | |
| #define SRAR_W4(RTYPE, in0, in1, in2, in3, shift) \ | |
| { \ | |
| SRAR_W2(RTYPE, in0, in1, shift) \ | |
| SRAR_W2(RTYPE, in2, in3, shift) \ | |
| } | |
| #define SRAR_W4_SW(...) SRAR_W4(v4i32, __VA_ARGS__) | |
| /* Description : Shift right arithmetic rounded (immediate) | |
| Arguments : Inputs - in0, in1, shift | |
| Outputs - in place operation | |
| Return Type - as per RTYPE | |
| Details : Each element of vector 'in0' is shifted right arithmetically by | |
| the value in 'shift'. The last discarded bit is added to the | |
| shifted value for rounding and the result is written in-place. | |
| 'shift' is an immediate value. | |
| */ | |
| #define SRARI_H2(RTYPE, in0, in1, shift) \ | |
| { \ | |
| in0 = (RTYPE)__msa_srari_h((v8i16)in0, shift); \ | |
| in1 = (RTYPE)__msa_srari_h((v8i16)in1, shift); \ | |
| } | |
| #define SRARI_H2_UH(...) SRARI_H2(v8u16, __VA_ARGS__) | |
| #define SRARI_H2_SH(...) SRARI_H2(v8i16, __VA_ARGS__) | |
| #define SRARI_H4(RTYPE, in0, in1, in2, in3, shift) \ | |
| { \ | |
| SRARI_H2(RTYPE, in0, in1, shift); \ | |
| SRARI_H2(RTYPE, in2, in3, shift); \ | |
| } | |
| #define SRARI_H4_UH(...) SRARI_H4(v8u16, __VA_ARGS__) | |
| #define SRARI_H4_SH(...) SRARI_H4(v8i16, __VA_ARGS__) | |
| #define SRARI_W2(RTYPE, in0, in1, shift) \ | |
| { \ | |
| in0 = (RTYPE)__msa_srari_w((v4i32)in0, shift); \ | |
| in1 = (RTYPE)__msa_srari_w((v4i32)in1, shift); \ | |
| } | |
| #define SRARI_W2_SW(...) SRARI_W2(v4i32, __VA_ARGS__) | |
| #define SRARI_W4(RTYPE, in0, in1, in2, in3, shift) \ | |
| { \ | |
| SRARI_W2(RTYPE, in0, in1, shift); \ | |
| SRARI_W2(RTYPE, in2, in3, shift); \ | |
| } | |
| #define SRARI_W4_SW(...) SRARI_W4(v4i32, __VA_ARGS__) | |
| /* Description : Logical shift right all elements of vector (immediate) | |
| Arguments : Inputs - in0, in1, in2, in3, shift | |
| Outputs - out0, out1, out2, out3 | |
| Return Type - as per RTYPE | |
| Details : Each element of vector 'in0' is right shifted by 'shift' and | |
| the result is written in-place. 'shift' is an immediate value. | |
| */ | |
| #define SRLI_H4(RTYPE, in0, in1, in2, in3, out0, out1, out2, out3, shift) \ | |
| { \ | |
| out0 = (RTYPE)__msa_srli_h((v8i16)in0, shift); \ | |
| out1 = (RTYPE)__msa_srli_h((v8i16)in1, shift); \ | |
| out2 = (RTYPE)__msa_srli_h((v8i16)in2, shift); \ | |
| out3 = (RTYPE)__msa_srli_h((v8i16)in3, shift); \ | |
| } | |
| #define SRLI_H4_SH(...) SRLI_H4(v8i16, __VA_ARGS__) | |
| /* Description : Multiplication of pairs of vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Details : Each element from 'in0' is multiplied with elements from 'in1' | |
| and the result is written to 'out0' | |
| */ | |
| #define MUL2(in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = in0 * in1; \ | |
| out1 = in2 * in3; \ | |
| } | |
| #define MUL4(in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, out2, out3) \ | |
| { \ | |
| MUL2(in0, in1, in2, in3, out0, out1); \ | |
| MUL2(in4, in5, in6, in7, out2, out3); \ | |
| } | |
| /* Description : Addition of 2 pairs of vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Details : Each element in 'in0' is added to 'in1' and result is written | |
| to 'out0'. | |
| */ | |
| #define ADD2(in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = in0 + in1; \ | |
| out1 = in2 + in3; \ | |
| } | |
| #define ADD4(in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, out2, out3) \ | |
| { \ | |
| ADD2(in0, in1, in2, in3, out0, out1); \ | |
| ADD2(in4, in5, in6, in7, out2, out3); \ | |
| } | |
| /* Description : Subtraction of 2 pairs of vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1 | |
| Details : Each element in 'in1' is subtracted from 'in0' and result is | |
| written to 'out0'. | |
| */ | |
| #define SUB2(in0, in1, in2, in3, out0, out1) \ | |
| { \ | |
| out0 = in0 - in1; \ | |
| out1 = in2 - in3; \ | |
| } | |
| #define SUB4(in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, out2, out3) \ | |
| { \ | |
| out0 = in0 - in1; \ | |
| out1 = in2 - in3; \ | |
| out2 = in4 - in5; \ | |
| out3 = in6 - in7; \ | |
| } | |
| /* Description : Sign extend halfword elements from right half of the vector | |
| Arguments : Input - in (halfword vector) | |
| Output - out (sign extended word vector) | |
| Return Type - signed word | |
| Details : Sign bit of halfword elements from input vector 'in' is | |
| extracted and interleaved with same vector 'in0' to generate | |
| 4 word elements keeping sign intact | |
| */ | |
| #define UNPCK_R_SH_SW(in, out) \ | |
| { \ | |
| v8i16 sign_m; \ | |
| \ | |
| sign_m = __msa_clti_s_h((v8i16)in, 0); \ | |
| out = (v4i32)__msa_ilvr_h(sign_m, (v8i16)in); \ | |
| } | |
| /* Description : Zero extend unsigned byte elements to halfword elements | |
| Arguments : Input - in (unsigned byte vector) | |
| Outputs - out0, out1 (unsigned halfword vectors) | |
| Return Type - signed halfword | |
| Details : Zero extended right half of vector is returned in 'out0' | |
| Zero extended left half of vector is returned in 'out1' | |
| */ | |
| #define UNPCK_UB_SH(in, out0, out1) \ | |
| { \ | |
| v16i8 zero_m = { 0 }; \ | |
| \ | |
| ILVRL_B2_SH(zero_m, in, out0, out1); \ | |
| } | |
| /* Description : Sign extend halfword elements from input vector and return | |
| the result in pair of vectors | |
| Arguments : Input - in (halfword vector) | |
| Outputs - out0, out1 (sign extended word vectors) | |
| Return Type - signed word | |
| Details : Sign bit of halfword elements from input vector 'in' is | |
| extracted and interleaved right with same vector 'in0' to | |
| generate 4 signed word elements in 'out0' | |
| Then interleaved left with same vector 'in0' to | |
| generate 4 signed word elements in 'out1' | |
| */ | |
| #define UNPCK_SH_SW(in, out0, out1) \ | |
| { \ | |
| v8i16 tmp_m; \ | |
| \ | |
| tmp_m = __msa_clti_s_h((v8i16)in, 0); \ | |
| ILVRL_H2_SW(tmp_m, in, out0, out1); \ | |
| } | |
| /* Description : Butterfly of 4 input vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1, out2, out3 | |
| Details : Butterfly operation | |
| */ | |
| #define BUTTERFLY_4(in0, in1, in2, in3, out0, out1, out2, out3) \ | |
| { \ | |
| out0 = in0 + in3; \ | |
| out1 = in1 + in2; \ | |
| \ | |
| out2 = in1 - in2; \ | |
| out3 = in0 - in3; \ | |
| } | |
| /* Description : Butterfly of 8 input vectors | |
| Arguments : Inputs - in0 ... in7 | |
| Outputs - out0 .. out7 | |
| Details : Butterfly operation | |
| */ | |
| #define BUTTERFLY_8(in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, out2, \ | |
| out3, out4, out5, out6, out7) \ | |
| { \ | |
| out0 = in0 + in7; \ | |
| out1 = in1 + in6; \ | |
| out2 = in2 + in5; \ | |
| out3 = in3 + in4; \ | |
| \ | |
| out4 = in3 - in4; \ | |
| out5 = in2 - in5; \ | |
| out6 = in1 - in6; \ | |
| out7 = in0 - in7; \ | |
| } | |
| /* Description : Butterfly of 16 input vectors | |
| Arguments : Inputs - in0 ... in15 | |
| Outputs - out0 .. out15 | |
| Details : Butterfly operation | |
| */ | |
| #define BUTTERFLY_16(in0, in1, in2, in3, in4, in5, in6, in7, in8, in9, in10, \ | |
| in11, in12, in13, in14, in15, out0, out1, out2, out3, \ | |
| out4, out5, out6, out7, out8, out9, out10, out11, out12, \ | |
| out13, out14, out15) \ | |
| { \ | |
| out0 = in0 + in15; \ | |
| out1 = in1 + in14; \ | |
| out2 = in2 + in13; \ | |
| out3 = in3 + in12; \ | |
| out4 = in4 + in11; \ | |
| out5 = in5 + in10; \ | |
| out6 = in6 + in9; \ | |
| out7 = in7 + in8; \ | |
| \ | |
| out8 = in7 - in8; \ | |
| out9 = in6 - in9; \ | |
| out10 = in5 - in10; \ | |
| out11 = in4 - in11; \ | |
| out12 = in3 - in12; \ | |
| out13 = in2 - in13; \ | |
| out14 = in1 - in14; \ | |
| out15 = in0 - in15; \ | |
| } | |
| /* Description : Transpose input 8x8 byte block | |
| Arguments : Inputs - in0, in1, in2, in3, in4, in5, in6, in7 | |
| Outputs - out0, out1, out2, out3, out4, out5, out6, out7 | |
| Return Type - as per RTYPE | |
| */ | |
| #define TRANSPOSE8x8_UB(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, \ | |
| out1, out2, out3, out4, out5, out6, out7) \ | |
| { \ | |
| v16i8 tmp0_m, tmp1_m, tmp2_m, tmp3_m; \ | |
| v16i8 tmp4_m, tmp5_m, tmp6_m, tmp7_m; \ | |
| \ | |
| ILVR_B4_SB(in2, in0, in3, in1, in6, in4, in7, in5, tmp0_m, tmp1_m, tmp2_m, \ | |
| tmp3_m); \ | |
| ILVRL_B2_SB(tmp1_m, tmp0_m, tmp4_m, tmp5_m); \ | |
| ILVRL_B2_SB(tmp3_m, tmp2_m, tmp6_m, tmp7_m); \ | |
| ILVRL_W2(RTYPE, tmp6_m, tmp4_m, out0, out2); \ | |
| ILVRL_W2(RTYPE, tmp7_m, tmp5_m, out4, out6); \ | |
| SLDI_B2_0(RTYPE, out0, out2, out1, out3, 8); \ | |
| SLDI_B2_0(RTYPE, out4, out6, out5, out7, 8); \ | |
| } | |
| #define TRANSPOSE8x8_UB_UB(...) TRANSPOSE8x8_UB(v16u8, __VA_ARGS__) | |
| /* Description : Transpose 16x8 block into 8x16 with byte elements in vectors | |
| Arguments : Inputs - in0, in1, in2, in3, in4, in5, in6, in7, | |
| in8, in9, in10, in11, in12, in13, in14, in15 | |
| Outputs - out0, out1, out2, out3, out4, out5, out6, out7 | |
| Return Type - unsigned byte | |
| */ | |
| #define TRANSPOSE16x8_UB_UB(in0, in1, in2, in3, in4, in5, in6, in7, in8, in9, \ | |
| in10, in11, in12, in13, in14, in15, out0, out1, \ | |
| out2, out3, out4, out5, out6, out7) \ | |
| { \ | |
| v16u8 tmp0_m, tmp1_m, tmp2_m, tmp3_m; \ | |
| v16u8 tmp4_m, tmp5_m, tmp6_m, tmp7_m; \ | |
| \ | |
| ILVEV_D2_UB(in0, in8, in1, in9, out7, out6); \ | |
| ILVEV_D2_UB(in2, in10, in3, in11, out5, out4); \ | |
| ILVEV_D2_UB(in4, in12, in5, in13, out3, out2); \ | |
| ILVEV_D2_UB(in6, in14, in7, in15, out1, out0); \ | |
| \ | |
| tmp0_m = (v16u8)__msa_ilvev_b((v16i8)out6, (v16i8)out7); \ | |
| tmp4_m = (v16u8)__msa_ilvod_b((v16i8)out6, (v16i8)out7); \ | |
| tmp1_m = (v16u8)__msa_ilvev_b((v16i8)out4, (v16i8)out5); \ | |
| tmp5_m = (v16u8)__msa_ilvod_b((v16i8)out4, (v16i8)out5); \ | |
| out5 = (v16u8)__msa_ilvev_b((v16i8)out2, (v16i8)out3); \ | |
| tmp6_m = (v16u8)__msa_ilvod_b((v16i8)out2, (v16i8)out3); \ | |
| out7 = (v16u8)__msa_ilvev_b((v16i8)out0, (v16i8)out1); \ | |
| tmp7_m = (v16u8)__msa_ilvod_b((v16i8)out0, (v16i8)out1); \ | |
| \ | |
| ILVEV_H2_UB(tmp0_m, tmp1_m, out5, out7, tmp2_m, tmp3_m); \ | |
| out0 = (v16u8)__msa_ilvev_w((v4i32)tmp3_m, (v4i32)tmp2_m); \ | |
| out4 = (v16u8)__msa_ilvod_w((v4i32)tmp3_m, (v4i32)tmp2_m); \ | |
| \ | |
| tmp2_m = (v16u8)__msa_ilvod_h((v8i16)tmp1_m, (v8i16)tmp0_m); \ | |
| tmp3_m = (v16u8)__msa_ilvod_h((v8i16)out7, (v8i16)out5); \ | |
| out2 = (v16u8)__msa_ilvev_w((v4i32)tmp3_m, (v4i32)tmp2_m); \ | |
| out6 = (v16u8)__msa_ilvod_w((v4i32)tmp3_m, (v4i32)tmp2_m); \ | |
| \ | |
| ILVEV_H2_UB(tmp4_m, tmp5_m, tmp6_m, tmp7_m, tmp2_m, tmp3_m); \ | |
| out1 = (v16u8)__msa_ilvev_w((v4i32)tmp3_m, (v4i32)tmp2_m); \ | |
| out5 = (v16u8)__msa_ilvod_w((v4i32)tmp3_m, (v4i32)tmp2_m); \ | |
| \ | |
| tmp2_m = (v16u8)__msa_ilvod_h((v8i16)tmp5_m, (v8i16)tmp4_m); \ | |
| tmp2_m = (v16u8)__msa_ilvod_h((v8i16)tmp5_m, (v8i16)tmp4_m); \ | |
| tmp3_m = (v16u8)__msa_ilvod_h((v8i16)tmp7_m, (v8i16)tmp6_m); \ | |
| tmp3_m = (v16u8)__msa_ilvod_h((v8i16)tmp7_m, (v8i16)tmp6_m); \ | |
| out3 = (v16u8)__msa_ilvev_w((v4i32)tmp3_m, (v4i32)tmp2_m); \ | |
| out7 = (v16u8)__msa_ilvod_w((v4i32)tmp3_m, (v4i32)tmp2_m); \ | |
| } | |
| /* Description : Transpose 4x4 block with half word elements in vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1, out2, out3 | |
| Return Type - signed halfword | |
| */ | |
| #define TRANSPOSE4x4_SH_SH(in0, in1, in2, in3, out0, out1, out2, out3) \ | |
| { \ | |
| v8i16 s0_m, s1_m; \ | |
| \ | |
| ILVR_H2_SH(in1, in0, in3, in2, s0_m, s1_m); \ | |
| ILVRL_W2_SH(s1_m, s0_m, out0, out2); \ | |
| out1 = (v8i16)__msa_ilvl_d((v2i64)out0, (v2i64)out0); \ | |
| out3 = (v8i16)__msa_ilvl_d((v2i64)out0, (v2i64)out2); \ | |
| } | |
| /* Description : Transpose 4x8 block with half word elements in vectors | |
| Arguments : Inputs - in0, in1, in2, in3, in4, in5, in6, in7 | |
| Outputs - out0, out1, out2, out3, out4, out5, out6, out7 | |
| Return Type - signed halfword | |
| */ | |
| #define TRANSPOSE4X8_SH_SH(in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, \ | |
| out2, out3, out4, out5, out6, out7) \ | |
| { \ | |
| v8i16 tmp0_m, tmp1_m, tmp2_m, tmp3_m; \ | |
| v8i16 tmp0_n, tmp1_n, tmp2_n, tmp3_n; \ | |
| v8i16 zero_m = { 0 }; \ | |
| \ | |
| ILVR_H4_SH(in1, in0, in3, in2, in5, in4, in7, in6, tmp0_n, tmp1_n, tmp2_n, \ | |
| tmp3_n); \ | |
| ILVRL_W2_SH(tmp1_n, tmp0_n, tmp0_m, tmp2_m); \ | |
| ILVRL_W2_SH(tmp3_n, tmp2_n, tmp1_m, tmp3_m); \ | |
| \ | |
| out0 = (v8i16)__msa_ilvr_d((v2i64)tmp1_m, (v2i64)tmp0_m); \ | |
| out1 = (v8i16)__msa_ilvl_d((v2i64)tmp1_m, (v2i64)tmp0_m); \ | |
| out2 = (v8i16)__msa_ilvr_d((v2i64)tmp3_m, (v2i64)tmp2_m); \ | |
| out3 = (v8i16)__msa_ilvl_d((v2i64)tmp3_m, (v2i64)tmp2_m); \ | |
| \ | |
| out4 = zero_m; \ | |
| out5 = zero_m; \ | |
| out6 = zero_m; \ | |
| out7 = zero_m; \ | |
| } | |
| /* Description : Transpose 8x4 block with half word elements in vectors | |
| Arguments : Inputs - in0, in1, in2, in3, in4, in5, in6, in7 | |
| Outputs - out0, out1, out2, out3, out4, out5, out6, out7 | |
| Return Type - signed halfword | |
| */ | |
| #define TRANSPOSE8X4_SH_SH(in0, in1, in2, in3, out0, out1, out2, out3) \ | |
| { \ | |
| v8i16 tmp0_m, tmp1_m, tmp2_m, tmp3_m; \ | |
| \ | |
| ILVR_H2_SH(in1, in0, in3, in2, tmp0_m, tmp1_m); \ | |
| ILVL_H2_SH(in1, in0, in3, in2, tmp2_m, tmp3_m); \ | |
| ILVR_W2_SH(tmp1_m, tmp0_m, tmp3_m, tmp2_m, out0, out2); \ | |
| ILVL_W2_SH(tmp1_m, tmp0_m, tmp3_m, tmp2_m, out1, out3); \ | |
| } | |
| /* Description : Transpose 8x8 block with half word elements in vectors | |
| Arguments : Inputs - in0, in1, in2, in3, in4, in5, in6, in7 | |
| Outputs - out0, out1, out2, out3, out4, out5, out6, out7 | |
| Return Type - as per RTYPE | |
| */ | |
| #define TRANSPOSE8x8_H(RTYPE, in0, in1, in2, in3, in4, in5, in6, in7, out0, \ | |
| out1, out2, out3, out4, out5, out6, out7) \ | |
| { \ | |
| v8i16 s0_m, s1_m; \ | |
| v8i16 tmp0_m, tmp1_m, tmp2_m, tmp3_m; \ | |
| v8i16 tmp4_m, tmp5_m, tmp6_m, tmp7_m; \ | |
| \ | |
| ILVR_H2_SH(in6, in4, in7, in5, s0_m, s1_m); \ | |
| ILVRL_H2_SH(s1_m, s0_m, tmp0_m, tmp1_m); \ | |
| ILVL_H2_SH(in6, in4, in7, in5, s0_m, s1_m); \ | |
| ILVRL_H2_SH(s1_m, s0_m, tmp2_m, tmp3_m); \ | |
| ILVR_H2_SH(in2, in0, in3, in1, s0_m, s1_m); \ | |
| ILVRL_H2_SH(s1_m, s0_m, tmp4_m, tmp5_m); \ | |
| ILVL_H2_SH(in2, in0, in3, in1, s0_m, s1_m); \ | |
| ILVRL_H2_SH(s1_m, s0_m, tmp6_m, tmp7_m); \ | |
| PCKEV_D4(RTYPE, tmp0_m, tmp4_m, tmp1_m, tmp5_m, tmp2_m, tmp6_m, tmp3_m, \ | |
| tmp7_m, out0, out2, out4, out6); \ | |
| out1 = (RTYPE)__msa_pckod_d((v2i64)tmp0_m, (v2i64)tmp4_m); \ | |
| out3 = (RTYPE)__msa_pckod_d((v2i64)tmp1_m, (v2i64)tmp5_m); \ | |
| out5 = (RTYPE)__msa_pckod_d((v2i64)tmp2_m, (v2i64)tmp6_m); \ | |
| out7 = (RTYPE)__msa_pckod_d((v2i64)tmp3_m, (v2i64)tmp7_m); \ | |
| } | |
| #define TRANSPOSE8x8_SH_SH(...) TRANSPOSE8x8_H(v8i16, __VA_ARGS__) | |
| /* Description : Transpose 4x4 block with word elements in vectors | |
| Arguments : Inputs - in0, in1, in2, in3 | |
| Outputs - out0, out1, out2, out3 | |
| Return Type - signed word | |
| */ | |
| #define TRANSPOSE4x4_SW_SW(in0, in1, in2, in3, out0, out1, out2, out3) \ | |
| { \ | |
| v4i32 s0_m, s1_m, s2_m, s3_m; \ | |
| \ | |
| ILVRL_W2_SW(in1, in0, s0_m, s1_m); \ | |
| ILVRL_W2_SW(in3, in2, s2_m, s3_m); \ | |
| \ | |
| out0 = (v4i32)__msa_ilvr_d((v2i64)s2_m, (v2i64)s0_m); \ | |
| out1 = (v4i32)__msa_ilvl_d((v2i64)s2_m, (v2i64)s0_m); \ | |
| out2 = (v4i32)__msa_ilvr_d((v2i64)s3_m, (v2i64)s1_m); \ | |
| out3 = (v4i32)__msa_ilvl_d((v2i64)s3_m, (v2i64)s1_m); \ | |
| } | |
| /* Description : Add block 4x4 | |
| Arguments : Inputs - in0, in1, in2, in3, pdst, stride | |
| Details : Least significant 4 bytes from each input vector are added to | |
| the destination bytes, clipped between 0-255 and stored. | |
| */ | |
| #define ADDBLK_ST4x4_UB(in0, in1, in2, in3, pdst, stride) \ | |
| { \ | |
| uint32_t src0_m, src1_m, src2_m, src3_m; \ | |
| v8i16 inp0_m, inp1_m, res0_m, res1_m; \ | |
| v16i8 dst0_m = { 0 }; \ | |
| v16i8 dst1_m = { 0 }; \ | |
| v16i8 zero_m = { 0 }; \ | |
| \ | |
| ILVR_D2_SH(in1, in0, in3, in2, inp0_m, inp1_m) \ | |
| LW4(pdst, stride, src0_m, src1_m, src2_m, src3_m); \ | |
| INSERT_W2_SB(src0_m, src1_m, dst0_m); \ | |
| INSERT_W2_SB(src2_m, src3_m, dst1_m); \ | |
| ILVR_B2_SH(zero_m, dst0_m, zero_m, dst1_m, res0_m, res1_m); \ | |
| ADD2(res0_m, inp0_m, res1_m, inp1_m, res0_m, res1_m); \ | |
| CLIP_SH2_0_255(res0_m, res1_m); \ | |
| PCKEV_B2_SB(res0_m, res0_m, res1_m, res1_m, dst0_m, dst1_m); \ | |
| ST4x4_UB(dst0_m, dst1_m, 0, 1, 0, 1, pdst, stride); \ | |
| } | |
| /* Description : Pack even elements of input vectors & xor with 128 | |
| Arguments : Inputs - in0, in1 | |
| Output - out_m | |
| Return Type - unsigned byte | |
| Details : Signed byte even elements from 'in0' and 'in1' are packed | |
| together in one vector and the resulting vector is xor'ed with | |
| 128 to shift the range from signed to unsigned byte | |
| */ | |
| #define PCKEV_XORI128_UB(in0, in1) \ | |
| ({ \ | |
| v16u8 out_m; \ | |
| \ | |
| out_m = (v16u8)__msa_pckev_b((v16i8)in1, (v16i8)in0); \ | |
| out_m = (v16u8)__msa_xori_b((v16u8)out_m, 128); \ | |
| out_m; \ | |
| }) | |
| /* Description : Converts inputs to unsigned bytes, interleave, average & store | |
| as 8x4 unsigned byte block | |
| Arguments : Inputs - in0, in1, in2, in3, dst0, dst1, dst2, dst3, | |
| pdst, stride | |
| */ | |
| #define CONVERT_UB_AVG_ST8x4_UB(in0, in1, in2, in3, dst0, dst1, dst2, dst3, \ | |
| pdst, stride) \ | |
| { \ | |
| v16u8 tmp0_m, tmp1_m, tmp2_m, tmp3_m; \ | |
| \ | |
| tmp0_m = PCKEV_XORI128_UB(in0, in1); \ | |
| tmp1_m = PCKEV_XORI128_UB(in2, in3); \ | |
| ILVR_D2_UB(dst1, dst0, dst3, dst2, tmp2_m, tmp3_m); \ | |
| AVER_UB2_UB(tmp0_m, tmp2_m, tmp1_m, tmp3_m, tmp0_m, tmp1_m); \ | |
| ST8x4_UB(tmp0_m, tmp1_m, pdst, stride); \ | |
| } | |
| /* Description : Pack even byte elements and store byte vector in destination | |
| memory | |
| Arguments : Inputs - in0, in1, pdst | |
| */ | |
| #define PCKEV_ST_SB(in0, in1, pdst) \ | |
| { \ | |
| v16i8 tmp_m; \ | |
| \ | |
| tmp_m = __msa_pckev_b((v16i8)in1, (v16i8)in0); \ | |
| ST_SB(tmp_m, (pdst)); \ | |
| } | |
| /* Description : Horizontal 2 tap filter kernel code | |
| Arguments : Inputs - in0, in1, mask, coeff, shift | |
| */ | |
| #define HORIZ_2TAP_FILT_UH(in0, in1, mask, coeff, shift) \ | |
| ({ \ | |
| v16i8 tmp0_m; \ | |
| v8u16 tmp1_m; \ | |
| \ | |
| tmp0_m = __msa_vshf_b((v16i8)mask, (v16i8)in1, (v16i8)in0); \ | |
| tmp1_m = __msa_dotp_u_h((v16u8)tmp0_m, (v16u8)coeff); \ | |
| tmp1_m = (v8u16)__msa_srari_h((v8i16)tmp1_m, shift); \ | |
| \ | |
| tmp1_m; \ | |
| }) | |
| #endif /* AOM_DSP_MIPS_MACROS_MSA_H_ */ |