Hi Hongtao, Thanks for the review and pointing out the remaining uses of force_reg that I'd overlooked. Here's a revised version of the patch that incorporates your feedback. One minor change was that rather than using memory_operand, which as you point out also needs to include bcst_mem_operand, it's simpler to invert the logic to check for register_operand [i.e. the first operand must be a register].
This patch has been tested on x86_64-pc-linux-gnu with make bootstrap and make -k check, both with and without --target_board=unix{-m32} with no new failures. Ok for mainline? 2024-07-11 Roger Sayle <ro...@nextmovesoftware.com> Hongtao Liu <hongtao....@intel.com> gcc/ChangeLog * config/i386/i386-expand.cc (ix86_broadcast_from_constant): Use CONST_VECTOR_P instead of comparison against GET_CODE. (ix86_gen_bcst_mem): Likewise. (ix86_ternlog_leaf_p): Likewise. (ix86_ternlog_operand_p): ix86_ternlog_leaf_p is always true for vector_all_ones_operand. (ix86_expand_ternlog_bin_op): Use CONST_VECTOR_P instead of equality comparison against GET_CODE. Replace call to force_reg with gen_reg_rtx and emit_move_insn (for VEC_DUPLICATE broadcast). Check for !register_operand instead of memory_operand. Support CONST_VECTORs by calling force_const_mem. (ix86_expand_ternlog): Fix indentation whitespace. Allow ix86_ternlog_leaf_p as ix86_expand_ternlog_andnot's second operand. Use CONST_VECTOR_P instead of equality against GET_CODE. Use gen_reg_rtx and emit_move_insn for ~a, ~b and ~c cases. Thanks again, Roger > -----Original Message----- > From: Hongtao Liu <crazy...@gmail.com> > Sent: 08 July 2024 02:55 > To: Roger Sayle <ro...@nextmovesoftware.com> > Cc: gcc-patches@gcc.gnu.org; Uros Bizjak <ubiz...@gmail.com> > Subject: Re: [x86 SSE PATCH] Some AVX512 ternlog expansion refinements. > > On Sun, Jul 7, 2024 at 5:00 PM Roger Sayle <ro...@nextmovesoftware.com> > wrote: > > Hi Hongtao, > > This should address concerns about the remaining use of force_reg. > > > 51@@ -25793,15 +25792,20 @@ ix86_expand_ternlog_binop (enum rtx_code > code, machine_mode mode, > 52 if (GET_MODE (op1) != mode) > 53 op1 = gen_lowpart (mode, op1); > 54 > 55- if (GET_CODE (op0) == CONST_VECTOR) 56+ if (CONST_VECTOR_P (op0)) > 57 op0 = validize_mem (force_const_mem (mode, op0)); > 58- if (GET_CODE (op1) == CONST_VECTOR) 59+ if (CONST_VECTOR_P (op1)) > 60 op1 = validize_mem (force_const_mem (mode, op1)); > 61 > 62 if (memory_operand (op0, mode)) > 63 { > 64 if (memory_operand (op1, mode)) > 65- op0 = force_reg (mode, op0); > 66+ { > 67+ /* We can't use force_reg (op0, mode). */ > 68+ rtx reg = gen_reg_rtx (mode); > 69+ emit_move_insn (reg, op0); > 70+ op0 = reg; > 71+ } > Shouldn't we handle bcst_mem_operand instead of > memory_operand(bcst_memory_operand is not a memory_operand)? > so maybe > if (memory_operand (op0, mode0) || bcst_mem_operand (op0, mode0) > if (memory_operand (op1, mode) || bcst_mem_operand (op1, mode0)? > 72 else > 73 std::swap (op0, op1); > 74 } > > Also there's force_reg in below 3 cases, are there any restrictions to avoid > bcst_mem_operand into them? > case 0x0f: /* ~a */ > case 0x33: /* ~b */ > case 0x33: /* ~b */ > .. > if (!TARGET_64BIT && !register_operand (op2, mode)) > op2 = force_reg (mode, op2); > > -- > BR, > Hongtao
diff --git a/gcc/config/i386/i386-expand.cc b/gcc/config/i386/i386-expand.cc index dd2c3a8..7818d40 100644 --- a/gcc/config/i386/i386-expand.cc +++ b/gcc/config/i386/i386-expand.cc @@ -612,7 +612,7 @@ ix86_broadcast_from_constant (machine_mode mode, rtx op) return nullptr; rtx constant = get_pool_constant (XEXP (op, 0)); - if (GET_CODE (constant) != CONST_VECTOR) + if (!CONST_VECTOR_P (constant)) return nullptr; /* There could be some rtx like @@ -622,7 +622,7 @@ ix86_broadcast_from_constant (machine_mode mode, rtx op) { constant = simplify_subreg (mode, constant, GET_MODE (constant), 0); - if (constant == nullptr || GET_CODE (constant) != CONST_VECTOR) + if (constant == nullptr || !CONST_VECTOR_P (constant)) return nullptr; } @@ -25532,7 +25532,7 @@ static rtx ix86_gen_bcst_mem (machine_mode mode, rtx x) { if (!TARGET_AVX512F - || GET_CODE (x) != CONST_VECTOR + || !CONST_VECTOR_P (x) || (!TARGET_AVX512VL && (GET_MODE_SIZE (mode) != 64 || !TARGET_EVEX512)) || !VALID_BCST_MODE_P (GET_MODE_INNER (mode)) @@ -25722,7 +25722,7 @@ ix86_ternlog_leaf_p (rtx op, machine_mode mode) problems splitting instructions. */ return register_operand (op, mode) || MEM_P (op) - || GET_CODE (op) == CONST_VECTOR + || CONST_VECTOR_P (op) || bcst_mem_operand (op, mode); } @@ -25772,8 +25772,7 @@ ix86_ternlog_operand_p (rtx op) op1 = XEXP (op, 1); /* Prefer pxor, or one_cmpl<vmode>2. */ if (ix86_ternlog_leaf_p (XEXP (op, 0), mode) - && (ix86_ternlog_leaf_p (op1, mode) - || vector_all_ones_operand (op1, mode))) + && ix86_ternlog_leaf_p (XEXP (op, 1), mode)) return false; break; @@ -25793,15 +25792,20 @@ ix86_expand_ternlog_binop (enum rtx_code code, machine_mode mode, if (GET_MODE (op1) != mode) op1 = gen_lowpart (mode, op1); - if (GET_CODE (op0) == CONST_VECTOR) + if (CONST_VECTOR_P (op0)) op0 = validize_mem (force_const_mem (mode, op0)); - if (GET_CODE (op1) == CONST_VECTOR) + if (CONST_VECTOR_P (op1)) op1 = validize_mem (force_const_mem (mode, op1)); - if (memory_operand (op0, mode)) + if (!register_operand (op0, mode)) { - if (memory_operand (op1, mode)) - op0 = force_reg (mode, op0); + if (!register_operand (op1, mode)) + { + /* We can't use force_reg (op0, mode). */ + rtx reg = gen_reg_rtx (mode); + emit_move_insn (reg, op0); + op0 = reg; + } else std::swap (op0, op1); } @@ -25820,6 +25824,8 @@ ix86_expand_ternlog_andnot (machine_mode mode, rtx op0, rtx op1, rtx target) op0 = gen_rtx_NOT (mode, op0); if (GET_MODE (op1) != mode) op1 = gen_lowpart (mode, op1); + if (CONST_VECTOR_P (op1)) + op1 = validize_mem (force_const_mem (mode, op1)); emit_move_insn (target, gen_rtx_AND (mode, op0, op1)); return target; } @@ -25856,9 +25862,9 @@ ix86_expand_ternlog (machine_mode mode, rtx op0, rtx op1, rtx op2, int idx, { case 0x00: if ((!op0 || !side_effects_p (op0)) - && (!op1 || !side_effects_p (op1)) - && (!op2 || !side_effects_p (op2))) - { + && (!op1 || !side_effects_p (op1)) + && (!op2 || !side_effects_p (op2))) + { emit_move_insn (target, CONST0_RTX (mode)); return target; } @@ -25867,26 +25873,31 @@ ix86_expand_ternlog (machine_mode mode, rtx op0, rtx op1, rtx op2, int idx, case 0x0a: /* ~a&c */ if ((!op1 || !side_effects_p (op1)) && op0 && register_operand (op0, mode) - && op2 && register_operand (op2, mode)) + && op2 && ix86_ternlog_leaf_p (op2, mode)) return ix86_expand_ternlog_andnot (mode, op0, op2, target); break; case 0x0c: /* ~a&b */ if ((!op2 || !side_effects_p (op2)) && op0 && register_operand (op0, mode) - && op1 && register_operand (op1, mode)) + && op1 && ix86_ternlog_leaf_p (op1, mode)) return ix86_expand_ternlog_andnot (mode, op0, op1, target); break; case 0x0f: /* ~a */ if ((!op1 || !side_effects_p (op1)) && (!op2 || !side_effects_p (op2)) - && op0) + && op0) { if (GET_MODE (op0) != mode) op0 = gen_lowpart (mode, op0); if (!TARGET_64BIT && !register_operand (op0, mode)) - op0 = force_reg (mode, op0); + { + /* Avoid force_reg (mode, op0). */ + rtx reg = gen_reg_rtx (mode); + emit_move_insn (reg, op0); + op0 = reg; + } emit_move_insn (target, gen_rtx_XOR (mode, op0, CONSTM1_RTX (mode))); return target; } @@ -25895,13 +25906,13 @@ ix86_expand_ternlog (machine_mode mode, rtx op0, rtx op1, rtx op2, int idx, case 0x22: /* ~b&c */ if ((!op0 || !side_effects_p (op0)) && op1 && register_operand (op1, mode) - && op2 && register_operand (op2, mode)) + && op2 && ix86_ternlog_leaf_p (op2, mode)) return ix86_expand_ternlog_andnot (mode, op1, op2, target); break; case 0x30: /* ~b&a */ if ((!op2 || !side_effects_p (op2)) - && op0 && register_operand (op0, mode) + && op0 && ix86_ternlog_leaf_p (op0, mode) && op1 && register_operand (op1, mode)) return ix86_expand_ternlog_andnot (mode, op1, op0, target); break; @@ -25909,33 +25920,39 @@ ix86_expand_ternlog (machine_mode mode, rtx op0, rtx op1, rtx op2, int idx, case 0x33: /* ~b */ if ((!op0 || !side_effects_p (op0)) && (!op2 || !side_effects_p (op2)) - && op1) + && op1) { if (GET_MODE (op1) != mode) op1 = gen_lowpart (mode, op1); if (!TARGET_64BIT && !register_operand (op1, mode)) - op1 = force_reg (mode, op1); + { + /* Avoid force_reg (mode, op1). */ + rtx reg = gen_reg_rtx (mode); + emit_move_insn (reg, op1); + op1 = reg; + } emit_move_insn (target, gen_rtx_XOR (mode, op1, CONSTM1_RTX (mode))); return target; } break; case 0x3c: /* a^b */ - if (op0 && op1 - && (!op2 || !side_effects_p (op2))) + if (op0 && ix86_ternlog_leaf_p (op0, mode) + && op1 && ix86_ternlog_leaf_p (op1, mode) + && (!op2 || !side_effects_p (op2))) return ix86_expand_ternlog_binop (XOR, mode, op0, op1, target); break; case 0x44: /* ~c&b */ if ((!op0 || !side_effects_p (op0)) - && op1 && register_operand (op1, mode) + && op1 && ix86_ternlog_leaf_p (op1, mode) && op2 && register_operand (op2, mode)) return ix86_expand_ternlog_andnot (mode, op2, op1, target); break; case 0x50: /* ~c&a */ if ((!op1 || !side_effects_p (op1)) - && op0 && register_operand (op0, mode) + && op0 && ix86_ternlog_leaf_p (op0, mode) && op2 && register_operand (op2, mode)) return ix86_expand_ternlog_andnot (mode, op2, op0, target); break; @@ -25943,45 +25960,54 @@ ix86_expand_ternlog (machine_mode mode, rtx op0, rtx op1, rtx op2, int idx, case 0x55: /* ~c */ if ((!op0 || !side_effects_p (op0)) && (!op1 || !side_effects_p (op1)) - && op2) + && op2) { if (GET_MODE (op2) != mode) op2 = gen_lowpart (mode, op2); if (!TARGET_64BIT && !register_operand (op2, mode)) - op2 = force_reg (mode, op2); + { + /* Avoid force_reg (mode, op2). */ + rtx reg = gen_reg_rtx (mode); + emit_move_insn (reg, op2); + op2 = reg; + } emit_move_insn (target, gen_rtx_XOR (mode, op2, CONSTM1_RTX (mode))); return target; } break; case 0x5a: /* a^c */ - if (op0 && op2 - && (!op1 || !side_effects_p (op1))) + if (op0 && ix86_ternlog_leaf_p (op0, mode) + && op2 && ix86_ternlog_leaf_p (op2, mode) + && (!op1 || !side_effects_p (op1))) return ix86_expand_ternlog_binop (XOR, mode, op0, op2, target); break; case 0x66: /* b^c */ if ((!op0 || !side_effects_p (op0)) - && op1 && op2) + && op1 && ix86_ternlog_leaf_p (op1, mode) + && op2 && ix86_ternlog_leaf_p (op2, mode)) return ix86_expand_ternlog_binop (XOR, mode, op1, op2, target); break; case 0x88: /* b&c */ if ((!op0 || !side_effects_p (op0)) - && op1 && op2) + && op1 && ix86_ternlog_leaf_p (op1, mode) + && op2 && ix86_ternlog_leaf_p (op2, mode)) return ix86_expand_ternlog_binop (AND, mode, op1, op2, target); break; case 0xa0: /* a&c */ if ((!op1 || !side_effects_p (op1)) - && op0 && op2) + && op0 && ix86_ternlog_leaf_p (op0, mode) + && op2 && ix86_ternlog_leaf_p (op2, mode)) return ix86_expand_ternlog_binop (AND, mode, op0, op2, target); break; case 0xaa: /* c */ if ((!op0 || !side_effects_p (op0)) && (!op1 || !side_effects_p (op1)) - && op2) + && op2) { if (GET_MODE (op2) != mode) op2 = gen_lowpart (mode, op2); @@ -25991,14 +26017,15 @@ ix86_expand_ternlog (machine_mode mode, rtx op0, rtx op1, rtx op2, int idx, break; case 0xc0: /* a&b */ - if (op0 && op1 - && (!op2 || !side_effects_p (op2))) + if (op0 && ix86_ternlog_leaf_p (op0, mode) + && op1 && ix86_ternlog_leaf_p (op1, mode) + && (!op2 || !side_effects_p (op2))) return ix86_expand_ternlog_binop (AND, mode, op0, op1, target); break; case 0xcc: /* b */ if ((!op0 || !side_effects_p (op0)) - && op1 + && op1 && (!op2 || !side_effects_p (op2))) { if (GET_MODE (op1) != mode) @@ -26010,13 +26037,14 @@ ix86_expand_ternlog (machine_mode mode, rtx op0, rtx op1, rtx op2, int idx, case 0xee: /* b|c */ if ((!op0 || !side_effects_p (op0)) - && op1 && op2) + && op1 && ix86_ternlog_leaf_p (op1, mode) + && op2 && ix86_ternlog_leaf_p (op2, mode)) return ix86_expand_ternlog_binop (IOR, mode, op1, op2, target); break; case 0xf0: /* a */ if (op0 - && (!op1 || !side_effects_p (op1)) + && (!op1 || !side_effects_p (op1)) && (!op2 || !side_effects_p (op2))) { if (GET_MODE (op0) != mode) @@ -26027,23 +26055,25 @@ ix86_expand_ternlog (machine_mode mode, rtx op0, rtx op1, rtx op2, int idx, break; case 0xfa: /* a|c */ - if (op0 && op2 - && (!op1 || !side_effects_p (op1))) + if (op0 && ix86_ternlog_leaf_p (op0, mode) + && op2 && ix86_ternlog_leaf_p (op2, mode) + && (!op1 || !side_effects_p (op1))) return ix86_expand_ternlog_binop (IOR, mode, op0, op2, target); break; case 0xfc: /* a|b */ - if (op0 && op1 - && (!op2 || !side_effects_p (op2))) + if (op0 && ix86_ternlog_leaf_p (op0, mode) + && op1 && ix86_ternlog_leaf_p (op1, mode) + && (!op2 || !side_effects_p (op2))) return ix86_expand_ternlog_binop (IOR, mode, op0, op1, target); break; case 0xff: if ((!op0 || !side_effects_p (op0)) - && (!op1 || !side_effects_p (op1)) - && (!op2 || !side_effects_p (op2))) - { - emit_move_insn (target, CONSTM1_RTX (mode)); + && (!op1 || !side_effects_p (op1)) + && (!op2 || !side_effects_p (op2))) + { + emit_move_insn (target, CONSTM1_RTX (mode)); return target; } break; @@ -26066,7 +26096,7 @@ ix86_expand_ternlog (machine_mode mode, rtx op0, rtx op1, rtx op2, int idx, tmp2 = copy_rtx (tmp0); else if (rtx_equal_p (op1, op2)) tmp2 = copy_rtx (tmp1); - else if (GET_CODE (op2) == CONST_VECTOR) + else if (CONST_VECTOR_P (op2)) { if (GET_MODE (op2) != mode) op2 = gen_lowpart (mode, op2);