mirror of
https://gitlab.com/qemu-project/qemu
synced 2024-11-05 20:35:44 +00:00
tcg: Add support for vector absolute value
Reviewed-by: Alex Bennée <alex.bennee@linaro.org> Signed-off-by: Richard Henderson <richard.henderson@linaro.org>
This commit is contained in:
parent
ff1f11f7f8
commit
bcefc90208
11 changed files with 167 additions and 0 deletions
|
@ -398,6 +398,54 @@ void HELPER(gvec_neg64)(void *d, void *a, uint32_t desc)
|
|||
clear_high(d, oprsz, desc);
|
||||
}
|
||||
|
||||
void HELPER(gvec_abs8)(void *d, void *a, uint32_t desc)
|
||||
{
|
||||
intptr_t oprsz = simd_oprsz(desc);
|
||||
intptr_t i;
|
||||
|
||||
for (i = 0; i < oprsz; i += sizeof(int8_t)) {
|
||||
int8_t aa = *(int8_t *)(a + i);
|
||||
*(int8_t *)(d + i) = aa < 0 ? -aa : aa;
|
||||
}
|
||||
clear_high(d, oprsz, desc);
|
||||
}
|
||||
|
||||
void HELPER(gvec_abs16)(void *d, void *a, uint32_t desc)
|
||||
{
|
||||
intptr_t oprsz = simd_oprsz(desc);
|
||||
intptr_t i;
|
||||
|
||||
for (i = 0; i < oprsz; i += sizeof(int16_t)) {
|
||||
int16_t aa = *(int16_t *)(a + i);
|
||||
*(int16_t *)(d + i) = aa < 0 ? -aa : aa;
|
||||
}
|
||||
clear_high(d, oprsz, desc);
|
||||
}
|
||||
|
||||
void HELPER(gvec_abs32)(void *d, void *a, uint32_t desc)
|
||||
{
|
||||
intptr_t oprsz = simd_oprsz(desc);
|
||||
intptr_t i;
|
||||
|
||||
for (i = 0; i < oprsz; i += sizeof(int32_t)) {
|
||||
int32_t aa = *(int32_t *)(a + i);
|
||||
*(int32_t *)(d + i) = aa < 0 ? -aa : aa;
|
||||
}
|
||||
clear_high(d, oprsz, desc);
|
||||
}
|
||||
|
||||
void HELPER(gvec_abs64)(void *d, void *a, uint32_t desc)
|
||||
{
|
||||
intptr_t oprsz = simd_oprsz(desc);
|
||||
intptr_t i;
|
||||
|
||||
for (i = 0; i < oprsz; i += sizeof(int64_t)) {
|
||||
int64_t aa = *(int64_t *)(a + i);
|
||||
*(int64_t *)(d + i) = aa < 0 ? -aa : aa;
|
||||
}
|
||||
clear_high(d, oprsz, desc);
|
||||
}
|
||||
|
||||
void HELPER(gvec_mov)(void *d, void *a, uint32_t desc)
|
||||
{
|
||||
intptr_t oprsz = simd_oprsz(desc);
|
||||
|
|
|
@ -225,6 +225,11 @@ DEF_HELPER_FLAGS_3(gvec_neg16, TCG_CALL_NO_RWG, void, ptr, ptr, i32)
|
|||
DEF_HELPER_FLAGS_3(gvec_neg32, TCG_CALL_NO_RWG, void, ptr, ptr, i32)
|
||||
DEF_HELPER_FLAGS_3(gvec_neg64, TCG_CALL_NO_RWG, void, ptr, ptr, i32)
|
||||
|
||||
DEF_HELPER_FLAGS_3(gvec_abs8, TCG_CALL_NO_RWG, void, ptr, ptr, i32)
|
||||
DEF_HELPER_FLAGS_3(gvec_abs16, TCG_CALL_NO_RWG, void, ptr, ptr, i32)
|
||||
DEF_HELPER_FLAGS_3(gvec_abs32, TCG_CALL_NO_RWG, void, ptr, ptr, i32)
|
||||
DEF_HELPER_FLAGS_3(gvec_abs64, TCG_CALL_NO_RWG, void, ptr, ptr, i32)
|
||||
|
||||
DEF_HELPER_FLAGS_3(gvec_not, TCG_CALL_NO_RWG, void, ptr, ptr, i32)
|
||||
DEF_HELPER_FLAGS_4(gvec_and, TCG_CALL_NO_RWG, void, ptr, ptr, ptr, i32)
|
||||
DEF_HELPER_FLAGS_4(gvec_or, TCG_CALL_NO_RWG, void, ptr, ptr, ptr, i32)
|
||||
|
|
|
@ -561,6 +561,10 @@ E.g. VECL=1 -> 64 << 1 -> v128, and VECE=2 -> 1 << 2 -> i32.
|
|||
|
||||
Similarly, v0 = -v1.
|
||||
|
||||
* abs_vec v0, v1
|
||||
|
||||
Similarly, v0 = v1 < 0 ? -v1 : v1, in elements across the vector.
|
||||
|
||||
* smin_vec:
|
||||
* umin_vec:
|
||||
|
||||
|
|
|
@ -132,6 +132,7 @@ typedef enum {
|
|||
#define TCG_TARGET_HAS_orc_vec 1
|
||||
#define TCG_TARGET_HAS_not_vec 1
|
||||
#define TCG_TARGET_HAS_neg_vec 1
|
||||
#define TCG_TARGET_HAS_abs_vec 0
|
||||
#define TCG_TARGET_HAS_shi_vec 1
|
||||
#define TCG_TARGET_HAS_shs_vec 0
|
||||
#define TCG_TARGET_HAS_shv_vec 1
|
||||
|
|
|
@ -182,6 +182,7 @@ extern bool have_avx2;
|
|||
#define TCG_TARGET_HAS_orc_vec 0
|
||||
#define TCG_TARGET_HAS_not_vec 0
|
||||
#define TCG_TARGET_HAS_neg_vec 0
|
||||
#define TCG_TARGET_HAS_abs_vec 0
|
||||
#define TCG_TARGET_HAS_shi_vec 1
|
||||
#define TCG_TARGET_HAS_shs_vec 1
|
||||
#define TCG_TARGET_HAS_shv_vec have_avx2
|
||||
|
|
|
@ -2177,6 +2177,69 @@ void tcg_gen_gvec_neg(unsigned vece, uint32_t dofs, uint32_t aofs,
|
|||
tcg_gen_gvec_2(dofs, aofs, oprsz, maxsz, &g[vece]);
|
||||
}
|
||||
|
||||
static void gen_absv_mask(TCGv_i64 d, TCGv_i64 b, unsigned vece)
|
||||
{
|
||||
TCGv_i64 t = tcg_temp_new_i64();
|
||||
int nbit = 8 << vece;
|
||||
|
||||
/* Create -1 for each negative element. */
|
||||
tcg_gen_shri_i64(t, b, nbit - 1);
|
||||
tcg_gen_andi_i64(t, t, dup_const(vece, 1));
|
||||
tcg_gen_muli_i64(t, t, (1 << nbit) - 1);
|
||||
|
||||
/*
|
||||
* Invert (via xor -1) and add one (via sub -1).
|
||||
* Because of the ordering the msb is cleared,
|
||||
* so we never have carry into the next element.
|
||||
*/
|
||||
tcg_gen_xor_i64(d, b, t);
|
||||
tcg_gen_sub_i64(d, d, t);
|
||||
|
||||
tcg_temp_free_i64(t);
|
||||
}
|
||||
|
||||
static void tcg_gen_vec_abs8_i64(TCGv_i64 d, TCGv_i64 b)
|
||||
{
|
||||
gen_absv_mask(d, b, MO_8);
|
||||
}
|
||||
|
||||
static void tcg_gen_vec_abs16_i64(TCGv_i64 d, TCGv_i64 b)
|
||||
{
|
||||
gen_absv_mask(d, b, MO_16);
|
||||
}
|
||||
|
||||
void tcg_gen_gvec_abs(unsigned vece, uint32_t dofs, uint32_t aofs,
|
||||
uint32_t oprsz, uint32_t maxsz)
|
||||
{
|
||||
static const TCGOpcode vecop_list[] = { INDEX_op_abs_vec, 0 };
|
||||
static const GVecGen2 g[4] = {
|
||||
{ .fni8 = tcg_gen_vec_abs8_i64,
|
||||
.fniv = tcg_gen_abs_vec,
|
||||
.fno = gen_helper_gvec_abs8,
|
||||
.opt_opc = vecop_list,
|
||||
.vece = MO_8 },
|
||||
{ .fni8 = tcg_gen_vec_abs16_i64,
|
||||
.fniv = tcg_gen_abs_vec,
|
||||
.fno = gen_helper_gvec_abs16,
|
||||
.opt_opc = vecop_list,
|
||||
.vece = MO_16 },
|
||||
{ .fni4 = tcg_gen_abs_i32,
|
||||
.fniv = tcg_gen_abs_vec,
|
||||
.fno = gen_helper_gvec_abs32,
|
||||
.opt_opc = vecop_list,
|
||||
.vece = MO_32 },
|
||||
{ .fni8 = tcg_gen_abs_i64,
|
||||
.fniv = tcg_gen_abs_vec,
|
||||
.fno = gen_helper_gvec_abs64,
|
||||
.opt_opc = vecop_list,
|
||||
.prefer_i64 = TCG_TARGET_REG_BITS == 64,
|
||||
.vece = MO_64 },
|
||||
};
|
||||
|
||||
tcg_debug_assert(vece <= MO_64);
|
||||
tcg_gen_gvec_2(dofs, aofs, oprsz, maxsz, &g[vece]);
|
||||
}
|
||||
|
||||
void tcg_gen_gvec_and(unsigned vece, uint32_t dofs, uint32_t aofs,
|
||||
uint32_t bofs, uint32_t oprsz, uint32_t maxsz)
|
||||
{
|
||||
|
|
|
@ -228,6 +228,8 @@ void tcg_gen_gvec_not(unsigned vece, uint32_t dofs, uint32_t aofs,
|
|||
uint32_t oprsz, uint32_t maxsz);
|
||||
void tcg_gen_gvec_neg(unsigned vece, uint32_t dofs, uint32_t aofs,
|
||||
uint32_t oprsz, uint32_t maxsz);
|
||||
void tcg_gen_gvec_abs(unsigned vece, uint32_t dofs, uint32_t aofs,
|
||||
uint32_t oprsz, uint32_t maxsz);
|
||||
|
||||
void tcg_gen_gvec_add(unsigned vece, uint32_t dofs, uint32_t aofs,
|
||||
uint32_t bofs, uint32_t oprsz, uint32_t maxsz);
|
||||
|
|
|
@ -110,6 +110,14 @@ bool tcg_can_emit_vecop_list(const TCGOpcode *list,
|
|||
continue;
|
||||
}
|
||||
break;
|
||||
case INDEX_op_abs_vec:
|
||||
if (tcg_can_emit_vec_op(INDEX_op_sub_vec, type, vece)
|
||||
&& (tcg_can_emit_vec_op(INDEX_op_smax_vec, type, vece) > 0
|
||||
|| tcg_can_emit_vec_op(INDEX_op_sari_vec, type, vece) > 0
|
||||
|| tcg_can_emit_vec_op(INDEX_op_cmp_vec, type, vece))) {
|
||||
continue;
|
||||
}
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
@ -429,6 +437,37 @@ void tcg_gen_neg_vec(unsigned vece, TCGv_vec r, TCGv_vec a)
|
|||
tcg_swap_vecop_list(hold_list);
|
||||
}
|
||||
|
||||
void tcg_gen_abs_vec(unsigned vece, TCGv_vec r, TCGv_vec a)
|
||||
{
|
||||
const TCGOpcode *hold_list;
|
||||
|
||||
tcg_assert_listed_vecop(INDEX_op_abs_vec);
|
||||
hold_list = tcg_swap_vecop_list(NULL);
|
||||
|
||||
if (!do_op2(vece, r, a, INDEX_op_abs_vec)) {
|
||||
TCGType type = tcgv_vec_temp(r)->base_type;
|
||||
TCGv_vec t = tcg_temp_new_vec(type);
|
||||
|
||||
tcg_debug_assert(tcg_can_emit_vec_op(INDEX_op_sub_vec, type, vece));
|
||||
if (tcg_can_emit_vec_op(INDEX_op_smax_vec, type, vece) > 0) {
|
||||
tcg_gen_neg_vec(vece, t, a);
|
||||
tcg_gen_smax_vec(vece, r, a, t);
|
||||
} else {
|
||||
if (tcg_can_emit_vec_op(INDEX_op_sari_vec, type, vece) > 0) {
|
||||
tcg_gen_sari_vec(vece, t, a, (8 << vece) - 1);
|
||||
} else {
|
||||
do_dupi_vec(t, MO_REG, 0);
|
||||
tcg_gen_cmp_vec(TCG_COND_LT, vece, t, a, t);
|
||||
}
|
||||
tcg_gen_xor_vec(vece, r, a, t);
|
||||
tcg_gen_sub_vec(vece, r, r, t);
|
||||
}
|
||||
|
||||
tcg_temp_free_vec(t);
|
||||
}
|
||||
tcg_swap_vecop_list(hold_list);
|
||||
}
|
||||
|
||||
static void do_shifti(TCGOpcode opc, unsigned vece,
|
||||
TCGv_vec r, TCGv_vec a, int64_t i)
|
||||
{
|
||||
|
|
|
@ -225,6 +225,7 @@ DEF(add_vec, 1, 2, 0, IMPLVEC)
|
|||
DEF(sub_vec, 1, 2, 0, IMPLVEC)
|
||||
DEF(mul_vec, 1, 2, 0, IMPLVEC | IMPL(TCG_TARGET_HAS_mul_vec))
|
||||
DEF(neg_vec, 1, 1, 0, IMPLVEC | IMPL(TCG_TARGET_HAS_neg_vec))
|
||||
DEF(abs_vec, 1, 1, 0, IMPLVEC | IMPL(TCG_TARGET_HAS_abs_vec))
|
||||
DEF(ssadd_vec, 1, 2, 0, IMPLVEC | IMPL(TCG_TARGET_HAS_sat_vec))
|
||||
DEF(usadd_vec, 1, 2, 0, IMPLVEC | IMPL(TCG_TARGET_HAS_sat_vec))
|
||||
DEF(sssub_vec, 1, 2, 0, IMPLVEC | IMPL(TCG_TARGET_HAS_sat_vec))
|
||||
|
|
|
@ -1616,6 +1616,8 @@ bool tcg_op_supported(TCGOpcode op)
|
|||
return have_vec && TCG_TARGET_HAS_not_vec;
|
||||
case INDEX_op_neg_vec:
|
||||
return have_vec && TCG_TARGET_HAS_neg_vec;
|
||||
case INDEX_op_abs_vec:
|
||||
return have_vec && TCG_TARGET_HAS_abs_vec;
|
||||
case INDEX_op_andc_vec:
|
||||
return have_vec && TCG_TARGET_HAS_andc_vec;
|
||||
case INDEX_op_orc_vec:
|
||||
|
|
|
@ -176,6 +176,7 @@ typedef uint64_t TCGRegSet;
|
|||
&& !defined(TCG_TARGET_HAS_v128) \
|
||||
&& !defined(TCG_TARGET_HAS_v256)
|
||||
#define TCG_TARGET_MAYBE_vec 0
|
||||
#define TCG_TARGET_HAS_abs_vec 0
|
||||
#define TCG_TARGET_HAS_neg_vec 0
|
||||
#define TCG_TARGET_HAS_not_vec 0
|
||||
#define TCG_TARGET_HAS_andc_vec 0
|
||||
|
|
Loading…
Reference in a new issue