Re: [PATCH] vect: Don't clear base_misaligned in update_epilogue_loop_vinfo [PR114566]

2024-04-05 Thread Richard Biener
On Fri, 5 Apr 2024, Jakub Jelinek wrote:

> Hi!
> 
> The following testcase is miscompiled, because in the vectorized
> epilogue the vectorizer assumes it can use aligned loads/stores
> (if the base decl gets alignment increased), but it actually doesn't
> increase that.
> This is because r10-4203-g97c1460367 added the hunk following
> patch removes.  The explanation feels reasonable, but actually it
> is not true as the testcase proves.
> The thing is, we vectorize the main loop with 64-byte vectors
> and the corresponding data refs have base_alignment 16 (the
> a array has DECL_ALIGN 128) and offset_alignment 32.  Now, because
> of the offset_alignment 32 rather than 64, we need to use unaligned
> loads/stores in the main loop (and ditto in the first load/store
> in vectorized epilogue).  But the second load/store in the vectorized
> epilogue uses only 32-byte vectors and because it is a multiple
> of offset_alignment, it checks if we could increase alignment of the
> a VAR_DECL, the function returns true, sets base_misaligned = true
> and says the access is then aligned.
> But when update_epilogue_loop_vinfo clears base_misaligned with the
> assumption that the var had to have the alignment increased already,
> the update of DECL_ALIGN doesn't happen anymore.
> 
> Now, I'd think this base_alignment = false was needed before
> r10-4030-gd2db7f7901 change was committed where it incorrectly
> overwrote DECL_ALIGN even if it was already larger, rather than
> just always increasing it.  But with that change in, it doesn't
> make sense to me anymore.

I think it was incorrect before - basically it assumed that the
main loop would never use misaligned accesses and the epilogue
analysis would align.  But for this we'd need to make sure to
never request such during epilogue analysis.

> Note, the testcase is latent on the trunk, but reproduces on the 13
> branch.
> 
> Bootstrapped/regtested on x86_64-linux and i686-linux on the trunk,
> plus tested with the testcase on 13 branch with -m32/-m64 without/with
> the tree-vect-loop.cc patch (where it FAILed before and now PASSes).
> Ok for trunk?

OK.

Thanks,
Richard.

> 2024-04-05  Jakub Jelinek  
> 
>   PR tree-optimization/114566
>   * tree-vect-loop.cc (update_epilogue_loop_vinfo): Don't clear
>   base_misaligned.
> 
>   * gcc.target/i386/avx512f-pr114566.c: New test.
> 
> --- gcc/tree-vect-loop.cc.jj  2024-04-04 00:48:05.932072711 +0200
> +++ gcc/tree-vect-loop.cc 2024-04-05 00:59:33.743101468 +0200
> @@ -11590,9 +11590,7 @@ find_in_mapping (tree t, void *context)
> corresponding dr_vec_info need to be reconnected to the EPILOGUE's
> stmt_vec_infos, their statements need to point to their corresponding 
> copy,
> if they are gather loads or scatter stores then their reference needs to 
> be
> -   updated to point to its corresponding copy and finally we set
> -   'base_misaligned' to false as we have already peeled for alignment in the
> -   prologue of the main loop.  */
> +   updated to point to its corresponding copy.  */
>  
>  static void
>  update_epilogue_loop_vinfo (class loop *epilogue, tree advance)
> @@ -11736,10 +11734,6 @@ update_epilogue_loop_vinfo (class loop *
>   }
>DR_STMT (dr) = STMT_VINFO_STMT (stmt_vinfo);
>stmt_vinfo->dr_aux.stmt = stmt_vinfo;
> -  /* The vector size of the epilogue is smaller than that of the main 
> loop
> -  so the alignment is either the same or lower. This means the dr will
> -  thus by definition be aligned.  */
> -  STMT_VINFO_DR_INFO (stmt_vinfo)->base_misaligned = false;
>  }
>  
>epilogue_vinfo->shared->datarefs_copy.release ();
> --- gcc/testsuite/gcc.target/i386/avx512f-pr114566.c.jj   2024-04-05 
> 11:21:04.282639386 +0200
> +++ gcc/testsuite/gcc.target/i386/avx512f-pr114566.c  2024-04-05 
> 11:21:04.282639386 +0200
> @@ -0,0 +1,34 @@
> +/* PR tree-optimization/114566 */
> +/* { dg-do run } */
> +/* { dg-options "-O3 -mavx512f" } */
> +/* { dg-additional-options "-fstack-protector-strong" { target 
> fstack_protector } } */
> +/* { dg-require-effective-target avx512f } */
> +
> +#define AVX512F
> +#include "avx512f-helper.h"
> +
> +__attribute__((noipa)) int
> +foo (float x, float y)
> +{
> +  float a[8][56];
> +  __builtin_memset (a, 0, sizeof (a));
> +
> +  for (int j = 0; j < 8; j++)
> +for (int k = 0; k < 56; k++)
> +  {
> + float b = k * y;
> + if (b < 0.)
> +   b = 0.;
> + if (b > 0.)
> +   b = 0.;
> + a[j][k] += b;
> +  }
> +
> +  return __builtin_log (x);
> +}
> +
> +void
> +TEST (void)
> +{
> +  foo (86.25f, 0.625f);
> +}
> 
>   Jakub
> 
> 

-- 
Richard Biener 
SUSE Software Solutions Germany GmbH,
Frankenstrasse 146, 90461 Nuernberg, Germany;
GF: Ivo Totev, Andrew McDonald, Werner Knoblich; (HRB 36809, AG Nuernberg)


[PATCH] vect: Don't clear base_misaligned in update_epilogue_loop_vinfo [PR114566]

2024-04-05 Thread Jakub Jelinek
Hi!

The following testcase is miscompiled, because in the vectorized
epilogue the vectorizer assumes it can use aligned loads/stores
(if the base decl gets alignment increased), but it actually doesn't
increase that.
This is because r10-4203-g97c1460367 added the hunk following
patch removes.  The explanation feels reasonable, but actually it
is not true as the testcase proves.
The thing is, we vectorize the main loop with 64-byte vectors
and the corresponding data refs have base_alignment 16 (the
a array has DECL_ALIGN 128) and offset_alignment 32.  Now, because
of the offset_alignment 32 rather than 64, we need to use unaligned
loads/stores in the main loop (and ditto in the first load/store
in vectorized epilogue).  But the second load/store in the vectorized
epilogue uses only 32-byte vectors and because it is a multiple
of offset_alignment, it checks if we could increase alignment of the
a VAR_DECL, the function returns true, sets base_misaligned = true
and says the access is then aligned.
But when update_epilogue_loop_vinfo clears base_misaligned with the
assumption that the var had to have the alignment increased already,
the update of DECL_ALIGN doesn't happen anymore.

Now, I'd think this base_alignment = false was needed before
r10-4030-gd2db7f7901 change was committed where it incorrectly
overwrote DECL_ALIGN even if it was already larger, rather than
just always increasing it.  But with that change in, it doesn't
make sense to me anymore.

Note, the testcase is latent on the trunk, but reproduces on the 13
branch.

Bootstrapped/regtested on x86_64-linux and i686-linux on the trunk,
plus tested with the testcase on 13 branch with -m32/-m64 without/with
the tree-vect-loop.cc patch (where it FAILed before and now PASSes).
Ok for trunk?

2024-04-05  Jakub Jelinek  

PR tree-optimization/114566
* tree-vect-loop.cc (update_epilogue_loop_vinfo): Don't clear
base_misaligned.

* gcc.target/i386/avx512f-pr114566.c: New test.

--- gcc/tree-vect-loop.cc.jj2024-04-04 00:48:05.932072711 +0200
+++ gcc/tree-vect-loop.cc   2024-04-05 00:59:33.743101468 +0200
@@ -11590,9 +11590,7 @@ find_in_mapping (tree t, void *context)
corresponding dr_vec_info need to be reconnected to the EPILOGUE's
stmt_vec_infos, their statements need to point to their corresponding copy,
if they are gather loads or scatter stores then their reference needs to be
-   updated to point to its corresponding copy and finally we set
-   'base_misaligned' to false as we have already peeled for alignment in the
-   prologue of the main loop.  */
+   updated to point to its corresponding copy.  */
 
 static void
 update_epilogue_loop_vinfo (class loop *epilogue, tree advance)
@@ -11736,10 +11734,6 @@ update_epilogue_loop_vinfo (class loop *
}
   DR_STMT (dr) = STMT_VINFO_STMT (stmt_vinfo);
   stmt_vinfo->dr_aux.stmt = stmt_vinfo;
-  /* The vector size of the epilogue is smaller than that of the main loop
-so the alignment is either the same or lower. This means the dr will
-thus by definition be aligned.  */
-  STMT_VINFO_DR_INFO (stmt_vinfo)->base_misaligned = false;
 }
 
   epilogue_vinfo->shared->datarefs_copy.release ();
--- gcc/testsuite/gcc.target/i386/avx512f-pr114566.c.jj 2024-04-05 
11:21:04.282639386 +0200
+++ gcc/testsuite/gcc.target/i386/avx512f-pr114566.c2024-04-05 
11:21:04.282639386 +0200
@@ -0,0 +1,34 @@
+/* PR tree-optimization/114566 */
+/* { dg-do run } */
+/* { dg-options "-O3 -mavx512f" } */
+/* { dg-additional-options "-fstack-protector-strong" { target 
fstack_protector } } */
+/* { dg-require-effective-target avx512f } */
+
+#define AVX512F
+#include "avx512f-helper.h"
+
+__attribute__((noipa)) int
+foo (float x, float y)
+{
+  float a[8][56];
+  __builtin_memset (a, 0, sizeof (a));
+
+  for (int j = 0; j < 8; j++)
+for (int k = 0; k < 56; k++)
+  {
+   float b = k * y;
+   if (b < 0.)
+ b = 0.;
+   if (b > 0.)
+ b = 0.;
+   a[j][k] += b;
+  }
+
+  return __builtin_log (x);
+}
+
+void
+TEST (void)
+{
+  foo (86.25f, 0.625f);
+}

Jakub