Hi Richard,
Many thanks for the review and recommendation to use nop_convert?.
This revised patch implements that suggestion, which required a little
experimentation/tweaking as ranger/EVRP records the ranges on the
useless type conversions rather than the multiplications.

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-14  Roger Sayle  <ro...@nextmovesoftware.com>
            Richard Biener  <rguent...@suse.de>

gcc/ChangeLog
        PR tree-optimization/114661
        * match.pd ((X*C1)|(X*C2) to X*(C1+C2)): Allow optional useless
        type conversions around multiplicaitions, such as those inserted
        by this transformation.

gcc/testsuite/ChangeLog
        PR tree-optimization/114661
        * gcc.dg/pr114661.c: New test case.


Thanks again,
Roger
--

> -----Original Message-----
> From: Richard Biener <richard.guent...@gmail.com>
> Sent: 10 July 2024 12:34
> To: Roger Sayle <ro...@nextmovesoftware.com>
> Cc: gcc-patches@gcc.gnu.org
> Subject: Re: [match.pd PATCH] PR tree-optimization/114661: Generalize
> MULT_EXPR recognition.
> 
> On Wed, Jul 10, 2024 at 12:28 AM Roger Sayle <ro...@nextmovesoftware.com>
> wrote:
> >
> > This patch resolves PR tree-optimization/114661, by generalizing the
> > set of expressions that we canonicalize to multiplication.  This
> > extends the
> > optimization(s) contributed (by me) back in July 2021.
> > https://gcc.gnu.org/pipermail/gcc-patches/2021-July/575999.html
> >
> > The existing transformation folds (X*C1)^(X<<C2) into X*C3 when
> > allowed.  A subtlety is that for non-wrapping integer types, we
> > actually fold this into (int)((unsigned)X*C3) so that we don't
> > introduce an undefined overflow that wasn't in the original.
> > Unfortunately, this transformation confuses itself, as the type-safe
> > multiplication isn't recognized when further combining bit operations.
> > Fixed here by adding transforms to turn (int)((unsigned)X*C1)^(X<<C2)
> > into (int)((unsigned)X*C3) so that match.pd and EVRP can continue to
> > construct multiplications.
> >
> > For the example given in the PR:
> >
> > unsigned mul(unsigned char c) {
> >     if (c > 3) __builtin_unreachable();
> >     return c << 18 | c << 15 |
> >            c << 12 | c << 9 |
> >            c << 6 | c << 3 | c;
> > }
> >
> > GCC on x86_64 with -O2 previously generated:
> >
> > mul:    movzbl  %dil, %edi
> >         leal    (%rdi,%rdi,8), %edx
> >         leal    0(,%rdx,8), %eax
> >         movl    %edx, %ecx
> >         sall    $15, %edx
> >         orl     %edi, %eax
> >         sall    $9, %ecx
> >         orl     %ecx, %eax
> >         orl     %edx, %eax
> >         ret
> >
> > with this patch we now generate:
> >
> > mul:    movzbl  %dil, %eax
> >         imull   $299593, %eax, %eax
> >         ret
> >
> > 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?
> 
> I'm looking at the difference between the existing
> 
>  (simplify
>   (op:c (mult:s@0 @1 INTEGER_CST@2)
>         (lshift:s@3 @1 INTEGER_CST@4))
>   (if (INTEGRAL_TYPE_P (type) && TYPE_OVERFLOW_WRAPS (type)
>        && tree_int_cst_sgn (@4) > 0
>        && (tree_nonzero_bits (@0) & tree_nonzero_bits (@3)) == 0)
>    (with { wide_int wone = wi::one (TYPE_PRECISION (type));
>            wide_int c = wi::add (wi::to_wide (@2),
>                                  wi::lshift (wone, wi::to_wide (@4))); }
>     (mult @1 { wide_int_to_tree (type, c); }))))
> 
> and
> 
> + (simplify
> +  (op:c (convert:s@0 (mult:s@1 (convert @2) INTEGER_CST@3))
> +       (lshift:s@4 @2 INTEGER_CST@5))
> +  (if (INTEGRAL_TYPE_P (type)
> +       && INTEGRAL_TYPE_P (TREE_TYPE (@1))
> +       && TREE_TYPE (@2) == type
> +       && TYPE_UNSIGNED (TREE_TYPE (@1))
> +       && TYPE_PRECISION (type) == TYPE_PRECISION (TREE_TYPE (@1))
> +       && tree_int_cst_sgn (@5) > 0
> +       && (tree_nonzero_bits (@0) & tree_nonzero_bits (@4)) == 0)
> +   (with { tree t = TREE_TYPE (@1);
> +          wide_int wone = wi::one (TYPE_PRECISION (t));
> +          wide_int c = wi::add (wi::to_wide (@3),
> +                                wi::lshift (wone, wi::to_wide (@5))); }
> +    (convert (mult:t (convert:t @2) { wide_int_to_tree (t, c); })))))
> 
> and wonder whether wrapping of the multiplication is required for correctness,
> specifically the former seems to allow signed types with -fwrapv while the 
> latter
> won't.  It also looks the patterns could be merged doing
> 
>  (simplify
>   (op:c (nop_convert:s? (mult:s@0 (nop_convert? @1) INTEGER_CST@2)
>         (lshift:s@3 @1 INTEGER_CST@4))
> 
> and by using nop_convert instead of convert simplify the condition?
> 
> Richard.
> 
> >
> > 2024-07-09  Roger Sayle  <ro...@nextmovesoftware.com>
> >
> > gcc/ChangeLog
> >         PR tree-optimization/114661
> >         * match.pd ((X*C1)|(X*C2) to X*(C1+C2)): Additionally recognize
> >         multiplications surrounded by casts to an unsigned type and back
> >         such as those generated by these transformations.
> >
> > gcc/testsuite/ChangeLog
> >         PR tree-optimization/114661
> >         * gcc.dg/pr114661.c: New test case.
> >
> >
> > Thanks in advance,
> > Roger
> > --
> >
diff --git a/gcc/match.pd b/gcc/match.pd
index 4edfa2a..a66b00a 100644
--- a/gcc/match.pd
+++ b/gcc/match.pd
@@ -4156,30 +4156,39 @@ DEFINE_INT_AND_FLOAT_ROUND_FN (RINT)
    Likewise, handle (X<<C3) and X as legitimate variants of X*C.  */
 (for op (bit_ior bit_xor)
  (simplify
-  (op (mult:s@0 @1 INTEGER_CST@2)
-      (mult:s@3 @1 INTEGER_CST@4))
-  (if (INTEGRAL_TYPE_P (type) && TYPE_OVERFLOW_WRAPS (type)
-       && (tree_nonzero_bits (@0) & tree_nonzero_bits (@3)) == 0)
-   (mult @1
-        { wide_int_to_tree (type, wi::to_wide (@2) + wi::to_wide (@4)); })))
+  (op (nop_convert?:s@6 (mult:s@0 (nop_convert? @1) INTEGER_CST@2))
+      (nop_convert?:s@7 (mult:s@3 (nop_convert? @5) INTEGER_CST@4)))
+  (if (INTEGRAL_TYPE_P (type)
+       && operand_equal_p (@1, @5, 0)
+       && (tree_nonzero_bits (@6) & tree_nonzero_bits (@7)) == 0)
+   (with { tree t = type;
+          if (!TYPE_OVERFLOW_WRAPS (t))
+            t = unsigned_type_for (t);
+          wide_int c = wi::add (wi::to_wide (@2), wi::to_wide (@4)); }
+    (convert (mult:t (convert:t @1) { wide_int_to_tree (t, c); })))))
  (simplify
-  (op:c (mult:s@0 @1 INTEGER_CST@2)
+  (op:c (nop_convert?:s@5 (mult:s@0 (nop_convert? @1) INTEGER_CST@2))
        (lshift:s@3 @1 INTEGER_CST@4))
-  (if (INTEGRAL_TYPE_P (type) && TYPE_OVERFLOW_WRAPS (type)
+  (if (INTEGRAL_TYPE_P (type)
        && tree_int_cst_sgn (@4) > 0
-       && (tree_nonzero_bits (@0) & tree_nonzero_bits (@3)) == 0)
-   (with { wide_int wone = wi::one (TYPE_PRECISION (type));
+       && (tree_nonzero_bits (@5) & tree_nonzero_bits (@3)) == 0)
+   (with { tree t = type;
+          if (!TYPE_OVERFLOW_WRAPS (t))
+            t = unsigned_type_for (t);
+          wide_int wone = wi::one (TYPE_PRECISION (type));
           wide_int c = wi::add (wi::to_wide (@2),
                                 wi::lshift (wone, wi::to_wide (@4))); }
-    (mult @1 { wide_int_to_tree (type, c); }))))
+    (convert (mult:t (convert:t @1) { wide_int_to_tree (t, c); })))))
  (simplify
-  (op:c (mult:s@0 @1 INTEGER_CST@2)
+  (op:c (nop_convert?:s@3 (mult:s@0 (nop_convert? @1) INTEGER_CST@2))
        @1)
-  (if (INTEGRAL_TYPE_P (type) && TYPE_OVERFLOW_WRAPS (type)
-       && (tree_nonzero_bits (@0) & tree_nonzero_bits (@1)) == 0)
-   (mult @1
-        { wide_int_to_tree (type,
-                            wi::add (wi::to_wide (@2), 1)); })))
+  (if (INTEGRAL_TYPE_P (type)
+       && (tree_nonzero_bits (@3) & tree_nonzero_bits (@1)) == 0)
+   (with { tree t = type;
+          if (!TYPE_OVERFLOW_WRAPS (t))
+            t = unsigned_type_for (t);
+          wide_int c = wi::add (wi::to_wide (@2), 1); }
+    (convert (mult:t (convert:t @1) { wide_int_to_tree (t, c); })))))
  (simplify
   (op (lshift:s@0 @1 INTEGER_CST@2)
       (lshift:s@3 @1 INTEGER_CST@4))
diff --git a/gcc/testsuite/gcc.dg/pr114661.c b/gcc/testsuite/gcc.dg/pr114661.c
new file mode 100644
index 0000000..e6b5c69
--- /dev/null
+++ b/gcc/testsuite/gcc.dg/pr114661.c
@@ -0,0 +1,10 @@
+/* { dg-do compile } */
+/* { dg-options "-O2 -fdump-tree-evrp" } */
+
+unsigned mul(unsigned char c) {
+    if (c > 3) __builtin_unreachable();
+    return c << 18 | c << 15 |
+        c << 12 | c << 9 |
+        c << 6 | c << 3 | c;
+}
+/* { dg-final { scan-tree-dump-times " \\* 299593" 1 "evrp" } } */

Reply via email to