While working something else, I noticed we could improve the following function code generation: ``` unsigned f(unsigned t) { if (t & ~(1<<30)) __builtin_unreachable(); return t != 0; } ``` Right know we just emit a comparison against 0 instead of just a shift right by 30. There is code in do_store_flag which already optimizes `(t & 1<<30) != 0` to `(t >> 30) & 1`. This patch extends it to handle the case where we know t has a nonzero of just one bit set.
OK? Bootstrapped and tested on x86_64-linux-gnu with no regressions. gcc/ChangeLog: * expr.cc (do_store_flag): Extend the one bit checking case to handle the case where we don't have an and but rather still one bit is known to be non-zero. --- gcc/expr.cc | 27 +++++++++++++++++++++------ 1 file changed, 21 insertions(+), 6 deletions(-) diff --git a/gcc/expr.cc b/gcc/expr.cc index 5ede094e705..91528e734e7 100644 --- a/gcc/expr.cc +++ b/gcc/expr.cc @@ -13083,15 +13083,30 @@ do_store_flag (sepops ops, rtx target, machine_mode mode) && integer_zerop (arg1) && (TYPE_PRECISION (ops->type) != 1 || TYPE_UNSIGNED (ops->type))) { - gimple *srcstmt = get_def_for_expr (arg0, BIT_AND_EXPR); - if (srcstmt - && integer_pow2p (gimple_assign_rhs2 (srcstmt))) + wide_int nz = tree_nonzero_bits (arg0); + + if (wi::popcount (nz) == 1) { + tree op0; + tree op1; + gimple *srcstmt = get_def_for_expr (arg0, BIT_AND_EXPR); + /* If the defining statement was (x & POW2), then remove the and + as we are going to add it back. */ + if (srcstmt + && integer_pow2p (gimple_assign_rhs2 (srcstmt))) + { + op0 = gimple_assign_rhs1 (srcstmt); + op1 = gimple_assign_rhs2 (srcstmt); + } + else + { + op0 = arg0; + op1 = wide_int_to_tree (TREE_TYPE (op0), nz); + } enum tree_code tcode = code == NE ? NE_EXPR : EQ_EXPR; type = lang_hooks.types.type_for_mode (mode, unsignedp); - tree temp = fold_build2_loc (loc, BIT_AND_EXPR, TREE_TYPE (arg1), - gimple_assign_rhs1 (srcstmt), - gimple_assign_rhs2 (srcstmt)); + tree temp = fold_build2_loc (loc, BIT_AND_EXPR, TREE_TYPE (op0), + op0, op1); temp = fold_single_bit_test (loc, tcode, temp, arg1, type); if (temp) return expand_expr (temp, target, VOIDmode, EXPAND_NORMAL); -- 2.31.1