[PATCH] tree-optimization/104716 - check if we can copy loop in loop distribution

[Richard Biener via Gcc-patches]

The following checks whether we can copy the loop before attempting
to do so in loop distribution.  In the testcase there's a computed
goto and thus abnormal edges which we cannot redirect.

Bootstrapped on x86_64-unknown-linux-gnu, testing in progress.

2022-03-01  Richard Biener  <rguent...@suse.de>

        PR tree-optimization/104716
        * tree-loop-distribution.cc (find_seed_stmts_for_distribution):
        Check if we can copy the loop.

        * gfortran.dg/pr104716.f: New testcase.
---
 gcc/testsuite/gfortran.dg/pr104716.f | 31 ++++++++++++++++++++++++++++
 gcc/tree-loop-distribution.cc        | 14 +++++++++++++
 2 files changed, 45 insertions(+)
 create mode 100644 gcc/testsuite/gfortran.dg/pr104716.f

diff --git a/gcc/testsuite/gfortran.dg/pr104716.f 
b/gcc/testsuite/gfortran.dg/pr104716.f
new file mode 100644
index 00000000000..97f899a9201
--- /dev/null
+++ b/gcc/testsuite/gfortran.dg/pr104716.f
@@ -0,0 +1,31 @@
+! { dg-do compile }
+! { dg-options "-std=legacy -O2 -ftree-loop-distribution -fno-move-loop-stores 
-fno-tree-dominator-opts" }
+
+      SUBROUTINE FOO()
+      
+      COMMON /WORK/ C2(2, 2)
+      
+      DIMENSION D11(2)
+
+      EQUIVALENCE (D11(1), C2(1, 1))
+
+      DO 40 I = 1, 2
+         DO 30 J = 1, 2
+            ASSIGN 10 TO ILBL
+            IF (C2(J, I) .NE. 0.0) THEN
+               ASSIGN 20 TO ILBL
+            ENDIF
+            GO TO ILBL
+ 10         CONTINUE
+ 20         CONTINUE
+            C2(J, I) = C2(J, I) + 1
+ 30      CONTINUE
+ 40   CONTINUE
+
+      DO 50 I = 1, 2
+         PRINT 90, I
+ 50   CONTINUE
+      
+      RETURN
+ 90   FORMAT(I5)
+      END
diff --git a/gcc/tree-loop-distribution.cc b/gcc/tree-loop-distribution.cc
index 8ee40d88816..c2ca746e237 100644
--- a/gcc/tree-loop-distribution.cc
+++ b/gcc/tree-loop-distribution.cc
@@ -3237,6 +3237,20 @@ find_seed_stmts_for_distribution (class loop *loop, 
vec<gimple *> *work_list)
          work_list->truncate (0);
          break;
        }
+      edge_iterator ei;
+      edge e;
+      /* We cannot redirect abnormal edges so loop copying will fail.  */
+      FOR_EACH_EDGE (e, ei, bbs[i]->succs)
+       if (e->flags & EDGE_ABNORMAL)
+         {
+           if (dump_file && (dump_flags & TDF_DETAILS))
+             fprintf (dump_file, "loop %d is the source of abnormal edges.\n",
+                      loop->num);
+           work_list->truncate (0);
+           break;
+         }
+      if (e)
+       break;
       for (gphi_iterator gsi = gsi_start_phis (bbs[i]);
           !gsi_end_p (gsi); gsi_next (&gsi))
        {
-- 
2.34.1