OpenCores
URL https://opencores.org/ocsvn/openrisc/openrisc/trunk

Subversion Repositories openrisc

[/] [openrisc/] [trunk/] [gnu-dev/] [or1k-gcc/] [gcc/] [tree-vect-loop-manip.c] - Blame information for rev 746

Go to most recent revision | Details | Compare with Previous | View Log

Line No. Rev Author Line
1 684 jeremybenn
/* Vectorizer Specific Loop Manipulations
2
   Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2012
3
   Free Software Foundation, Inc.
4
   Contributed by Dorit Naishlos <dorit@il.ibm.com>
5
   and Ira Rosen <irar@il.ibm.com>
6
 
7
This file is part of GCC.
8
 
9
GCC is free software; you can redistribute it and/or modify it under
10
the terms of the GNU General Public License as published by the Free
11
Software Foundation; either version 3, or (at your option) any later
12
version.
13
 
14
GCC is distributed in the hope that it will be useful, but WITHOUT ANY
15
WARRANTY; without even the implied warranty of MERCHANTABILITY or
16
FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
17
for more details.
18
 
19
You should have received a copy of the GNU General Public License
20
along with GCC; see the file COPYING3.  If not see
21
<http://www.gnu.org/licenses/>.  */
22
 
23
#include "config.h"
24
#include "system.h"
25
#include "coretypes.h"
26
#include "tm.h"
27
#include "ggc.h"
28
#include "tree.h"
29
#include "basic-block.h"
30
#include "tree-pretty-print.h"
31
#include "gimple-pretty-print.h"
32
#include "tree-flow.h"
33
#include "tree-dump.h"
34
#include "cfgloop.h"
35
#include "cfglayout.h"
36
#include "diagnostic-core.h"
37
#include "tree-scalar-evolution.h"
38
#include "tree-vectorizer.h"
39
#include "langhooks.h"
40
 
41
/*************************************************************************
42
  Simple Loop Peeling Utilities
43
 
44
  Utilities to support loop peeling for vectorization purposes.
45
 *************************************************************************/
46
 
47
 
48
/* Renames the use *OP_P.  */
49
 
50
static void
51
rename_use_op (use_operand_p op_p)
52
{
53
  tree new_name;
54
 
55
  if (TREE_CODE (USE_FROM_PTR (op_p)) != SSA_NAME)
56
    return;
57
 
58
  new_name = get_current_def (USE_FROM_PTR (op_p));
59
 
60
  /* Something defined outside of the loop.  */
61
  if (!new_name)
62
    return;
63
 
64
  /* An ordinary ssa name defined in the loop.  */
65
 
66
  SET_USE (op_p, new_name);
67
}
68
 
69
 
70
/* Renames the variables in basic block BB.  */
71
 
72
void
73
rename_variables_in_bb (basic_block bb)
74
{
75
  gimple_stmt_iterator gsi;
76
  gimple stmt;
77
  use_operand_p use_p;
78
  ssa_op_iter iter;
79
  edge e;
80
  edge_iterator ei;
81
  struct loop *loop = bb->loop_father;
82
 
83
  for (gsi = gsi_start_bb (bb); !gsi_end_p (gsi); gsi_next (&gsi))
84
    {
85
      stmt = gsi_stmt (gsi);
86
      FOR_EACH_SSA_USE_OPERAND (use_p, stmt, iter, SSA_OP_ALL_USES)
87
        rename_use_op (use_p);
88
    }
89
 
90
  FOR_EACH_EDGE (e, ei, bb->succs)
91
    {
92
      if (!flow_bb_inside_loop_p (loop, e->dest))
93
        continue;
94
      for (gsi = gsi_start_phis (e->dest); !gsi_end_p (gsi); gsi_next (&gsi))
95
        rename_use_op (PHI_ARG_DEF_PTR_FROM_EDGE (gsi_stmt (gsi), e));
96
    }
97
}
98
 
99
 
100
/* Renames variables in new generated LOOP.  */
101
 
102
void
103
rename_variables_in_loop (struct loop *loop)
104
{
105
  unsigned i;
106
  basic_block *bbs;
107
 
108
  bbs = get_loop_body (loop);
109
 
110
  for (i = 0; i < loop->num_nodes; i++)
111
    rename_variables_in_bb (bbs[i]);
112
 
113
  free (bbs);
114
}
115
 
116
typedef struct
117
{
118
  tree from, to;
119
  basic_block bb;
120
} adjust_info;
121
 
122
DEF_VEC_O(adjust_info);
123
DEF_VEC_ALLOC_O_STACK(adjust_info);
124
#define VEC_adjust_info_stack_alloc(alloc) VEC_stack_alloc (adjust_info, alloc)
125
 
126
/* A stack of values to be adjusted in debug stmts.  We have to
127
   process them LIFO, so that the closest substitution applies.  If we
128
   processed them FIFO, without the stack, we might substitute uses
129
   with a PHI DEF that would soon become non-dominant, and when we got
130
   to the suitable one, it wouldn't have anything to substitute any
131
   more.  */
132
static VEC(adjust_info, stack) *adjust_vec;
133
 
134
/* Adjust any debug stmts that referenced AI->from values to use the
135
   loop-closed AI->to, if the references are dominated by AI->bb and
136
   not by the definition of AI->from.  */
137
 
138
static void
139
adjust_debug_stmts_now (adjust_info *ai)
140
{
141
  basic_block bbphi = ai->bb;
142
  tree orig_def = ai->from;
143
  tree new_def = ai->to;
144
  imm_use_iterator imm_iter;
145
  gimple stmt;
146
  basic_block bbdef = gimple_bb (SSA_NAME_DEF_STMT (orig_def));
147
 
148
  gcc_assert (dom_info_available_p (CDI_DOMINATORS));
149
 
150
  /* Adjust any debug stmts that held onto non-loop-closed
151
     references.  */
152
  FOR_EACH_IMM_USE_STMT (stmt, imm_iter, orig_def)
153
    {
154
      use_operand_p use_p;
155
      basic_block bbuse;
156
 
157
      if (!is_gimple_debug (stmt))
158
        continue;
159
 
160
      gcc_assert (gimple_debug_bind_p (stmt));
161
 
162
      bbuse = gimple_bb (stmt);
163
 
164
      if ((bbuse == bbphi
165
           || dominated_by_p (CDI_DOMINATORS, bbuse, bbphi))
166
          && !(bbuse == bbdef
167
               || dominated_by_p (CDI_DOMINATORS, bbuse, bbdef)))
168
        {
169
          if (new_def)
170
            FOR_EACH_IMM_USE_ON_STMT (use_p, imm_iter)
171
              SET_USE (use_p, new_def);
172
          else
173
            {
174
              gimple_debug_bind_reset_value (stmt);
175
              update_stmt (stmt);
176
            }
177
        }
178
    }
179
}
180
 
181
/* Adjust debug stmts as scheduled before.  */
182
 
183
static void
184
adjust_vec_debug_stmts (void)
185
{
186
  if (!MAY_HAVE_DEBUG_STMTS)
187
    return;
188
 
189
  gcc_assert (adjust_vec);
190
 
191
  while (!VEC_empty (adjust_info, adjust_vec))
192
    {
193
      adjust_debug_stmts_now (VEC_last (adjust_info, adjust_vec));
194
      VEC_pop (adjust_info, adjust_vec);
195
    }
196
 
197
  VEC_free (adjust_info, stack, adjust_vec);
198
}
199
 
200
/* Adjust any debug stmts that referenced FROM values to use the
201
   loop-closed TO, if the references are dominated by BB and not by
202
   the definition of FROM.  If adjust_vec is non-NULL, adjustments
203
   will be postponed until adjust_vec_debug_stmts is called.  */
204
 
205
static void
206
adjust_debug_stmts (tree from, tree to, basic_block bb)
207
{
208
  adjust_info ai;
209
 
210
  if (MAY_HAVE_DEBUG_STMTS && TREE_CODE (from) == SSA_NAME
211
      && SSA_NAME_VAR (from) != gimple_vop (cfun))
212
    {
213
      ai.from = from;
214
      ai.to = to;
215
      ai.bb = bb;
216
 
217
      if (adjust_vec)
218
        VEC_safe_push (adjust_info, stack, adjust_vec, &ai);
219
      else
220
        adjust_debug_stmts_now (&ai);
221
    }
222
}
223
 
224
/* Change E's phi arg in UPDATE_PHI to NEW_DEF, and record information
225
   to adjust any debug stmts that referenced the old phi arg,
226
   presumably non-loop-closed references left over from other
227
   transformations.  */
228
 
229
static void
230
adjust_phi_and_debug_stmts (gimple update_phi, edge e, tree new_def)
231
{
232
  tree orig_def = PHI_ARG_DEF_FROM_EDGE (update_phi, e);
233
 
234
  SET_PHI_ARG_DEF (update_phi, e->dest_idx, new_def);
235
 
236
  if (MAY_HAVE_DEBUG_STMTS)
237
    adjust_debug_stmts (orig_def, PHI_RESULT (update_phi),
238
                        gimple_bb (update_phi));
239
}
240
 
241
 
242
/* Update the PHI nodes of NEW_LOOP.
243
 
244
   NEW_LOOP is a duplicate of ORIG_LOOP.
245
   AFTER indicates whether NEW_LOOP executes before or after ORIG_LOOP:
246
   AFTER is true if NEW_LOOP executes after ORIG_LOOP, and false if it
247
   executes before it.  */
248
 
249
static void
250
slpeel_update_phis_for_duplicate_loop (struct loop *orig_loop,
251
                                       struct loop *new_loop, bool after)
252
{
253
  tree new_ssa_name;
254
  gimple phi_new, phi_orig;
255
  tree def;
256
  edge orig_loop_latch = loop_latch_edge (orig_loop);
257
  edge orig_entry_e = loop_preheader_edge (orig_loop);
258
  edge new_loop_exit_e = single_exit (new_loop);
259
  edge new_loop_entry_e = loop_preheader_edge (new_loop);
260
  edge entry_arg_e = (after ? orig_loop_latch : orig_entry_e);
261
  gimple_stmt_iterator gsi_new, gsi_orig;
262
 
263
  /*
264
     step 1. For each loop-header-phi:
265
             Add the first phi argument for the phi in NEW_LOOP
266
            (the one associated with the entry of NEW_LOOP)
267
 
268
     step 2. For each loop-header-phi:
269
             Add the second phi argument for the phi in NEW_LOOP
270
            (the one associated with the latch of NEW_LOOP)
271
 
272
     step 3. Update the phis in the successor block of NEW_LOOP.
273
 
274
        case 1: NEW_LOOP was placed before ORIG_LOOP:
275
                The successor block of NEW_LOOP is the header of ORIG_LOOP.
276
                Updating the phis in the successor block can therefore be done
277
                along with the scanning of the loop header phis, because the
278
                header blocks of ORIG_LOOP and NEW_LOOP have exactly the same
279
                phi nodes, organized in the same order.
280
 
281
        case 2: NEW_LOOP was placed after ORIG_LOOP:
282
                The successor block of NEW_LOOP is the original exit block of
283
                ORIG_LOOP - the phis to be updated are the loop-closed-ssa phis.
284
                We postpone updating these phis to a later stage (when
285
                loop guards are added).
286
   */
287
 
288
 
289
  /* Scan the phis in the headers of the old and new loops
290
     (they are organized in exactly the same order).  */
291
 
292
  for (gsi_new = gsi_start_phis (new_loop->header),
293
       gsi_orig = gsi_start_phis (orig_loop->header);
294
       !gsi_end_p (gsi_new) && !gsi_end_p (gsi_orig);
295
       gsi_next (&gsi_new), gsi_next (&gsi_orig))
296
    {
297
      source_location locus;
298
      phi_new = gsi_stmt (gsi_new);
299
      phi_orig = gsi_stmt (gsi_orig);
300
 
301
      /* step 1.  */
302
      def = PHI_ARG_DEF_FROM_EDGE (phi_orig, entry_arg_e);
303
      locus = gimple_phi_arg_location_from_edge (phi_orig, entry_arg_e);
304
      add_phi_arg (phi_new, def, new_loop_entry_e, locus);
305
 
306
      /* step 2.  */
307
      def = PHI_ARG_DEF_FROM_EDGE (phi_orig, orig_loop_latch);
308
      locus = gimple_phi_arg_location_from_edge (phi_orig, orig_loop_latch);
309
      if (TREE_CODE (def) != SSA_NAME)
310
        continue;
311
 
312
      new_ssa_name = get_current_def (def);
313
      if (!new_ssa_name)
314
        {
315
          /* This only happens if there are no definitions
316
             inside the loop. use the phi_result in this case.  */
317
          new_ssa_name = PHI_RESULT (phi_new);
318
        }
319
 
320
      /* An ordinary ssa name defined in the loop.  */
321
      add_phi_arg (phi_new, new_ssa_name, loop_latch_edge (new_loop), locus);
322
 
323
      /* Drop any debug references outside the loop, if they would
324
         become ill-formed SSA.  */
325
      adjust_debug_stmts (def, NULL, single_exit (orig_loop)->dest);
326
 
327
      /* step 3 (case 1).  */
328
      if (!after)
329
        {
330
          gcc_assert (new_loop_exit_e == orig_entry_e);
331
          adjust_phi_and_debug_stmts (phi_orig, new_loop_exit_e, new_ssa_name);
332
        }
333
    }
334
}
335
 
336
 
337
/* Update PHI nodes for a guard of the LOOP.
338
 
339
   Input:
340
   - LOOP, GUARD_EDGE: LOOP is a loop for which we added guard code that
341
        controls whether LOOP is to be executed.  GUARD_EDGE is the edge that
342
        originates from the guard-bb, skips LOOP and reaches the (unique) exit
343
        bb of LOOP.  This loop-exit-bb is an empty bb with one successor.
344
        We denote this bb NEW_MERGE_BB because before the guard code was added
345
        it had a single predecessor (the LOOP header), and now it became a merge
346
        point of two paths - the path that ends with the LOOP exit-edge, and
347
        the path that ends with GUARD_EDGE.
348
   - NEW_EXIT_BB: New basic block that is added by this function between LOOP
349
        and NEW_MERGE_BB. It is used to place loop-closed-ssa-form exit-phis.
350
 
351
   ===> The CFG before the guard-code was added:
352
        LOOP_header_bb:
353
          loop_body
354
          if (exit_loop) goto update_bb
355
          else           goto LOOP_header_bb
356
        update_bb:
357
 
358
   ==> The CFG after the guard-code was added:
359
        guard_bb:
360
          if (LOOP_guard_condition) goto new_merge_bb
361
          else                      goto LOOP_header_bb
362
        LOOP_header_bb:
363
          loop_body
364
          if (exit_loop_condition) goto new_merge_bb
365
          else                     goto LOOP_header_bb
366
        new_merge_bb:
367
          goto update_bb
368
        update_bb:
369
 
370
   ==> The CFG after this function:
371
        guard_bb:
372
          if (LOOP_guard_condition) goto new_merge_bb
373
          else                      goto LOOP_header_bb
374
        LOOP_header_bb:
375
          loop_body
376
          if (exit_loop_condition) goto new_exit_bb
377
          else                     goto LOOP_header_bb
378
        new_exit_bb:
379
        new_merge_bb:
380
          goto update_bb
381
        update_bb:
382
 
383
   This function:
384
   1. creates and updates the relevant phi nodes to account for the new
385
      incoming edge (GUARD_EDGE) into NEW_MERGE_BB. This involves:
386
      1.1. Create phi nodes at NEW_MERGE_BB.
387
      1.2. Update the phi nodes at the successor of NEW_MERGE_BB (denoted
388
           UPDATE_BB).  UPDATE_BB was the exit-bb of LOOP before NEW_MERGE_BB
389
   2. preserves loop-closed-ssa-form by creating the required phi nodes
390
      at the exit of LOOP (i.e, in NEW_EXIT_BB).
391
 
392
   There are two flavors to this function:
393
 
394
   slpeel_update_phi_nodes_for_guard1:
395
     Here the guard controls whether we enter or skip LOOP, where LOOP is a
396
     prolog_loop (loop1 below), and the new phis created in NEW_MERGE_BB are
397
     for variables that have phis in the loop header.
398
 
399
   slpeel_update_phi_nodes_for_guard2:
400
     Here the guard controls whether we enter or skip LOOP, where LOOP is an
401
     epilog_loop (loop2 below), and the new phis created in NEW_MERGE_BB are
402
     for variables that have phis in the loop exit.
403
 
404
   I.E., the overall structure is:
405
 
406
        loop1_preheader_bb:
407
                guard1 (goto loop1/merge1_bb)
408
        loop1
409
        loop1_exit_bb:
410
                guard2 (goto merge1_bb/merge2_bb)
411
        merge1_bb
412
        loop2
413
        loop2_exit_bb
414
        merge2_bb
415
        next_bb
416
 
417
   slpeel_update_phi_nodes_for_guard1 takes care of creating phis in
418
   loop1_exit_bb and merge1_bb. These are entry phis (phis for the vars
419
   that have phis in loop1->header).
420
 
421
   slpeel_update_phi_nodes_for_guard2 takes care of creating phis in
422
   loop2_exit_bb and merge2_bb. These are exit phis (phis for the vars
423
   that have phis in next_bb). It also adds some of these phis to
424
   loop1_exit_bb.
425
 
426
   slpeel_update_phi_nodes_for_guard1 is always called before
427
   slpeel_update_phi_nodes_for_guard2. They are both needed in order
428
   to create correct data-flow and loop-closed-ssa-form.
429
 
430
   Generally slpeel_update_phi_nodes_for_guard1 creates phis for variables
431
   that change between iterations of a loop (and therefore have a phi-node
432
   at the loop entry), whereas slpeel_update_phi_nodes_for_guard2 creates
433
   phis for variables that are used out of the loop (and therefore have
434
   loop-closed exit phis). Some variables may be both updated between
435
   iterations and used after the loop. This is why in loop1_exit_bb we
436
   may need both entry_phis (created by slpeel_update_phi_nodes_for_guard1)
437
   and exit phis (created by slpeel_update_phi_nodes_for_guard2).
438
 
439
   - IS_NEW_LOOP: if IS_NEW_LOOP is true, then LOOP is a newly created copy of
440
     an original loop. i.e., we have:
441
 
442
           orig_loop
443
           guard_bb (goto LOOP/new_merge)
444
           new_loop <-- LOOP
445
           new_exit
446
           new_merge
447
           next_bb
448
 
449
     If IS_NEW_LOOP is false, then LOOP is an original loop, in which case we
450
     have:
451
 
452
           new_loop
453
           guard_bb (goto LOOP/new_merge)
454
           orig_loop <-- LOOP
455
           new_exit
456
           new_merge
457
           next_bb
458
 
459
     The SSA names defined in the original loop have a current
460
     reaching definition that that records the corresponding new
461
     ssa-name used in the new duplicated loop copy.
462
  */
463
 
464
/* Function slpeel_update_phi_nodes_for_guard1
465
 
466
   Input:
467
   - GUARD_EDGE, LOOP, IS_NEW_LOOP, NEW_EXIT_BB - as explained above.
468
   - DEFS - a bitmap of ssa names to mark new names for which we recorded
469
            information.
470
 
471
   In the context of the overall structure, we have:
472
 
473
        loop1_preheader_bb:
474
                guard1 (goto loop1/merge1_bb)
475
LOOP->  loop1
476
        loop1_exit_bb:
477
                guard2 (goto merge1_bb/merge2_bb)
478
        merge1_bb
479
        loop2
480
        loop2_exit_bb
481
        merge2_bb
482
        next_bb
483
 
484
   For each name updated between loop iterations (i.e - for each name that has
485
   an entry (loop-header) phi in LOOP) we create a new phi in:
486
   1. merge1_bb (to account for the edge from guard1)
487
   2. loop1_exit_bb (an exit-phi to keep LOOP in loop-closed form)
488
*/
489
 
490
static void
491
slpeel_update_phi_nodes_for_guard1 (edge guard_edge, struct loop *loop,
492
                                    bool is_new_loop, basic_block *new_exit_bb,
493
                                    bitmap *defs)
494
{
495
  gimple orig_phi, new_phi;
496
  gimple update_phi, update_phi2;
497
  tree guard_arg, loop_arg;
498
  basic_block new_merge_bb = guard_edge->dest;
499
  edge e = EDGE_SUCC (new_merge_bb, 0);
500
  basic_block update_bb = e->dest;
501
  basic_block orig_bb = loop->header;
502
  edge new_exit_e;
503
  tree current_new_name;
504
  gimple_stmt_iterator gsi_orig, gsi_update;
505
 
506
  /* Create new bb between loop and new_merge_bb.  */
507
  *new_exit_bb = split_edge (single_exit (loop));
508
 
509
  new_exit_e = EDGE_SUCC (*new_exit_bb, 0);
510
 
511
  for (gsi_orig = gsi_start_phis (orig_bb),
512
       gsi_update = gsi_start_phis (update_bb);
513
       !gsi_end_p (gsi_orig) && !gsi_end_p (gsi_update);
514
       gsi_next (&gsi_orig), gsi_next (&gsi_update))
515
    {
516
      source_location loop_locus, guard_locus;
517
      orig_phi = gsi_stmt (gsi_orig);
518
      update_phi = gsi_stmt (gsi_update);
519
 
520
      /** 1. Handle new-merge-point phis  **/
521
 
522
      /* 1.1. Generate new phi node in NEW_MERGE_BB:  */
523
      new_phi = create_phi_node (SSA_NAME_VAR (PHI_RESULT (orig_phi)),
524
                                 new_merge_bb);
525
 
526
      /* 1.2. NEW_MERGE_BB has two incoming edges: GUARD_EDGE and the exit-edge
527
            of LOOP. Set the two phi args in NEW_PHI for these edges:  */
528
      loop_arg = PHI_ARG_DEF_FROM_EDGE (orig_phi, EDGE_SUCC (loop->latch, 0));
529
      loop_locus = gimple_phi_arg_location_from_edge (orig_phi,
530
                                                      EDGE_SUCC (loop->latch,
531
                                                                 0));
532
      guard_arg = PHI_ARG_DEF_FROM_EDGE (orig_phi, loop_preheader_edge (loop));
533
      guard_locus
534
        = gimple_phi_arg_location_from_edge (orig_phi,
535
                                             loop_preheader_edge (loop));
536
 
537
      add_phi_arg (new_phi, loop_arg, new_exit_e, loop_locus);
538
      add_phi_arg (new_phi, guard_arg, guard_edge, guard_locus);
539
 
540
      /* 1.3. Update phi in successor block.  */
541
      gcc_assert (PHI_ARG_DEF_FROM_EDGE (update_phi, e) == loop_arg
542
                  || PHI_ARG_DEF_FROM_EDGE (update_phi, e) == guard_arg);
543
      adjust_phi_and_debug_stmts (update_phi, e, PHI_RESULT (new_phi));
544
      update_phi2 = new_phi;
545
 
546
 
547
      /** 2. Handle loop-closed-ssa-form phis  **/
548
 
549
      if (!is_gimple_reg (PHI_RESULT (orig_phi)))
550
        continue;
551
 
552
      /* 2.1. Generate new phi node in NEW_EXIT_BB:  */
553
      new_phi = create_phi_node (SSA_NAME_VAR (PHI_RESULT (orig_phi)),
554
                                 *new_exit_bb);
555
 
556
      /* 2.2. NEW_EXIT_BB has one incoming edge: the exit-edge of the loop.  */
557
      add_phi_arg (new_phi, loop_arg, single_exit (loop), loop_locus);
558
 
559
      /* 2.3. Update phi in successor of NEW_EXIT_BB:  */
560
      gcc_assert (PHI_ARG_DEF_FROM_EDGE (update_phi2, new_exit_e) == loop_arg);
561
      adjust_phi_and_debug_stmts (update_phi2, new_exit_e,
562
                                  PHI_RESULT (new_phi));
563
 
564
      /* 2.4. Record the newly created name with set_current_def.
565
         We want to find a name such that
566
                name = get_current_def (orig_loop_name)
567
         and to set its current definition as follows:
568
                set_current_def (name, new_phi_name)
569
 
570
         If LOOP is a new loop then loop_arg is already the name we're
571
         looking for. If LOOP is the original loop, then loop_arg is
572
         the orig_loop_name and the relevant name is recorded in its
573
         current reaching definition.  */
574
      if (is_new_loop)
575
        current_new_name = loop_arg;
576
      else
577
        {
578
          current_new_name = get_current_def (loop_arg);
579
          /* current_def is not available only if the variable does not
580
             change inside the loop, in which case we also don't care
581
             about recording a current_def for it because we won't be
582
             trying to create loop-exit-phis for it.  */
583
          if (!current_new_name)
584
            continue;
585
        }
586
      gcc_assert (get_current_def (current_new_name) == NULL_TREE);
587
 
588
      set_current_def (current_new_name, PHI_RESULT (new_phi));
589
      bitmap_set_bit (*defs, SSA_NAME_VERSION (current_new_name));
590
    }
591
}
592
 
593
 
594
/* Function slpeel_update_phi_nodes_for_guard2
595
 
596
   Input:
597
   - GUARD_EDGE, LOOP, IS_NEW_LOOP, NEW_EXIT_BB - as explained above.
598
 
599
   In the context of the overall structure, we have:
600
 
601
        loop1_preheader_bb:
602
                guard1 (goto loop1/merge1_bb)
603
        loop1
604
        loop1_exit_bb:
605
                guard2 (goto merge1_bb/merge2_bb)
606
        merge1_bb
607
LOOP->  loop2
608
        loop2_exit_bb
609
        merge2_bb
610
        next_bb
611
 
612
   For each name used out side the loop (i.e - for each name that has an exit
613
   phi in next_bb) we create a new phi in:
614
   1. merge2_bb (to account for the edge from guard_bb)
615
   2. loop2_exit_bb (an exit-phi to keep LOOP in loop-closed form)
616
   3. guard2 bb (an exit phi to keep the preceding loop in loop-closed form),
617
      if needed (if it wasn't handled by slpeel_update_phis_nodes_for_phi1).
618
*/
619
 
620
static void
621
slpeel_update_phi_nodes_for_guard2 (edge guard_edge, struct loop *loop,
622
                                    bool is_new_loop, basic_block *new_exit_bb)
623
{
624
  gimple orig_phi, new_phi;
625
  gimple update_phi, update_phi2;
626
  tree guard_arg, loop_arg;
627
  basic_block new_merge_bb = guard_edge->dest;
628
  edge e = EDGE_SUCC (new_merge_bb, 0);
629
  basic_block update_bb = e->dest;
630
  edge new_exit_e;
631
  tree orig_def, orig_def_new_name;
632
  tree new_name, new_name2;
633
  tree arg;
634
  gimple_stmt_iterator gsi;
635
 
636
  /* Create new bb between loop and new_merge_bb.  */
637
  *new_exit_bb = split_edge (single_exit (loop));
638
 
639
  new_exit_e = EDGE_SUCC (*new_exit_bb, 0);
640
 
641
  for (gsi = gsi_start_phis (update_bb); !gsi_end_p (gsi); gsi_next (&gsi))
642
    {
643
      update_phi = gsi_stmt (gsi);
644
      orig_phi = update_phi;
645
      orig_def = PHI_ARG_DEF_FROM_EDGE (orig_phi, e);
646
      /* This loop-closed-phi actually doesn't represent a use
647
         out of the loop - the phi arg is a constant.  */
648
      if (TREE_CODE (orig_def) != SSA_NAME)
649
        continue;
650
      orig_def_new_name = get_current_def (orig_def);
651
      arg = NULL_TREE;
652
 
653
      /** 1. Handle new-merge-point phis  **/
654
 
655
      /* 1.1. Generate new phi node in NEW_MERGE_BB:  */
656
      new_phi = create_phi_node (SSA_NAME_VAR (PHI_RESULT (orig_phi)),
657
                                 new_merge_bb);
658
 
659
      /* 1.2. NEW_MERGE_BB has two incoming edges: GUARD_EDGE and the exit-edge
660
            of LOOP. Set the two PHI args in NEW_PHI for these edges:  */
661
      new_name = orig_def;
662
      new_name2 = NULL_TREE;
663
      if (orig_def_new_name)
664
        {
665
          new_name = orig_def_new_name;
666
          /* Some variables have both loop-entry-phis and loop-exit-phis.
667
             Such variables were given yet newer names by phis placed in
668
             guard_bb by slpeel_update_phi_nodes_for_guard1. I.e:
669
             new_name2 = get_current_def (get_current_def (orig_name)).  */
670
          new_name2 = get_current_def (new_name);
671
        }
672
 
673
      if (is_new_loop)
674
        {
675
          guard_arg = orig_def;
676
          loop_arg = new_name;
677
        }
678
      else
679
        {
680
          guard_arg = new_name;
681
          loop_arg = orig_def;
682
        }
683
      if (new_name2)
684
        guard_arg = new_name2;
685
 
686
      add_phi_arg (new_phi, loop_arg, new_exit_e, UNKNOWN_LOCATION);
687
      add_phi_arg (new_phi, guard_arg, guard_edge, UNKNOWN_LOCATION);
688
 
689
      /* 1.3. Update phi in successor block.  */
690
      gcc_assert (PHI_ARG_DEF_FROM_EDGE (update_phi, e) == orig_def);
691
      adjust_phi_and_debug_stmts (update_phi, e, PHI_RESULT (new_phi));
692
      update_phi2 = new_phi;
693
 
694
 
695
      /** 2. Handle loop-closed-ssa-form phis  **/
696
 
697
      /* 2.1. Generate new phi node in NEW_EXIT_BB:  */
698
      new_phi = create_phi_node (SSA_NAME_VAR (PHI_RESULT (orig_phi)),
699
                                 *new_exit_bb);
700
 
701
      /* 2.2. NEW_EXIT_BB has one incoming edge: the exit-edge of the loop.  */
702
      add_phi_arg (new_phi, loop_arg, single_exit (loop), UNKNOWN_LOCATION);
703
 
704
      /* 2.3. Update phi in successor of NEW_EXIT_BB:  */
705
      gcc_assert (PHI_ARG_DEF_FROM_EDGE (update_phi2, new_exit_e) == loop_arg);
706
      adjust_phi_and_debug_stmts (update_phi2, new_exit_e,
707
                                  PHI_RESULT (new_phi));
708
 
709
 
710
      /** 3. Handle loop-closed-ssa-form phis for first loop  **/
711
 
712
      /* 3.1. Find the relevant names that need an exit-phi in
713
         GUARD_BB, i.e. names for which
714
         slpeel_update_phi_nodes_for_guard1 had not already created a
715
         phi node. This is the case for names that are used outside
716
         the loop (and therefore need an exit phi) but are not updated
717
         across loop iterations (and therefore don't have a
718
         loop-header-phi).
719
 
720
         slpeel_update_phi_nodes_for_guard1 is responsible for
721
         creating loop-exit phis in GUARD_BB for names that have a
722
         loop-header-phi.  When such a phi is created we also record
723
         the new name in its current definition.  If this new name
724
         exists, then guard_arg was set to this new name (see 1.2
725
         above).  Therefore, if guard_arg is not this new name, this
726
         is an indication that an exit-phi in GUARD_BB was not yet
727
         created, so we take care of it here.  */
728
      if (guard_arg == new_name2)
729
        continue;
730
      arg = guard_arg;
731
 
732
      /* 3.2. Generate new phi node in GUARD_BB:  */
733
      new_phi = create_phi_node (SSA_NAME_VAR (PHI_RESULT (orig_phi)),
734
                                 guard_edge->src);
735
 
736
      /* 3.3. GUARD_BB has one incoming edge:  */
737
      gcc_assert (EDGE_COUNT (guard_edge->src->preds) == 1);
738
      add_phi_arg (new_phi, arg, EDGE_PRED (guard_edge->src, 0),
739
                   UNKNOWN_LOCATION);
740
 
741
      /* 3.4. Update phi in successor of GUARD_BB:  */
742
      gcc_assert (PHI_ARG_DEF_FROM_EDGE (update_phi2, guard_edge)
743
                                                                == guard_arg);
744
      adjust_phi_and_debug_stmts (update_phi2, guard_edge,
745
                                  PHI_RESULT (new_phi));
746
    }
747
}
748
 
749
 
750
/* Make the LOOP iterate NITERS times. This is done by adding a new IV
751
   that starts at zero, increases by one and its limit is NITERS.
752
 
753
   Assumption: the exit-condition of LOOP is the last stmt in the loop.  */
754
 
755
void
756
slpeel_make_loop_iterate_ntimes (struct loop *loop, tree niters)
757
{
758
  tree indx_before_incr, indx_after_incr;
759
  gimple cond_stmt;
760
  gimple orig_cond;
761
  edge exit_edge = single_exit (loop);
762
  gimple_stmt_iterator loop_cond_gsi;
763
  gimple_stmt_iterator incr_gsi;
764
  bool insert_after;
765
  tree init = build_int_cst (TREE_TYPE (niters), 0);
766
  tree step = build_int_cst (TREE_TYPE (niters), 1);
767
  LOC loop_loc;
768
  enum tree_code code;
769
 
770
  orig_cond = get_loop_exit_condition (loop);
771
  gcc_assert (orig_cond);
772
  loop_cond_gsi = gsi_for_stmt (orig_cond);
773
 
774
  standard_iv_increment_position (loop, &incr_gsi, &insert_after);
775
  create_iv (init, step, NULL_TREE, loop,
776
             &incr_gsi, insert_after, &indx_before_incr, &indx_after_incr);
777
 
778
  indx_after_incr = force_gimple_operand_gsi (&loop_cond_gsi, indx_after_incr,
779
                                              true, NULL_TREE, true,
780
                                              GSI_SAME_STMT);
781
  niters = force_gimple_operand_gsi (&loop_cond_gsi, niters, true, NULL_TREE,
782
                                     true, GSI_SAME_STMT);
783
 
784
  code = (exit_edge->flags & EDGE_TRUE_VALUE) ? GE_EXPR : LT_EXPR;
785
  cond_stmt = gimple_build_cond (code, indx_after_incr, niters, NULL_TREE,
786
                                 NULL_TREE);
787
 
788
  gsi_insert_before (&loop_cond_gsi, cond_stmt, GSI_SAME_STMT);
789
 
790
  /* Remove old loop exit test:  */
791
  gsi_remove (&loop_cond_gsi, true);
792
 
793
  loop_loc = find_loop_location (loop);
794
  if (dump_file && (dump_flags & TDF_DETAILS))
795
    {
796
      if (loop_loc != UNKNOWN_LOC)
797
        fprintf (dump_file, "\nloop at %s:%d: ",
798
                 LOC_FILE (loop_loc), LOC_LINE (loop_loc));
799
      print_gimple_stmt (dump_file, cond_stmt, 0, TDF_SLIM);
800
    }
801
 
802
  loop->nb_iterations = niters;
803
}
804
 
805
 
806
/* Given LOOP this function generates a new copy of it and puts it
807
   on E which is either the entry or exit of LOOP.  */
808
 
809
struct loop *
810
slpeel_tree_duplicate_loop_to_edge_cfg (struct loop *loop, edge e)
811
{
812
  struct loop *new_loop;
813
  basic_block *new_bbs, *bbs;
814
  bool at_exit;
815
  bool was_imm_dom;
816
  basic_block exit_dest;
817
  gimple phi;
818
  tree phi_arg;
819
  edge exit, new_exit;
820
  gimple_stmt_iterator gsi;
821
 
822
  at_exit = (e == single_exit (loop));
823
  if (!at_exit && e != loop_preheader_edge (loop))
824
    return NULL;
825
 
826
  bbs = get_loop_body (loop);
827
 
828
  /* Check whether duplication is possible.  */
829
  if (!can_copy_bbs_p (bbs, loop->num_nodes))
830
    {
831
      free (bbs);
832
      return NULL;
833
    }
834
 
835
  /* Generate new loop structure.  */
836
  new_loop = duplicate_loop (loop, loop_outer (loop));
837
  if (!new_loop)
838
    {
839
      free (bbs);
840
      return NULL;
841
    }
842
 
843
  exit_dest = single_exit (loop)->dest;
844
  was_imm_dom = (get_immediate_dominator (CDI_DOMINATORS,
845
                                          exit_dest) == loop->header ?
846
                 true : false);
847
 
848
  new_bbs = XNEWVEC (basic_block, loop->num_nodes);
849
 
850
  exit = single_exit (loop);
851
  copy_bbs (bbs, loop->num_nodes, new_bbs,
852
            &exit, 1, &new_exit, NULL,
853
            e->src);
854
 
855
  /* Duplicating phi args at exit bbs as coming
856
     also from exit of duplicated loop.  */
857
  for (gsi = gsi_start_phis (exit_dest); !gsi_end_p (gsi); gsi_next (&gsi))
858
    {
859
      phi = gsi_stmt (gsi);
860
      phi_arg = PHI_ARG_DEF_FROM_EDGE (phi, single_exit (loop));
861
      if (phi_arg)
862
        {
863
          edge new_loop_exit_edge;
864
          source_location locus;
865
 
866
          locus = gimple_phi_arg_location_from_edge (phi, single_exit (loop));
867
          if (EDGE_SUCC (new_loop->header, 0)->dest == new_loop->latch)
868
            new_loop_exit_edge = EDGE_SUCC (new_loop->header, 1);
869
          else
870
            new_loop_exit_edge = EDGE_SUCC (new_loop->header, 0);
871
 
872
          add_phi_arg (phi, phi_arg, new_loop_exit_edge, locus);
873
        }
874
    }
875
 
876
  if (at_exit) /* Add the loop copy at exit.  */
877
    {
878
      redirect_edge_and_branch_force (e, new_loop->header);
879
      PENDING_STMT (e) = NULL;
880
      set_immediate_dominator (CDI_DOMINATORS, new_loop->header, e->src);
881
      if (was_imm_dom)
882
        set_immediate_dominator (CDI_DOMINATORS, exit_dest, new_loop->header);
883
    }
884
  else /* Add the copy at entry.  */
885
    {
886
      edge new_exit_e;
887
      edge entry_e = loop_preheader_edge (loop);
888
      basic_block preheader = entry_e->src;
889
 
890
      if (!flow_bb_inside_loop_p (new_loop,
891
                                  EDGE_SUCC (new_loop->header, 0)->dest))
892
        new_exit_e = EDGE_SUCC (new_loop->header, 0);
893
      else
894
        new_exit_e = EDGE_SUCC (new_loop->header, 1);
895
 
896
      redirect_edge_and_branch_force (new_exit_e, loop->header);
897
      PENDING_STMT (new_exit_e) = NULL;
898
      set_immediate_dominator (CDI_DOMINATORS, loop->header,
899
                               new_exit_e->src);
900
 
901
      /* We have to add phi args to the loop->header here as coming
902
         from new_exit_e edge.  */
903
      for (gsi = gsi_start_phis (loop->header);
904
           !gsi_end_p (gsi);
905
           gsi_next (&gsi))
906
        {
907
          phi = gsi_stmt (gsi);
908
          phi_arg = PHI_ARG_DEF_FROM_EDGE (phi, entry_e);
909
          if (phi_arg)
910
            add_phi_arg (phi, phi_arg, new_exit_e,
911
                         gimple_phi_arg_location_from_edge (phi, entry_e));
912
        }
913
 
914
      redirect_edge_and_branch_force (entry_e, new_loop->header);
915
      PENDING_STMT (entry_e) = NULL;
916
      set_immediate_dominator (CDI_DOMINATORS, new_loop->header, preheader);
917
    }
918
 
919
  free (new_bbs);
920
  free (bbs);
921
 
922
  return new_loop;
923
}
924
 
925
 
926
/* Given the condition statement COND, put it as the last statement
927
   of GUARD_BB; EXIT_BB is the basic block to skip the loop;
928
   Assumes that this is the single exit of the guarded loop.
929
   Returns the skip edge, inserts new stmts on the COND_EXPR_STMT_LIST.  */
930
 
931
static edge
932
slpeel_add_loop_guard (basic_block guard_bb, tree cond,
933
                       gimple_seq cond_expr_stmt_list,
934
                       basic_block exit_bb, basic_block dom_bb)
935
{
936
  gimple_stmt_iterator gsi;
937
  edge new_e, enter_e;
938
  gimple cond_stmt;
939
  gimple_seq gimplify_stmt_list = NULL;
940
 
941
  enter_e = EDGE_SUCC (guard_bb, 0);
942
  enter_e->flags &= ~EDGE_FALLTHRU;
943
  enter_e->flags |= EDGE_FALSE_VALUE;
944
  gsi = gsi_last_bb (guard_bb);
945
 
946
  cond = force_gimple_operand (cond, &gimplify_stmt_list, true, NULL_TREE);
947
  if (gimplify_stmt_list)
948
    gimple_seq_add_seq (&cond_expr_stmt_list, gimplify_stmt_list);
949
  cond_stmt = gimple_build_cond (NE_EXPR,
950
                                 cond, build_int_cst (TREE_TYPE (cond), 0),
951
                                 NULL_TREE, NULL_TREE);
952
  if (cond_expr_stmt_list)
953
    gsi_insert_seq_after (&gsi, cond_expr_stmt_list, GSI_NEW_STMT);
954
 
955
  gsi = gsi_last_bb (guard_bb);
956
  gsi_insert_after (&gsi, cond_stmt, GSI_NEW_STMT);
957
 
958
  /* Add new edge to connect guard block to the merge/loop-exit block.  */
959
  new_e = make_edge (guard_bb, exit_bb, EDGE_TRUE_VALUE);
960
  set_immediate_dominator (CDI_DOMINATORS, exit_bb, dom_bb);
961
  return new_e;
962
}
963
 
964
 
965
/* This function verifies that the following restrictions apply to LOOP:
966
   (1) it is innermost
967
   (2) it consists of exactly 2 basic blocks - header, and an empty latch.
968
   (3) it is single entry, single exit
969
   (4) its exit condition is the last stmt in the header
970
   (5) E is the entry/exit edge of LOOP.
971
 */
972
 
973
bool
974
slpeel_can_duplicate_loop_p (const struct loop *loop, const_edge e)
975
{
976
  edge exit_e = single_exit (loop);
977
  edge entry_e = loop_preheader_edge (loop);
978
  gimple orig_cond = get_loop_exit_condition (loop);
979
  gimple_stmt_iterator loop_exit_gsi = gsi_last_bb (exit_e->src);
980
 
981
  if (need_ssa_update_p (cfun))
982
    return false;
983
 
984
  if (loop->inner
985
      /* All loops have an outer scope; the only case loop->outer is NULL is for
986
         the function itself.  */
987
      || !loop_outer (loop)
988
      || loop->num_nodes != 2
989
      || !empty_block_p (loop->latch)
990
      || !single_exit (loop)
991
      /* Verify that new loop exit condition can be trivially modified.  */
992
      || (!orig_cond || orig_cond != gsi_stmt (loop_exit_gsi))
993
      || (e != exit_e && e != entry_e))
994
    return false;
995
 
996
  return true;
997
}
998
 
999
#ifdef ENABLE_CHECKING
1000
static void
1001
slpeel_verify_cfg_after_peeling (struct loop *first_loop,
1002
                                 struct loop *second_loop)
1003
{
1004
  basic_block loop1_exit_bb = single_exit (first_loop)->dest;
1005
  basic_block loop2_entry_bb = loop_preheader_edge (second_loop)->src;
1006
  basic_block loop1_entry_bb = loop_preheader_edge (first_loop)->src;
1007
 
1008
  /* A guard that controls whether the second_loop is to be executed or skipped
1009
     is placed in first_loop->exit.  first_loop->exit therefore has two
1010
     successors - one is the preheader of second_loop, and the other is a bb
1011
     after second_loop.
1012
   */
1013
  gcc_assert (EDGE_COUNT (loop1_exit_bb->succs) == 2);
1014
 
1015
  /* 1. Verify that one of the successors of first_loop->exit is the preheader
1016
        of second_loop.  */
1017
 
1018
  /* The preheader of new_loop is expected to have two predecessors:
1019
     first_loop->exit and the block that precedes first_loop.  */
1020
 
1021
  gcc_assert (EDGE_COUNT (loop2_entry_bb->preds) == 2
1022
              && ((EDGE_PRED (loop2_entry_bb, 0)->src == loop1_exit_bb
1023
                   && EDGE_PRED (loop2_entry_bb, 1)->src == loop1_entry_bb)
1024
               || (EDGE_PRED (loop2_entry_bb, 1)->src ==  loop1_exit_bb
1025
                   && EDGE_PRED (loop2_entry_bb, 0)->src == loop1_entry_bb)));
1026
 
1027
  /* Verify that the other successor of first_loop->exit is after the
1028
     second_loop.  */
1029
  /* TODO */
1030
}
1031
#endif
1032
 
1033
/* If the run time cost model check determines that vectorization is
1034
   not profitable and hence scalar loop should be generated then set
1035
   FIRST_NITERS to prologue peeled iterations. This will allow all the
1036
   iterations to be executed in the prologue peeled scalar loop.  */
1037
 
1038
static void
1039
set_prologue_iterations (basic_block bb_before_first_loop,
1040
                         tree *first_niters,
1041
                         struct loop *loop,
1042
                         unsigned int th)
1043
{
1044
  edge e;
1045
  basic_block cond_bb, then_bb;
1046
  tree var, prologue_after_cost_adjust_name;
1047
  gimple_stmt_iterator gsi;
1048
  gimple newphi;
1049
  edge e_true, e_false, e_fallthru;
1050
  gimple cond_stmt;
1051
  gimple_seq gimplify_stmt_list = NULL, stmts = NULL;
1052
  tree cost_pre_condition = NULL_TREE;
1053
  tree scalar_loop_iters =
1054
    unshare_expr (LOOP_VINFO_NITERS_UNCHANGED (loop_vec_info_for_loop (loop)));
1055
 
1056
  e = single_pred_edge (bb_before_first_loop);
1057
  cond_bb = split_edge(e);
1058
 
1059
  e = single_pred_edge (bb_before_first_loop);
1060
  then_bb = split_edge(e);
1061
  set_immediate_dominator (CDI_DOMINATORS, then_bb, cond_bb);
1062
 
1063
  e_false = make_single_succ_edge (cond_bb, bb_before_first_loop,
1064
                                   EDGE_FALSE_VALUE);
1065
  set_immediate_dominator (CDI_DOMINATORS, bb_before_first_loop, cond_bb);
1066
 
1067
  e_true = EDGE_PRED (then_bb, 0);
1068
  e_true->flags &= ~EDGE_FALLTHRU;
1069
  e_true->flags |= EDGE_TRUE_VALUE;
1070
 
1071
  e_fallthru = EDGE_SUCC (then_bb, 0);
1072
 
1073
  cost_pre_condition =
1074
    fold_build2 (LE_EXPR, boolean_type_node, scalar_loop_iters,
1075
                 build_int_cst (TREE_TYPE (scalar_loop_iters), th));
1076
  cost_pre_condition =
1077
    force_gimple_operand (cost_pre_condition, &gimplify_stmt_list,
1078
                          true, NULL_TREE);
1079
  cond_stmt = gimple_build_cond (NE_EXPR, cost_pre_condition,
1080
                                 build_int_cst (TREE_TYPE (cost_pre_condition),
1081
                                                0), NULL_TREE, NULL_TREE);
1082
 
1083
  gsi = gsi_last_bb (cond_bb);
1084
  if (gimplify_stmt_list)
1085
    gsi_insert_seq_after (&gsi, gimplify_stmt_list, GSI_NEW_STMT);
1086
 
1087
  gsi = gsi_last_bb (cond_bb);
1088
  gsi_insert_after (&gsi, cond_stmt, GSI_NEW_STMT);
1089
 
1090
  var = create_tmp_var (TREE_TYPE (scalar_loop_iters),
1091
                        "prologue_after_cost_adjust");
1092
  add_referenced_var (var);
1093
  prologue_after_cost_adjust_name =
1094
    force_gimple_operand (scalar_loop_iters, &stmts, false, var);
1095
 
1096
  gsi = gsi_last_bb (then_bb);
1097
  if (stmts)
1098
    gsi_insert_seq_after (&gsi, stmts, GSI_NEW_STMT);
1099
 
1100
  newphi = create_phi_node (var, bb_before_first_loop);
1101
  add_phi_arg (newphi, prologue_after_cost_adjust_name, e_fallthru,
1102
               UNKNOWN_LOCATION);
1103
  add_phi_arg (newphi, *first_niters, e_false, UNKNOWN_LOCATION);
1104
 
1105
  *first_niters = PHI_RESULT (newphi);
1106
}
1107
 
1108
/* Function slpeel_tree_peel_loop_to_edge.
1109
 
1110
   Peel the first (last) iterations of LOOP into a new prolog (epilog) loop
1111
   that is placed on the entry (exit) edge E of LOOP. After this transformation
1112
   we have two loops one after the other - first-loop iterates FIRST_NITERS
1113
   times, and second-loop iterates the remainder NITERS - FIRST_NITERS times.
1114
   If the cost model indicates that it is profitable to emit a scalar
1115
   loop instead of the vector one, then the prolog (epilog) loop will iterate
1116
   for the entire unchanged scalar iterations of the loop.
1117
 
1118
   Input:
1119
   - LOOP: the loop to be peeled.
1120
   - E: the exit or entry edge of LOOP.
1121
        If it is the entry edge, we peel the first iterations of LOOP. In this
1122
        case first-loop is LOOP, and second-loop is the newly created loop.
1123
        If it is the exit edge, we peel the last iterations of LOOP. In this
1124
        case, first-loop is the newly created loop, and second-loop is LOOP.
1125
   - NITERS: the number of iterations that LOOP iterates.
1126
   - FIRST_NITERS: the number of iterations that the first-loop should iterate.
1127
   - UPDATE_FIRST_LOOP_COUNT:  specified whether this function is responsible
1128
        for updating the loop bound of the first-loop to FIRST_NITERS.  If it
1129
        is false, the caller of this function may want to take care of this
1130
        (this can be useful if we don't want new stmts added to first-loop).
1131
   - TH: cost model profitability threshold of iterations for vectorization.
1132
   - CHECK_PROFITABILITY: specify whether cost model check has not occurred
1133
                          during versioning and hence needs to occur during
1134
                          prologue generation or whether cost model check
1135
                          has not occurred during prologue generation and hence
1136
                          needs to occur during epilogue generation.
1137
 
1138
 
1139
   Output:
1140
   The function returns a pointer to the new loop-copy, or NULL if it failed
1141
   to perform the transformation.
1142
 
1143
   The function generates two if-then-else guards: one before the first loop,
1144
   and the other before the second loop:
1145
   The first guard is:
1146
     if (FIRST_NITERS == 0) then skip the first loop,
1147
     and go directly to the second loop.
1148
   The second guard is:
1149
     if (FIRST_NITERS == NITERS) then skip the second loop.
1150
 
1151
   If the optional COND_EXPR and COND_EXPR_STMT_LIST arguments are given
1152
   then the generated condition is combined with COND_EXPR and the
1153
   statements in COND_EXPR_STMT_LIST are emitted together with it.
1154
 
1155
   FORNOW only simple loops are supported (see slpeel_can_duplicate_loop_p).
1156
   FORNOW the resulting code will not be in loop-closed-ssa form.
1157
*/
1158
 
1159
static struct loop*
1160
slpeel_tree_peel_loop_to_edge (struct loop *loop,
1161
                               edge e, tree *first_niters,
1162
                               tree niters, bool update_first_loop_count,
1163
                               unsigned int th, bool check_profitability,
1164
                               tree cond_expr, gimple_seq cond_expr_stmt_list)
1165
{
1166
  struct loop *new_loop = NULL, *first_loop, *second_loop;
1167
  edge skip_e;
1168
  tree pre_condition = NULL_TREE;
1169
  bitmap definitions;
1170
  basic_block bb_before_second_loop, bb_after_second_loop;
1171
  basic_block bb_before_first_loop;
1172
  basic_block bb_between_loops;
1173
  basic_block new_exit_bb;
1174
  gimple_stmt_iterator gsi;
1175
  edge exit_e = single_exit (loop);
1176
  LOC loop_loc;
1177
  tree cost_pre_condition = NULL_TREE;
1178
 
1179
  if (!slpeel_can_duplicate_loop_p (loop, e))
1180
    return NULL;
1181
 
1182
  /* We have to initialize cfg_hooks. Then, when calling
1183
   cfg_hooks->split_edge, the function tree_split_edge
1184
   is actually called and, when calling cfg_hooks->duplicate_block,
1185
   the function tree_duplicate_bb is called.  */
1186
  gimple_register_cfg_hooks ();
1187
 
1188
  /* If the loop has a virtual PHI, but exit bb doesn't, create a virtual PHI
1189
     in the exit bb and rename all the uses after the loop.  This simplifies
1190
     the *guard[12] routines, which assume loop closed SSA form for all PHIs
1191
     (but normally loop closed SSA form doesn't require virtual PHIs to be
1192
     in the same form).  Doing this early simplifies the checking what
1193
     uses should be renamed.  */
1194
  for (gsi = gsi_start_phis (loop->header); !gsi_end_p (gsi); gsi_next (&gsi))
1195
    if (!is_gimple_reg (gimple_phi_result (gsi_stmt (gsi))))
1196
      {
1197
        gimple phi = gsi_stmt (gsi);
1198
        for (gsi = gsi_start_phis (exit_e->dest);
1199
             !gsi_end_p (gsi); gsi_next (&gsi))
1200
          if (!is_gimple_reg (gimple_phi_result (gsi_stmt (gsi))))
1201
            break;
1202
        if (gsi_end_p (gsi))
1203
          {
1204
            gimple new_phi = create_phi_node (SSA_NAME_VAR (PHI_RESULT (phi)),
1205
                                              exit_e->dest);
1206
            tree vop = PHI_ARG_DEF_FROM_EDGE (phi, EDGE_SUCC (loop->latch, 0));
1207
            imm_use_iterator imm_iter;
1208
            gimple stmt;
1209
            tree new_vop = make_ssa_name (SSA_NAME_VAR (PHI_RESULT (phi)),
1210
                                          new_phi);
1211
            use_operand_p use_p;
1212
 
1213
            add_phi_arg (new_phi, vop, exit_e, UNKNOWN_LOCATION);
1214
            gimple_phi_set_result (new_phi, new_vop);
1215
            FOR_EACH_IMM_USE_STMT (stmt, imm_iter, vop)
1216
              if (stmt != new_phi && gimple_bb (stmt) != loop->header)
1217
                FOR_EACH_IMM_USE_ON_STMT (use_p, imm_iter)
1218
                  SET_USE (use_p, new_vop);
1219
          }
1220
        break;
1221
      }
1222
 
1223
  /* 1. Generate a copy of LOOP and put it on E (E is the entry/exit of LOOP).
1224
        Resulting CFG would be:
1225
 
1226
        first_loop:
1227
        do {
1228
        } while ...
1229
 
1230
        second_loop:
1231
        do {
1232
        } while ...
1233
 
1234
        orig_exit_bb:
1235
   */
1236
 
1237
  if (!(new_loop = slpeel_tree_duplicate_loop_to_edge_cfg (loop, e)))
1238
    {
1239
      loop_loc = find_loop_location (loop);
1240
      if (dump_file && (dump_flags & TDF_DETAILS))
1241
        {
1242
          if (loop_loc != UNKNOWN_LOC)
1243
            fprintf (dump_file, "\n%s:%d: note: ",
1244
                     LOC_FILE (loop_loc), LOC_LINE (loop_loc));
1245
          fprintf (dump_file, "tree_duplicate_loop_to_edge_cfg failed.\n");
1246
        }
1247
      return NULL;
1248
    }
1249
 
1250
  if (MAY_HAVE_DEBUG_STMTS)
1251
    {
1252
      gcc_assert (!adjust_vec);
1253
      adjust_vec = VEC_alloc (adjust_info, stack, 32);
1254
    }
1255
 
1256
  if (e == exit_e)
1257
    {
1258
      /* NEW_LOOP was placed after LOOP.  */
1259
      first_loop = loop;
1260
      second_loop = new_loop;
1261
    }
1262
  else
1263
    {
1264
      /* NEW_LOOP was placed before LOOP.  */
1265
      first_loop = new_loop;
1266
      second_loop = loop;
1267
    }
1268
 
1269
  definitions = ssa_names_to_replace ();
1270
  slpeel_update_phis_for_duplicate_loop (loop, new_loop, e == exit_e);
1271
  rename_variables_in_loop (new_loop);
1272
 
1273
 
1274
  /* 2.  Add the guard code in one of the following ways:
1275
 
1276
     2.a Add the guard that controls whether the first loop is executed.
1277
         This occurs when this function is invoked for prologue or epilogue
1278
         generation and when the cost model check can be done at compile time.
1279
 
1280
         Resulting CFG would be:
1281
 
1282
         bb_before_first_loop:
1283
         if (FIRST_NITERS == 0) GOTO bb_before_second_loop
1284
                                GOTO first-loop
1285
 
1286
         first_loop:
1287
         do {
1288
         } while ...
1289
 
1290
         bb_before_second_loop:
1291
 
1292
         second_loop:
1293
         do {
1294
         } while ...
1295
 
1296
         orig_exit_bb:
1297
 
1298
     2.b Add the cost model check that allows the prologue
1299
         to iterate for the entire unchanged scalar
1300
         iterations of the loop in the event that the cost
1301
         model indicates that the scalar loop is more
1302
         profitable than the vector one. This occurs when
1303
         this function is invoked for prologue generation
1304
         and the cost model check needs to be done at run
1305
         time.
1306
 
1307
         Resulting CFG after prologue peeling would be:
1308
 
1309
         if (scalar_loop_iterations <= th)
1310
           FIRST_NITERS = scalar_loop_iterations
1311
 
1312
         bb_before_first_loop:
1313
         if (FIRST_NITERS == 0) GOTO bb_before_second_loop
1314
                                GOTO first-loop
1315
 
1316
         first_loop:
1317
         do {
1318
         } while ...
1319
 
1320
         bb_before_second_loop:
1321
 
1322
         second_loop:
1323
         do {
1324
         } while ...
1325
 
1326
         orig_exit_bb:
1327
 
1328
     2.c Add the cost model check that allows the epilogue
1329
         to iterate for the entire unchanged scalar
1330
         iterations of the loop in the event that the cost
1331
         model indicates that the scalar loop is more
1332
         profitable than the vector one. This occurs when
1333
         this function is invoked for epilogue generation
1334
         and the cost model check needs to be done at run
1335
         time.  This check is combined with any pre-existing
1336
         check in COND_EXPR to avoid versioning.
1337
 
1338
         Resulting CFG after prologue peeling would be:
1339
 
1340
         bb_before_first_loop:
1341
         if ((scalar_loop_iterations <= th)
1342
             ||
1343
             FIRST_NITERS == 0) GOTO bb_before_second_loop
1344
                                GOTO first-loop
1345
 
1346
         first_loop:
1347
         do {
1348
         } while ...
1349
 
1350
         bb_before_second_loop:
1351
 
1352
         second_loop:
1353
         do {
1354
         } while ...
1355
 
1356
         orig_exit_bb:
1357
  */
1358
 
1359
  bb_before_first_loop = split_edge (loop_preheader_edge (first_loop));
1360
  bb_before_second_loop = split_edge (single_exit (first_loop));
1361
 
1362
  /* Epilogue peeling.  */
1363
  if (!update_first_loop_count)
1364
    {
1365
      pre_condition =
1366
        fold_build2 (LE_EXPR, boolean_type_node, *first_niters,
1367
                     build_int_cst (TREE_TYPE (*first_niters), 0));
1368
      if (check_profitability)
1369
        {
1370
          tree scalar_loop_iters
1371
            = unshare_expr (LOOP_VINFO_NITERS_UNCHANGED
1372
                                        (loop_vec_info_for_loop (loop)));
1373
          cost_pre_condition =
1374
            fold_build2 (LE_EXPR, boolean_type_node, scalar_loop_iters,
1375
                         build_int_cst (TREE_TYPE (scalar_loop_iters), th));
1376
 
1377
          pre_condition = fold_build2 (TRUTH_OR_EXPR, boolean_type_node,
1378
                                       cost_pre_condition, pre_condition);
1379
        }
1380
      if (cond_expr)
1381
        {
1382
          pre_condition =
1383
            fold_build2 (TRUTH_OR_EXPR, boolean_type_node,
1384
                         pre_condition,
1385
                         fold_build1 (TRUTH_NOT_EXPR, boolean_type_node,
1386
                                      cond_expr));
1387
        }
1388
    }
1389
 
1390
  /* Prologue peeling.  */
1391
  else
1392
    {
1393
      if (check_profitability)
1394
        set_prologue_iterations (bb_before_first_loop, first_niters,
1395
                                 loop, th);
1396
 
1397
      pre_condition =
1398
        fold_build2 (LE_EXPR, boolean_type_node, *first_niters,
1399
                     build_int_cst (TREE_TYPE (*first_niters), 0));
1400
    }
1401
 
1402
  skip_e = slpeel_add_loop_guard (bb_before_first_loop, pre_condition,
1403
                                  cond_expr_stmt_list,
1404
                                  bb_before_second_loop, bb_before_first_loop);
1405
  slpeel_update_phi_nodes_for_guard1 (skip_e, first_loop,
1406
                                      first_loop == new_loop,
1407
                                      &new_exit_bb, &definitions);
1408
 
1409
 
1410
  /* 3. Add the guard that controls whether the second loop is executed.
1411
        Resulting CFG would be:
1412
 
1413
        bb_before_first_loop:
1414
        if (FIRST_NITERS == 0) GOTO bb_before_second_loop (skip first loop)
1415
                               GOTO first-loop
1416
 
1417
        first_loop:
1418
        do {
1419
        } while ...
1420
 
1421
        bb_between_loops:
1422
        if (FIRST_NITERS == NITERS) GOTO bb_after_second_loop (skip second loop)
1423
                                    GOTO bb_before_second_loop
1424
 
1425
        bb_before_second_loop:
1426
 
1427
        second_loop:
1428
        do {
1429
        } while ...
1430
 
1431
        bb_after_second_loop:
1432
 
1433
        orig_exit_bb:
1434
   */
1435
 
1436
  bb_between_loops = new_exit_bb;
1437
  bb_after_second_loop = split_edge (single_exit (second_loop));
1438
 
1439
  pre_condition =
1440
        fold_build2 (EQ_EXPR, boolean_type_node, *first_niters, niters);
1441
  skip_e = slpeel_add_loop_guard (bb_between_loops, pre_condition, NULL,
1442
                                  bb_after_second_loop, bb_before_first_loop);
1443
  slpeel_update_phi_nodes_for_guard2 (skip_e, second_loop,
1444
                                     second_loop == new_loop, &new_exit_bb);
1445
 
1446
  /* 4. Make first-loop iterate FIRST_NITERS times, if requested.
1447
   */
1448
  if (update_first_loop_count)
1449
    slpeel_make_loop_iterate_ntimes (first_loop, *first_niters);
1450
 
1451
  BITMAP_FREE (definitions);
1452
  delete_update_ssa ();
1453
 
1454
  adjust_vec_debug_stmts ();
1455
 
1456
  return new_loop;
1457
}
1458
 
1459
/* Function vect_get_loop_location.
1460
 
1461
   Extract the location of the loop in the source code.
1462
   If the loop is not well formed for vectorization, an estimated
1463
   location is calculated.
1464
   Return the loop location if succeed and NULL if not.  */
1465
 
1466
LOC
1467
find_loop_location (struct loop *loop)
1468
{
1469
  gimple stmt = NULL;
1470
  basic_block bb;
1471
  gimple_stmt_iterator si;
1472
 
1473
  if (!loop)
1474
    return UNKNOWN_LOC;
1475
 
1476
  stmt = get_loop_exit_condition (loop);
1477
 
1478
  if (stmt && gimple_location (stmt) != UNKNOWN_LOC)
1479
    return gimple_location (stmt);
1480
 
1481
  /* If we got here the loop is probably not "well formed",
1482
     try to estimate the loop location */
1483
 
1484
  if (!loop->header)
1485
    return UNKNOWN_LOC;
1486
 
1487
  bb = loop->header;
1488
 
1489
  for (si = gsi_start_bb (bb); !gsi_end_p (si); gsi_next (&si))
1490
    {
1491
      stmt = gsi_stmt (si);
1492
      if (gimple_location (stmt) != UNKNOWN_LOC)
1493
        return gimple_location (stmt);
1494
    }
1495
 
1496
  return UNKNOWN_LOC;
1497
}
1498
 
1499
 
1500
/* This function builds ni_name = number of iterations loop executes
1501
   on the loop preheader.  If SEQ is given the stmt is instead emitted
1502
   there.  */
1503
 
1504
static tree
1505
vect_build_loop_niters (loop_vec_info loop_vinfo, gimple_seq seq)
1506
{
1507
  tree ni_name, var;
1508
  gimple_seq stmts = NULL;
1509
  edge pe;
1510
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
1511
  tree ni = unshare_expr (LOOP_VINFO_NITERS (loop_vinfo));
1512
 
1513
  var = create_tmp_var (TREE_TYPE (ni), "niters");
1514
  add_referenced_var (var);
1515
  ni_name = force_gimple_operand (ni, &stmts, false, var);
1516
 
1517
  pe = loop_preheader_edge (loop);
1518
  if (stmts)
1519
    {
1520
      if (seq)
1521
        gimple_seq_add_seq (&seq, stmts);
1522
      else
1523
        {
1524
          basic_block new_bb = gsi_insert_seq_on_edge_immediate (pe, stmts);
1525
          gcc_assert (!new_bb);
1526
        }
1527
    }
1528
 
1529
  return ni_name;
1530
}
1531
 
1532
 
1533
/* This function generates the following statements:
1534
 
1535
 ni_name = number of iterations loop executes
1536
 ratio = ni_name / vf
1537
 ratio_mult_vf_name = ratio * vf
1538
 
1539
 and places them at the loop preheader edge or in COND_EXPR_STMT_LIST
1540
 if that is non-NULL.  */
1541
 
1542
static void
1543
vect_generate_tmps_on_preheader (loop_vec_info loop_vinfo,
1544
                                 tree *ni_name_ptr,
1545
                                 tree *ratio_mult_vf_name_ptr,
1546
                                 tree *ratio_name_ptr,
1547
                                 gimple_seq cond_expr_stmt_list)
1548
{
1549
 
1550
  edge pe;
1551
  basic_block new_bb;
1552
  gimple_seq stmts;
1553
  tree ni_name, ni_minus_gap_name;
1554
  tree var;
1555
  tree ratio_name;
1556
  tree ratio_mult_vf_name;
1557
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
1558
  tree ni = LOOP_VINFO_NITERS (loop_vinfo);
1559
  int vf = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
1560
  tree log_vf;
1561
 
1562
  pe = loop_preheader_edge (loop);
1563
 
1564
  /* Generate temporary variable that contains
1565
     number of iterations loop executes.  */
1566
 
1567
  ni_name = vect_build_loop_niters (loop_vinfo, cond_expr_stmt_list);
1568
  log_vf = build_int_cst (TREE_TYPE (ni), exact_log2 (vf));
1569
 
1570
  /* If epilogue loop is required because of data accesses with gaps, we
1571
     subtract one iteration from the total number of iterations here for
1572
     correct calculation of RATIO.  */
1573
  if (LOOP_VINFO_PEELING_FOR_GAPS (loop_vinfo))
1574
    {
1575
      ni_minus_gap_name = fold_build2 (MINUS_EXPR, TREE_TYPE (ni_name),
1576
                                       ni_name,
1577
                                       build_one_cst (TREE_TYPE (ni_name)));
1578
      if (!is_gimple_val (ni_minus_gap_name))
1579
        {
1580
          var = create_tmp_var (TREE_TYPE (ni), "ni_gap");
1581
          add_referenced_var (var);
1582
 
1583
          stmts = NULL;
1584
          ni_minus_gap_name = force_gimple_operand (ni_minus_gap_name, &stmts,
1585
                                                    true, var);
1586
          if (cond_expr_stmt_list)
1587
            gimple_seq_add_seq (&cond_expr_stmt_list, stmts);
1588
          else
1589
            {
1590
              pe = loop_preheader_edge (loop);
1591
              new_bb = gsi_insert_seq_on_edge_immediate (pe, stmts);
1592
              gcc_assert (!new_bb);
1593
            }
1594
        }
1595
    }
1596
  else
1597
    ni_minus_gap_name = ni_name;
1598
 
1599
  /* Create: ratio = ni >> log2(vf) */
1600
 
1601
  ratio_name = fold_build2 (RSHIFT_EXPR, TREE_TYPE (ni_minus_gap_name),
1602
                            ni_minus_gap_name, log_vf);
1603
  if (!is_gimple_val (ratio_name))
1604
    {
1605
      var = create_tmp_var (TREE_TYPE (ni), "bnd");
1606
      add_referenced_var (var);
1607
 
1608
      stmts = NULL;
1609
      ratio_name = force_gimple_operand (ratio_name, &stmts, true, var);
1610
      if (cond_expr_stmt_list)
1611
        gimple_seq_add_seq (&cond_expr_stmt_list, stmts);
1612
      else
1613
        {
1614
          pe = loop_preheader_edge (loop);
1615
          new_bb = gsi_insert_seq_on_edge_immediate (pe, stmts);
1616
          gcc_assert (!new_bb);
1617
        }
1618
    }
1619
 
1620
  /* Create: ratio_mult_vf = ratio << log2 (vf).  */
1621
 
1622
  ratio_mult_vf_name = fold_build2 (LSHIFT_EXPR, TREE_TYPE (ratio_name),
1623
                                    ratio_name, log_vf);
1624
  if (!is_gimple_val (ratio_mult_vf_name))
1625
    {
1626
      var = create_tmp_var (TREE_TYPE (ni), "ratio_mult_vf");
1627
      add_referenced_var (var);
1628
 
1629
      stmts = NULL;
1630
      ratio_mult_vf_name = force_gimple_operand (ratio_mult_vf_name, &stmts,
1631
                                                 true, var);
1632
      if (cond_expr_stmt_list)
1633
        gimple_seq_add_seq (&cond_expr_stmt_list, stmts);
1634
      else
1635
        {
1636
          pe = loop_preheader_edge (loop);
1637
          new_bb = gsi_insert_seq_on_edge_immediate (pe, stmts);
1638
          gcc_assert (!new_bb);
1639
        }
1640
    }
1641
 
1642
  *ni_name_ptr = ni_name;
1643
  *ratio_mult_vf_name_ptr = ratio_mult_vf_name;
1644
  *ratio_name_ptr = ratio_name;
1645
 
1646
  return;
1647
}
1648
 
1649
/* Function vect_can_advance_ivs_p
1650
 
1651
   In case the number of iterations that LOOP iterates is unknown at compile
1652
   time, an epilog loop will be generated, and the loop induction variables
1653
   (IVs) will be "advanced" to the value they are supposed to take just before
1654
   the epilog loop.  Here we check that the access function of the loop IVs
1655
   and the expression that represents the loop bound are simple enough.
1656
   These restrictions will be relaxed in the future.  */
1657
 
1658
bool
1659
vect_can_advance_ivs_p (loop_vec_info loop_vinfo)
1660
{
1661
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
1662
  basic_block bb = loop->header;
1663
  gimple phi;
1664
  gimple_stmt_iterator gsi;
1665
 
1666
  /* Analyze phi functions of the loop header.  */
1667
 
1668
  if (vect_print_dump_info (REPORT_DETAILS))
1669
    fprintf (vect_dump, "vect_can_advance_ivs_p:");
1670
 
1671
  for (gsi = gsi_start_phis (bb); !gsi_end_p (gsi); gsi_next (&gsi))
1672
    {
1673
      tree access_fn = NULL;
1674
      tree evolution_part;
1675
 
1676
      phi = gsi_stmt (gsi);
1677
      if (vect_print_dump_info (REPORT_DETAILS))
1678
        {
1679
          fprintf (vect_dump, "Analyze phi: ");
1680
          print_gimple_stmt (vect_dump, phi, 0, TDF_SLIM);
1681
        }
1682
 
1683
      /* Skip virtual phi's. The data dependences that are associated with
1684
         virtual defs/uses (i.e., memory accesses) are analyzed elsewhere.  */
1685
 
1686
      if (!is_gimple_reg (SSA_NAME_VAR (PHI_RESULT (phi))))
1687
        {
1688
          if (vect_print_dump_info (REPORT_DETAILS))
1689
            fprintf (vect_dump, "virtual phi. skip.");
1690
          continue;
1691
        }
1692
 
1693
      /* Skip reduction phis.  */
1694
 
1695
      if (STMT_VINFO_DEF_TYPE (vinfo_for_stmt (phi)) == vect_reduction_def)
1696
        {
1697
          if (vect_print_dump_info (REPORT_DETAILS))
1698
            fprintf (vect_dump, "reduc phi. skip.");
1699
          continue;
1700
        }
1701
 
1702
      /* Analyze the evolution function.  */
1703
 
1704
      access_fn = instantiate_parameters
1705
        (loop, analyze_scalar_evolution (loop, PHI_RESULT (phi)));
1706
 
1707
      if (!access_fn)
1708
        {
1709
          if (vect_print_dump_info (REPORT_DETAILS))
1710
            fprintf (vect_dump, "No Access function.");
1711
          return false;
1712
        }
1713
 
1714
      if (vect_print_dump_info (REPORT_DETAILS))
1715
        {
1716
          fprintf (vect_dump, "Access function of PHI: ");
1717
          print_generic_expr (vect_dump, access_fn, TDF_SLIM);
1718
        }
1719
 
1720
      evolution_part = evolution_part_in_loop_num (access_fn, loop->num);
1721
 
1722
      if (evolution_part == NULL_TREE)
1723
        {
1724
          if (vect_print_dump_info (REPORT_DETAILS))
1725
            fprintf (vect_dump, "No evolution.");
1726
          return false;
1727
        }
1728
 
1729
      /* FORNOW: We do not transform initial conditions of IVs
1730
         which evolution functions are a polynomial of degree >= 2.  */
1731
 
1732
      if (tree_is_chrec (evolution_part))
1733
        return false;
1734
    }
1735
 
1736
  return true;
1737
}
1738
 
1739
 
1740
/*   Function vect_update_ivs_after_vectorizer.
1741
 
1742
     "Advance" the induction variables of LOOP to the value they should take
1743
     after the execution of LOOP.  This is currently necessary because the
1744
     vectorizer does not handle induction variables that are used after the
1745
     loop.  Such a situation occurs when the last iterations of LOOP are
1746
     peeled, because:
1747
     1. We introduced new uses after LOOP for IVs that were not originally used
1748
        after LOOP: the IVs of LOOP are now used by an epilog loop.
1749
     2. LOOP is going to be vectorized; this means that it will iterate N/VF
1750
        times, whereas the loop IVs should be bumped N times.
1751
 
1752
     Input:
1753
     - LOOP - a loop that is going to be vectorized. The last few iterations
1754
              of LOOP were peeled.
1755
     - NITERS - the number of iterations that LOOP executes (before it is
1756
                vectorized). i.e, the number of times the ivs should be bumped.
1757
     - UPDATE_E - a successor edge of LOOP->exit that is on the (only) path
1758
                  coming out from LOOP on which there are uses of the LOOP ivs
1759
                  (this is the path from LOOP->exit to epilog_loop->preheader).
1760
 
1761
                  The new definitions of the ivs are placed in LOOP->exit.
1762
                  The phi args associated with the edge UPDATE_E in the bb
1763
                  UPDATE_E->dest are updated accordingly.
1764
 
1765
     Assumption 1: Like the rest of the vectorizer, this function assumes
1766
     a single loop exit that has a single predecessor.
1767
 
1768
     Assumption 2: The phi nodes in the LOOP header and in update_bb are
1769
     organized in the same order.
1770
 
1771
     Assumption 3: The access function of the ivs is simple enough (see
1772
     vect_can_advance_ivs_p).  This assumption will be relaxed in the future.
1773
 
1774
     Assumption 4: Exactly one of the successors of LOOP exit-bb is on a path
1775
     coming out of LOOP on which the ivs of LOOP are used (this is the path
1776
     that leads to the epilog loop; other paths skip the epilog loop).  This
1777
     path starts with the edge UPDATE_E, and its destination (denoted update_bb)
1778
     needs to have its phis updated.
1779
 */
1780
 
1781
static void
1782
vect_update_ivs_after_vectorizer (loop_vec_info loop_vinfo, tree niters,
1783
                                  edge update_e)
1784
{
1785
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
1786
  basic_block exit_bb = single_exit (loop)->dest;
1787
  gimple phi, phi1;
1788
  gimple_stmt_iterator gsi, gsi1;
1789
  basic_block update_bb = update_e->dest;
1790
 
1791
  /* gcc_assert (vect_can_advance_ivs_p (loop_vinfo)); */
1792
 
1793
  /* Make sure there exists a single-predecessor exit bb:  */
1794
  gcc_assert (single_pred_p (exit_bb));
1795
 
1796
  for (gsi = gsi_start_phis (loop->header), gsi1 = gsi_start_phis (update_bb);
1797
       !gsi_end_p (gsi) && !gsi_end_p (gsi1);
1798
       gsi_next (&gsi), gsi_next (&gsi1))
1799
    {
1800
      tree access_fn = NULL;
1801
      tree evolution_part;
1802
      tree init_expr;
1803
      tree step_expr, off;
1804
      tree type;
1805
      tree var, ni, ni_name;
1806
      gimple_stmt_iterator last_gsi;
1807
 
1808
      phi = gsi_stmt (gsi);
1809
      phi1 = gsi_stmt (gsi1);
1810
      if (vect_print_dump_info (REPORT_DETAILS))
1811
        {
1812
          fprintf (vect_dump, "vect_update_ivs_after_vectorizer: phi: ");
1813
          print_gimple_stmt (vect_dump, phi, 0, TDF_SLIM);
1814
        }
1815
 
1816
      /* Skip virtual phi's.  */
1817
      if (!is_gimple_reg (SSA_NAME_VAR (PHI_RESULT (phi))))
1818
        {
1819
          if (vect_print_dump_info (REPORT_DETAILS))
1820
            fprintf (vect_dump, "virtual phi. skip.");
1821
          continue;
1822
        }
1823
 
1824
      /* Skip reduction phis.  */
1825
      if (STMT_VINFO_DEF_TYPE (vinfo_for_stmt (phi)) == vect_reduction_def)
1826
        {
1827
          if (vect_print_dump_info (REPORT_DETAILS))
1828
            fprintf (vect_dump, "reduc phi. skip.");
1829
          continue;
1830
        }
1831
 
1832
      access_fn = analyze_scalar_evolution (loop, PHI_RESULT (phi));
1833
      gcc_assert (access_fn);
1834
      /* We can end up with an access_fn like
1835
           (short int) {(short unsigned int) i_49, +, 1}_1
1836
         for further analysis we need to strip the outer cast but we
1837
         need to preserve the original type.  */
1838
      type = TREE_TYPE (access_fn);
1839
      STRIP_NOPS (access_fn);
1840
      evolution_part =
1841
         unshare_expr (evolution_part_in_loop_num (access_fn, loop->num));
1842
      gcc_assert (evolution_part != NULL_TREE);
1843
 
1844
      /* FORNOW: We do not support IVs whose evolution function is a polynomial
1845
         of degree >= 2 or exponential.  */
1846
      gcc_assert (!tree_is_chrec (evolution_part));
1847
 
1848
      step_expr = evolution_part;
1849
      init_expr = unshare_expr (initial_condition_in_loop_num (access_fn,
1850
                                                               loop->num));
1851
      init_expr = fold_convert (type, init_expr);
1852
 
1853
      off = fold_build2 (MULT_EXPR, TREE_TYPE (step_expr),
1854
                         fold_convert (TREE_TYPE (step_expr), niters),
1855
                         step_expr);
1856
      if (POINTER_TYPE_P (TREE_TYPE (init_expr)))
1857
        ni = fold_build_pointer_plus (init_expr, off);
1858
      else
1859
        ni = fold_build2 (PLUS_EXPR, TREE_TYPE (init_expr),
1860
                          init_expr,
1861
                          fold_convert (TREE_TYPE (init_expr), off));
1862
 
1863
      var = create_tmp_var (TREE_TYPE (init_expr), "tmp");
1864
      add_referenced_var (var);
1865
 
1866
      last_gsi = gsi_last_bb (exit_bb);
1867
      ni_name = force_gimple_operand_gsi (&last_gsi, ni, false, var,
1868
                                          true, GSI_SAME_STMT);
1869
 
1870
      /* Fix phi expressions in the successor bb.  */
1871
      adjust_phi_and_debug_stmts (phi1, update_e, ni_name);
1872
    }
1873
}
1874
 
1875
/* Return the more conservative threshold between the
1876
   min_profitable_iters returned by the cost model and the user
1877
   specified threshold, if provided.  */
1878
 
1879
static unsigned int
1880
conservative_cost_threshold (loop_vec_info loop_vinfo,
1881
                             int min_profitable_iters)
1882
{
1883
  unsigned int th;
1884
  int min_scalar_loop_bound;
1885
 
1886
  min_scalar_loop_bound = ((PARAM_VALUE (PARAM_MIN_VECT_LOOP_BOUND)
1887
                            * LOOP_VINFO_VECT_FACTOR (loop_vinfo)) - 1);
1888
 
1889
  /* Use the cost model only if it is more conservative than user specified
1890
     threshold.  */
1891
  th = (unsigned) min_scalar_loop_bound;
1892
  if (min_profitable_iters
1893
      && (!min_scalar_loop_bound
1894
          || min_profitable_iters > min_scalar_loop_bound))
1895
    th = (unsigned) min_profitable_iters;
1896
 
1897
  if (th && vect_print_dump_info (REPORT_COST))
1898
    fprintf (vect_dump, "Profitability threshold is %u loop iterations.", th);
1899
 
1900
  return th;
1901
}
1902
 
1903
/* Function vect_do_peeling_for_loop_bound
1904
 
1905
   Peel the last iterations of the loop represented by LOOP_VINFO.
1906
   The peeled iterations form a new epilog loop.  Given that the loop now
1907
   iterates NITERS times, the new epilog loop iterates
1908
   NITERS % VECTORIZATION_FACTOR times.
1909
 
1910
   The original loop will later be made to iterate
1911
   NITERS / VECTORIZATION_FACTOR times (this value is placed into RATIO).
1912
 
1913
   COND_EXPR and COND_EXPR_STMT_LIST are combined with a new generated
1914
   test.  */
1915
 
1916
void
1917
vect_do_peeling_for_loop_bound (loop_vec_info loop_vinfo, tree *ratio,
1918
                                tree cond_expr, gimple_seq cond_expr_stmt_list)
1919
{
1920
  tree ni_name, ratio_mult_vf_name;
1921
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
1922
  struct loop *new_loop;
1923
  edge update_e;
1924
  basic_block preheader;
1925
  int loop_num;
1926
  bool check_profitability = false;
1927
  unsigned int th = 0;
1928
  int min_profitable_iters;
1929
 
1930
  if (vect_print_dump_info (REPORT_DETAILS))
1931
    fprintf (vect_dump, "=== vect_do_peeling_for_loop_bound ===");
1932
 
1933
  initialize_original_copy_tables ();
1934
 
1935
  /* Generate the following variables on the preheader of original loop:
1936
 
1937
     ni_name = number of iteration the original loop executes
1938
     ratio = ni_name / vf
1939
     ratio_mult_vf_name = ratio * vf  */
1940
  vect_generate_tmps_on_preheader (loop_vinfo, &ni_name,
1941
                                   &ratio_mult_vf_name, ratio,
1942
                                   cond_expr_stmt_list);
1943
 
1944
  loop_num  = loop->num;
1945
 
1946
  /* If cost model check not done during versioning and
1947
     peeling for alignment.  */
1948
  if (!LOOP_REQUIRES_VERSIONING_FOR_ALIGNMENT (loop_vinfo)
1949
      && !LOOP_REQUIRES_VERSIONING_FOR_ALIAS (loop_vinfo)
1950
      && !LOOP_PEELING_FOR_ALIGNMENT (loop_vinfo)
1951
      && !cond_expr)
1952
    {
1953
      check_profitability = true;
1954
 
1955
      /* Get profitability threshold for vectorized loop.  */
1956
      min_profitable_iters = LOOP_VINFO_COST_MODEL_MIN_ITERS (loop_vinfo);
1957
 
1958
      th = conservative_cost_threshold (loop_vinfo,
1959
                                        min_profitable_iters);
1960
    }
1961
 
1962
  new_loop = slpeel_tree_peel_loop_to_edge (loop, single_exit (loop),
1963
                                            &ratio_mult_vf_name, ni_name, false,
1964
                                            th, check_profitability,
1965
                                            cond_expr, cond_expr_stmt_list);
1966
  gcc_assert (new_loop);
1967
  gcc_assert (loop_num == loop->num);
1968
#ifdef ENABLE_CHECKING
1969
  slpeel_verify_cfg_after_peeling (loop, new_loop);
1970
#endif
1971
 
1972
  /* A guard that controls whether the new_loop is to be executed or skipped
1973
     is placed in LOOP->exit.  LOOP->exit therefore has two successors - one
1974
     is the preheader of NEW_LOOP, where the IVs from LOOP are used.  The other
1975
     is a bb after NEW_LOOP, where these IVs are not used.  Find the edge that
1976
     is on the path where the LOOP IVs are used and need to be updated.  */
1977
 
1978
  preheader = loop_preheader_edge (new_loop)->src;
1979
  if (EDGE_PRED (preheader, 0)->src == single_exit (loop)->dest)
1980
    update_e = EDGE_PRED (preheader, 0);
1981
  else
1982
    update_e = EDGE_PRED (preheader, 1);
1983
 
1984
  /* Update IVs of original loop as if they were advanced
1985
     by ratio_mult_vf_name steps.  */
1986
  vect_update_ivs_after_vectorizer (loop_vinfo, ratio_mult_vf_name, update_e);
1987
 
1988
  /* After peeling we have to reset scalar evolution analyzer.  */
1989
  scev_reset ();
1990
 
1991
  free_original_copy_tables ();
1992
}
1993
 
1994
 
1995
/* Function vect_gen_niters_for_prolog_loop
1996
 
1997
   Set the number of iterations for the loop represented by LOOP_VINFO
1998
   to the minimum between LOOP_NITERS (the original iteration count of the loop)
1999
   and the misalignment of DR - the data reference recorded in
2000
   LOOP_VINFO_UNALIGNED_DR (LOOP_VINFO).  As a result, after the execution of
2001
   this loop, the data reference DR will refer to an aligned location.
2002
 
2003
   The following computation is generated:
2004
 
2005
   If the misalignment of DR is known at compile time:
2006
     addr_mis = int mis = DR_MISALIGNMENT (dr);
2007
   Else, compute address misalignment in bytes:
2008
     addr_mis = addr & (vectype_size - 1)
2009
 
2010
   prolog_niters = min (LOOP_NITERS, ((VF - addr_mis/elem_size)&(VF-1))/step)
2011
 
2012
   (elem_size = element type size; an element is the scalar element whose type
2013
   is the inner type of the vectype)
2014
 
2015
   When the step of the data-ref in the loop is not 1 (as in interleaved data
2016
   and SLP), the number of iterations of the prolog must be divided by the step
2017
   (which is equal to the size of interleaved group).
2018
 
2019
   The above formulas assume that VF == number of elements in the vector. This
2020
   may not hold when there are multiple-types in the loop.
2021
   In this case, for some data-references in the loop the VF does not represent
2022
   the number of elements that fit in the vector.  Therefore, instead of VF we
2023
   use TYPE_VECTOR_SUBPARTS.  */
2024
 
2025
static tree
2026
vect_gen_niters_for_prolog_loop (loop_vec_info loop_vinfo, tree loop_niters)
2027
{
2028
  struct data_reference *dr = LOOP_VINFO_UNALIGNED_DR (loop_vinfo);
2029
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
2030
  tree var;
2031
  gimple_seq stmts;
2032
  tree iters, iters_name;
2033
  edge pe;
2034
  basic_block new_bb;
2035
  gimple dr_stmt = DR_STMT (dr);
2036
  stmt_vec_info stmt_info = vinfo_for_stmt (dr_stmt);
2037
  tree vectype = STMT_VINFO_VECTYPE (stmt_info);
2038
  int vectype_align = TYPE_ALIGN (vectype) / BITS_PER_UNIT;
2039
  tree niters_type = TREE_TYPE (loop_niters);
2040
  int nelements = TYPE_VECTOR_SUBPARTS (vectype);
2041
 
2042
  pe = loop_preheader_edge (loop);
2043
 
2044
  if (LOOP_PEELING_FOR_ALIGNMENT (loop_vinfo) > 0)
2045
    {
2046
      int npeel = LOOP_PEELING_FOR_ALIGNMENT (loop_vinfo);
2047
 
2048
      if (vect_print_dump_info (REPORT_DETAILS))
2049
        fprintf (vect_dump, "known peeling = %d.", npeel);
2050
 
2051
      iters = build_int_cst (niters_type, npeel);
2052
    }
2053
  else
2054
    {
2055
      gimple_seq new_stmts = NULL;
2056
      bool negative = tree_int_cst_compare (DR_STEP (dr), size_zero_node) < 0;
2057
      tree offset = negative
2058
          ? size_int (-TYPE_VECTOR_SUBPARTS (vectype) + 1) : NULL_TREE;
2059
      tree start_addr = vect_create_addr_base_for_vector_ref (dr_stmt,
2060
                                                &new_stmts, offset, loop);
2061
      tree ptr_type = TREE_TYPE (start_addr);
2062
      tree size = TYPE_SIZE (ptr_type);
2063
      tree type = lang_hooks.types.type_for_size (tree_low_cst (size, 1), 1);
2064
      tree vectype_size_minus_1 = build_int_cst (type, vectype_align - 1);
2065
      tree elem_size_log =
2066
        build_int_cst (type, exact_log2 (vectype_align/nelements));
2067
      tree nelements_minus_1 = build_int_cst (type, nelements - 1);
2068
      tree nelements_tree = build_int_cst (type, nelements);
2069
      tree byte_misalign;
2070
      tree elem_misalign;
2071
 
2072
      new_bb = gsi_insert_seq_on_edge_immediate (pe, new_stmts);
2073
      gcc_assert (!new_bb);
2074
 
2075
      /* Create:  byte_misalign = addr & (vectype_size - 1)  */
2076
      byte_misalign =
2077
        fold_build2 (BIT_AND_EXPR, type, fold_convert (type, start_addr),
2078
                     vectype_size_minus_1);
2079
 
2080
      /* Create:  elem_misalign = byte_misalign / element_size  */
2081
      elem_misalign =
2082
        fold_build2 (RSHIFT_EXPR, type, byte_misalign, elem_size_log);
2083
 
2084
      /* Create:  (niters_type) (nelements - elem_misalign)&(nelements - 1)  */
2085
      if (negative)
2086
        iters = fold_build2 (MINUS_EXPR, type, elem_misalign, nelements_tree);
2087
      else
2088
        iters = fold_build2 (MINUS_EXPR, type, nelements_tree, elem_misalign);
2089
      iters = fold_build2 (BIT_AND_EXPR, type, iters, nelements_minus_1);
2090
      iters = fold_convert (niters_type, iters);
2091
    }
2092
 
2093
  /* Create:  prolog_loop_niters = min (iters, loop_niters) */
2094
  /* If the loop bound is known at compile time we already verified that it is
2095
     greater than vf; since the misalignment ('iters') is at most vf, there's
2096
     no need to generate the MIN_EXPR in this case.  */
2097
  if (TREE_CODE (loop_niters) != INTEGER_CST)
2098
    iters = fold_build2 (MIN_EXPR, niters_type, iters, loop_niters);
2099
 
2100
  if (vect_print_dump_info (REPORT_DETAILS))
2101
    {
2102
      fprintf (vect_dump, "niters for prolog loop: ");
2103
      print_generic_expr (vect_dump, iters, TDF_SLIM);
2104
    }
2105
 
2106
  var = create_tmp_var (niters_type, "prolog_loop_niters");
2107
  add_referenced_var (var);
2108
  stmts = NULL;
2109
  iters_name = force_gimple_operand (iters, &stmts, false, var);
2110
 
2111
  /* Insert stmt on loop preheader edge.  */
2112
  if (stmts)
2113
    {
2114
      basic_block new_bb = gsi_insert_seq_on_edge_immediate (pe, stmts);
2115
      gcc_assert (!new_bb);
2116
    }
2117
 
2118
  return iters_name;
2119
}
2120
 
2121
 
2122
/* Function vect_update_init_of_dr
2123
 
2124
   NITERS iterations were peeled from LOOP.  DR represents a data reference
2125
   in LOOP.  This function updates the information recorded in DR to
2126
   account for the fact that the first NITERS iterations had already been
2127
   executed.  Specifically, it updates the OFFSET field of DR.  */
2128
 
2129
static void
2130
vect_update_init_of_dr (struct data_reference *dr, tree niters)
2131
{
2132
  tree offset = DR_OFFSET (dr);
2133
 
2134
  niters = fold_build2 (MULT_EXPR, sizetype,
2135
                        fold_convert (sizetype, niters),
2136
                        fold_convert (sizetype, DR_STEP (dr)));
2137
  offset = fold_build2 (PLUS_EXPR, sizetype,
2138
                        fold_convert (sizetype, offset), niters);
2139
  DR_OFFSET (dr) = offset;
2140
}
2141
 
2142
 
2143
/* Function vect_update_inits_of_drs
2144
 
2145
   NITERS iterations were peeled from the loop represented by LOOP_VINFO.
2146
   This function updates the information recorded for the data references in
2147
   the loop to account for the fact that the first NITERS iterations had
2148
   already been executed.  Specifically, it updates the initial_condition of
2149
   the access_function of all the data_references in the loop.  */
2150
 
2151
static void
2152
vect_update_inits_of_drs (loop_vec_info loop_vinfo, tree niters)
2153
{
2154
  unsigned int i;
2155
  VEC (data_reference_p, heap) *datarefs = LOOP_VINFO_DATAREFS (loop_vinfo);
2156
  struct data_reference *dr;
2157
 
2158
  if (vect_print_dump_info (REPORT_DETAILS))
2159
    fprintf (vect_dump, "=== vect_update_inits_of_dr ===");
2160
 
2161
  FOR_EACH_VEC_ELT (data_reference_p, datarefs, i, dr)
2162
    vect_update_init_of_dr (dr, niters);
2163
}
2164
 
2165
 
2166
/* Function vect_do_peeling_for_alignment
2167
 
2168
   Peel the first 'niters' iterations of the loop represented by LOOP_VINFO.
2169
   'niters' is set to the misalignment of one of the data references in the
2170
   loop, thereby forcing it to refer to an aligned location at the beginning
2171
   of the execution of this loop.  The data reference for which we are
2172
   peeling is recorded in LOOP_VINFO_UNALIGNED_DR.  */
2173
 
2174
void
2175
vect_do_peeling_for_alignment (loop_vec_info loop_vinfo)
2176
{
2177
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
2178
  tree niters_of_prolog_loop, ni_name;
2179
  tree n_iters;
2180
  tree wide_prolog_niters;
2181
  struct loop *new_loop;
2182
  unsigned int th = 0;
2183
  int min_profitable_iters;
2184
 
2185
  if (vect_print_dump_info (REPORT_DETAILS))
2186
    fprintf (vect_dump, "=== vect_do_peeling_for_alignment ===");
2187
 
2188
  initialize_original_copy_tables ();
2189
 
2190
  ni_name = vect_build_loop_niters (loop_vinfo, NULL);
2191
  niters_of_prolog_loop = vect_gen_niters_for_prolog_loop (loop_vinfo,
2192
                                                           ni_name);
2193
 
2194
  /* Get profitability threshold for vectorized loop.  */
2195
  min_profitable_iters = LOOP_VINFO_COST_MODEL_MIN_ITERS (loop_vinfo);
2196
  th = conservative_cost_threshold (loop_vinfo,
2197
                                    min_profitable_iters);
2198
 
2199
  /* Peel the prolog loop and iterate it niters_of_prolog_loop.  */
2200
  new_loop =
2201
    slpeel_tree_peel_loop_to_edge (loop, loop_preheader_edge (loop),
2202
                                   &niters_of_prolog_loop, ni_name, true,
2203
                                   th, true, NULL_TREE, NULL);
2204
 
2205
  gcc_assert (new_loop);
2206
#ifdef ENABLE_CHECKING
2207
  slpeel_verify_cfg_after_peeling (new_loop, loop);
2208
#endif
2209
 
2210
  /* Update number of times loop executes.  */
2211
  n_iters = LOOP_VINFO_NITERS (loop_vinfo);
2212
  LOOP_VINFO_NITERS (loop_vinfo) = fold_build2 (MINUS_EXPR,
2213
                TREE_TYPE (n_iters), n_iters, niters_of_prolog_loop);
2214
 
2215
  if (types_compatible_p (sizetype, TREE_TYPE (niters_of_prolog_loop)))
2216
    wide_prolog_niters = niters_of_prolog_loop;
2217
  else
2218
    {
2219
      gimple_seq seq = NULL;
2220
      edge pe = loop_preheader_edge (loop);
2221
      tree wide_iters = fold_convert (sizetype, niters_of_prolog_loop);
2222
      tree var = create_tmp_var (sizetype, "prolog_loop_adjusted_niters");
2223
      add_referenced_var (var);
2224
      wide_prolog_niters = force_gimple_operand (wide_iters, &seq, false,
2225
                                                 var);
2226
      if (seq)
2227
        {
2228
          /* Insert stmt on loop preheader edge.  */
2229
          basic_block new_bb = gsi_insert_seq_on_edge_immediate (pe, seq);
2230
          gcc_assert (!new_bb);
2231
        }
2232
    }
2233
 
2234
  /* Update the init conditions of the access functions of all data refs.  */
2235
  vect_update_inits_of_drs (loop_vinfo, wide_prolog_niters);
2236
 
2237
  /* After peeling we have to reset scalar evolution analyzer.  */
2238
  scev_reset ();
2239
 
2240
  free_original_copy_tables ();
2241
}
2242
 
2243
 
2244
/* Function vect_create_cond_for_align_checks.
2245
 
2246
   Create a conditional expression that represents the alignment checks for
2247
   all of data references (array element references) whose alignment must be
2248
   checked at runtime.
2249
 
2250
   Input:
2251
   COND_EXPR  - input conditional expression.  New conditions will be chained
2252
                with logical AND operation.
2253
   LOOP_VINFO - two fields of the loop information are used.
2254
                LOOP_VINFO_PTR_MASK is the mask used to check the alignment.
2255
                LOOP_VINFO_MAY_MISALIGN_STMTS contains the refs to be checked.
2256
 
2257
   Output:
2258
   COND_EXPR_STMT_LIST - statements needed to construct the conditional
2259
                         expression.
2260
   The returned value is the conditional expression to be used in the if
2261
   statement that controls which version of the loop gets executed at runtime.
2262
 
2263
   The algorithm makes two assumptions:
2264
     1) The number of bytes "n" in a vector is a power of 2.
2265
     2) An address "a" is aligned if a%n is zero and that this
2266
        test can be done as a&(n-1) == 0.  For example, for 16
2267
        byte vectors the test is a&0xf == 0.  */
2268
 
2269
static void
2270
vect_create_cond_for_align_checks (loop_vec_info loop_vinfo,
2271
                                   tree *cond_expr,
2272
                                   gimple_seq *cond_expr_stmt_list)
2273
{
2274
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
2275
  VEC(gimple,heap) *may_misalign_stmts
2276
    = LOOP_VINFO_MAY_MISALIGN_STMTS (loop_vinfo);
2277
  gimple ref_stmt;
2278
  int mask = LOOP_VINFO_PTR_MASK (loop_vinfo);
2279
  tree mask_cst;
2280
  unsigned int i;
2281
  tree psize;
2282
  tree int_ptrsize_type;
2283
  char tmp_name[20];
2284
  tree or_tmp_name = NULL_TREE;
2285
  tree and_tmp, and_tmp_name;
2286
  gimple and_stmt;
2287
  tree ptrsize_zero;
2288
  tree part_cond_expr;
2289
 
2290
  /* Check that mask is one less than a power of 2, i.e., mask is
2291
     all zeros followed by all ones.  */
2292
  gcc_assert ((mask != 0) && ((mask & (mask+1)) == 0));
2293
 
2294
  /* CHECKME: what is the best integer or unsigned type to use to hold a
2295
     cast from a pointer value?  */
2296
  psize = TYPE_SIZE (ptr_type_node);
2297
  int_ptrsize_type
2298
    = lang_hooks.types.type_for_size (tree_low_cst (psize, 1), 0);
2299
 
2300
  /* Create expression (mask & (dr_1 || ... || dr_n)) where dr_i is the address
2301
     of the first vector of the i'th data reference. */
2302
 
2303
  FOR_EACH_VEC_ELT (gimple, may_misalign_stmts, i, ref_stmt)
2304
    {
2305
      gimple_seq new_stmt_list = NULL;
2306
      tree addr_base;
2307
      tree addr_tmp, addr_tmp_name;
2308
      tree or_tmp, new_or_tmp_name;
2309
      gimple addr_stmt, or_stmt;
2310
      stmt_vec_info stmt_vinfo = vinfo_for_stmt (ref_stmt);
2311
      tree vectype = STMT_VINFO_VECTYPE (stmt_vinfo);
2312
      bool negative = tree_int_cst_compare
2313
        (DR_STEP (STMT_VINFO_DATA_REF (stmt_vinfo)), size_zero_node) < 0;
2314
      tree offset = negative
2315
        ? size_int (-TYPE_VECTOR_SUBPARTS (vectype) + 1) : NULL_TREE;
2316
 
2317
      /* create: addr_tmp = (int)(address_of_first_vector) */
2318
      addr_base =
2319
        vect_create_addr_base_for_vector_ref (ref_stmt, &new_stmt_list,
2320
                                              offset, loop);
2321
      if (new_stmt_list != NULL)
2322
        gimple_seq_add_seq (cond_expr_stmt_list, new_stmt_list);
2323
 
2324
      sprintf (tmp_name, "%s%d", "addr2int", i);
2325
      addr_tmp = create_tmp_var (int_ptrsize_type, tmp_name);
2326
      add_referenced_var (addr_tmp);
2327
      addr_tmp_name = make_ssa_name (addr_tmp, NULL);
2328
      addr_stmt = gimple_build_assign_with_ops (NOP_EXPR, addr_tmp_name,
2329
                                                addr_base, NULL_TREE);
2330
      SSA_NAME_DEF_STMT (addr_tmp_name) = addr_stmt;
2331
      gimple_seq_add_stmt (cond_expr_stmt_list, addr_stmt);
2332
 
2333
      /* The addresses are OR together.  */
2334
 
2335
      if (or_tmp_name != NULL_TREE)
2336
        {
2337
          /* create: or_tmp = or_tmp | addr_tmp */
2338
          sprintf (tmp_name, "%s%d", "orptrs", i);
2339
          or_tmp = create_tmp_var (int_ptrsize_type, tmp_name);
2340
          add_referenced_var (or_tmp);
2341
          new_or_tmp_name = make_ssa_name (or_tmp, NULL);
2342
          or_stmt = gimple_build_assign_with_ops (BIT_IOR_EXPR,
2343
                                                  new_or_tmp_name,
2344
                                                  or_tmp_name, addr_tmp_name);
2345
          SSA_NAME_DEF_STMT (new_or_tmp_name) = or_stmt;
2346
          gimple_seq_add_stmt (cond_expr_stmt_list, or_stmt);
2347
          or_tmp_name = new_or_tmp_name;
2348
        }
2349
      else
2350
        or_tmp_name = addr_tmp_name;
2351
 
2352
    } /* end for i */
2353
 
2354
  mask_cst = build_int_cst (int_ptrsize_type, mask);
2355
 
2356
  /* create: and_tmp = or_tmp & mask  */
2357
  and_tmp = create_tmp_var (int_ptrsize_type, "andmask" );
2358
  add_referenced_var (and_tmp);
2359
  and_tmp_name = make_ssa_name (and_tmp, NULL);
2360
 
2361
  and_stmt = gimple_build_assign_with_ops (BIT_AND_EXPR, and_tmp_name,
2362
                                           or_tmp_name, mask_cst);
2363
  SSA_NAME_DEF_STMT (and_tmp_name) = and_stmt;
2364
  gimple_seq_add_stmt (cond_expr_stmt_list, and_stmt);
2365
 
2366
  /* Make and_tmp the left operand of the conditional test against zero.
2367
     if and_tmp has a nonzero bit then some address is unaligned.  */
2368
  ptrsize_zero = build_int_cst (int_ptrsize_type, 0);
2369
  part_cond_expr = fold_build2 (EQ_EXPR, boolean_type_node,
2370
                                and_tmp_name, ptrsize_zero);
2371
  if (*cond_expr)
2372
    *cond_expr = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
2373
                              *cond_expr, part_cond_expr);
2374
  else
2375
    *cond_expr = part_cond_expr;
2376
}
2377
 
2378
 
2379
/* Function vect_vfa_segment_size.
2380
 
2381
   Create an expression that computes the size of segment
2382
   that will be accessed for a data reference.  The functions takes into
2383
   account that realignment loads may access one more vector.
2384
 
2385
   Input:
2386
     DR: The data reference.
2387
     LENGTH_FACTOR: segment length to consider.
2388
 
2389
   Return an expression whose value is the size of segment which will be
2390
   accessed by DR.  */
2391
 
2392
static tree
2393
vect_vfa_segment_size (struct data_reference *dr, tree length_factor)
2394
{
2395
  tree segment_length;
2396
 
2397
  if (!compare_tree_int (DR_STEP (dr), 0))
2398
    segment_length = TYPE_SIZE_UNIT (TREE_TYPE (DR_REF (dr)));
2399
  else
2400
    segment_length = size_binop (MULT_EXPR,
2401
                                 fold_convert (sizetype, DR_STEP (dr)),
2402
                                 fold_convert (sizetype, length_factor));
2403
 
2404
  if (vect_supportable_dr_alignment (dr, false)
2405
        == dr_explicit_realign_optimized)
2406
    {
2407
      tree vector_size = TYPE_SIZE_UNIT
2408
                          (STMT_VINFO_VECTYPE (vinfo_for_stmt (DR_STMT (dr))));
2409
 
2410
      segment_length = size_binop (PLUS_EXPR, segment_length, vector_size);
2411
    }
2412
  return segment_length;
2413
}
2414
 
2415
 
2416
/* Function vect_create_cond_for_alias_checks.
2417
 
2418
   Create a conditional expression that represents the run-time checks for
2419
   overlapping of address ranges represented by a list of data references
2420
   relations passed as input.
2421
 
2422
   Input:
2423
   COND_EXPR  - input conditional expression.  New conditions will be chained
2424
                with logical AND operation.
2425
   LOOP_VINFO - field LOOP_VINFO_MAY_ALIAS_STMTS contains the list of ddrs
2426
                to be checked.
2427
 
2428
   Output:
2429
   COND_EXPR - conditional expression.
2430
   COND_EXPR_STMT_LIST - statements needed to construct the conditional
2431
                         expression.
2432
 
2433
 
2434
   The returned value is the conditional expression to be used in the if
2435
   statement that controls which version of the loop gets executed at runtime.
2436
*/
2437
 
2438
static void
2439
vect_create_cond_for_alias_checks (loop_vec_info loop_vinfo,
2440
                                   tree * cond_expr,
2441
                                   gimple_seq * cond_expr_stmt_list)
2442
{
2443
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
2444
  VEC (ddr_p, heap) * may_alias_ddrs =
2445
    LOOP_VINFO_MAY_ALIAS_DDRS (loop_vinfo);
2446
  int vect_factor = LOOP_VINFO_VECT_FACTOR (loop_vinfo);
2447
  tree scalar_loop_iters = LOOP_VINFO_NITERS (loop_vinfo);
2448
 
2449
  ddr_p ddr;
2450
  unsigned int i;
2451
  tree part_cond_expr, length_factor;
2452
 
2453
  /* Create expression
2454
     ((store_ptr_0 + store_segment_length_0) <= load_ptr_0)
2455
     || (load_ptr_0 + load_segment_length_0) <= store_ptr_0))
2456
     &&
2457
     ...
2458
     &&
2459
     ((store_ptr_n + store_segment_length_n) <= load_ptr_n)
2460
     || (load_ptr_n + load_segment_length_n) <= store_ptr_n))  */
2461
 
2462
  if (VEC_empty (ddr_p, may_alias_ddrs))
2463
    return;
2464
 
2465
  FOR_EACH_VEC_ELT (ddr_p, may_alias_ddrs, i, ddr)
2466
    {
2467
      struct data_reference *dr_a, *dr_b;
2468
      gimple dr_group_first_a, dr_group_first_b;
2469
      tree addr_base_a, addr_base_b;
2470
      tree segment_length_a, segment_length_b;
2471
      gimple stmt_a, stmt_b;
2472
      tree seg_a_min, seg_a_max, seg_b_min, seg_b_max;
2473
 
2474
      dr_a = DDR_A (ddr);
2475
      stmt_a = DR_STMT (DDR_A (ddr));
2476
      dr_group_first_a = GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt_a));
2477
      if (dr_group_first_a)
2478
        {
2479
          stmt_a = dr_group_first_a;
2480
          dr_a = STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt_a));
2481
        }
2482
 
2483
      dr_b = DDR_B (ddr);
2484
      stmt_b = DR_STMT (DDR_B (ddr));
2485
      dr_group_first_b = GROUP_FIRST_ELEMENT (vinfo_for_stmt (stmt_b));
2486
      if (dr_group_first_b)
2487
        {
2488
          stmt_b = dr_group_first_b;
2489
          dr_b = STMT_VINFO_DATA_REF (vinfo_for_stmt (stmt_b));
2490
        }
2491
 
2492
      addr_base_a =
2493
        vect_create_addr_base_for_vector_ref (stmt_a, cond_expr_stmt_list,
2494
                                              NULL_TREE, loop);
2495
      addr_base_b =
2496
        vect_create_addr_base_for_vector_ref (stmt_b, cond_expr_stmt_list,
2497
                                              NULL_TREE, loop);
2498
 
2499
      if (!operand_equal_p (DR_STEP (dr_a), DR_STEP (dr_b), 0))
2500
        length_factor = scalar_loop_iters;
2501
      else
2502
        length_factor = size_int (vect_factor);
2503
      segment_length_a = vect_vfa_segment_size (dr_a, length_factor);
2504
      segment_length_b = vect_vfa_segment_size (dr_b, length_factor);
2505
 
2506
      if (vect_print_dump_info (REPORT_DR_DETAILS))
2507
        {
2508
          fprintf (vect_dump,
2509
                   "create runtime check for data references ");
2510
          print_generic_expr (vect_dump, DR_REF (dr_a), TDF_SLIM);
2511
          fprintf (vect_dump, " and ");
2512
          print_generic_expr (vect_dump, DR_REF (dr_b), TDF_SLIM);
2513
        }
2514
 
2515
      seg_a_min = addr_base_a;
2516
      seg_a_max = fold_build_pointer_plus (addr_base_a, segment_length_a);
2517
      if (tree_int_cst_compare (DR_STEP (dr_a), size_zero_node) < 0)
2518
        seg_a_min = seg_a_max, seg_a_max = addr_base_a;
2519
 
2520
      seg_b_min = addr_base_b;
2521
      seg_b_max = fold_build_pointer_plus (addr_base_b, segment_length_b);
2522
      if (tree_int_cst_compare (DR_STEP (dr_b), size_zero_node) < 0)
2523
        seg_b_min = seg_b_max, seg_b_max = addr_base_b;
2524
 
2525
      part_cond_expr =
2526
        fold_build2 (TRUTH_OR_EXPR, boolean_type_node,
2527
          fold_build2 (LE_EXPR, boolean_type_node, seg_a_max, seg_b_min),
2528
          fold_build2 (LE_EXPR, boolean_type_node, seg_b_max, seg_a_min));
2529
 
2530
      if (*cond_expr)
2531
        *cond_expr = fold_build2 (TRUTH_AND_EXPR, boolean_type_node,
2532
                                  *cond_expr, part_cond_expr);
2533
      else
2534
        *cond_expr = part_cond_expr;
2535
    }
2536
 
2537
  if (vect_print_dump_info (REPORT_VECTORIZED_LOCATIONS))
2538
    fprintf (vect_dump, "created %u versioning for alias checks.\n",
2539
             VEC_length (ddr_p, may_alias_ddrs));
2540
}
2541
 
2542
 
2543
/* Function vect_loop_versioning.
2544
 
2545
   If the loop has data references that may or may not be aligned or/and
2546
   has data reference relations whose independence was not proven then
2547
   two versions of the loop need to be generated, one which is vectorized
2548
   and one which isn't.  A test is then generated to control which of the
2549
   loops is executed.  The test checks for the alignment of all of the
2550
   data references that may or may not be aligned.  An additional
2551
   sequence of runtime tests is generated for each pairs of DDRs whose
2552
   independence was not proven.  The vectorized version of loop is
2553
   executed only if both alias and alignment tests are passed.
2554
 
2555
   The test generated to check which version of loop is executed
2556
   is modified to also check for profitability as indicated by the
2557
   cost model initially.
2558
 
2559
   The versioning precondition(s) are placed in *COND_EXPR and
2560
   *COND_EXPR_STMT_LIST.  If DO_VERSIONING is true versioning is
2561
   also performed, otherwise only the conditions are generated.  */
2562
 
2563
void
2564
vect_loop_versioning (loop_vec_info loop_vinfo, bool do_versioning,
2565
                      tree *cond_expr, gimple_seq *cond_expr_stmt_list)
2566
{
2567
  struct loop *loop = LOOP_VINFO_LOOP (loop_vinfo);
2568
  basic_block condition_bb;
2569
  gimple_stmt_iterator gsi, cond_exp_gsi;
2570
  basic_block merge_bb;
2571
  basic_block new_exit_bb;
2572
  edge new_exit_e, e;
2573
  gimple orig_phi, new_phi;
2574
  tree arg;
2575
  unsigned prob = 4 * REG_BR_PROB_BASE / 5;
2576
  gimple_seq gimplify_stmt_list = NULL;
2577
  tree scalar_loop_iters = LOOP_VINFO_NITERS (loop_vinfo);
2578
  int min_profitable_iters = 0;
2579
  unsigned int th;
2580
 
2581
  /* Get profitability threshold for vectorized loop.  */
2582
  min_profitable_iters = LOOP_VINFO_COST_MODEL_MIN_ITERS (loop_vinfo);
2583
 
2584
  th = conservative_cost_threshold (loop_vinfo,
2585
                                    min_profitable_iters);
2586
 
2587
  *cond_expr =
2588
    fold_build2 (GT_EXPR, boolean_type_node, scalar_loop_iters,
2589
                 build_int_cst (TREE_TYPE (scalar_loop_iters), th));
2590
 
2591
  *cond_expr = force_gimple_operand (*cond_expr, cond_expr_stmt_list,
2592
                                     false, NULL_TREE);
2593
 
2594
  if (LOOP_REQUIRES_VERSIONING_FOR_ALIGNMENT (loop_vinfo))
2595
      vect_create_cond_for_align_checks (loop_vinfo, cond_expr,
2596
                                         cond_expr_stmt_list);
2597
 
2598
  if (LOOP_REQUIRES_VERSIONING_FOR_ALIAS (loop_vinfo))
2599
    vect_create_cond_for_alias_checks (loop_vinfo, cond_expr,
2600
                                       cond_expr_stmt_list);
2601
 
2602
  *cond_expr =
2603
    fold_build2 (NE_EXPR, boolean_type_node, *cond_expr, integer_zero_node);
2604
  *cond_expr =
2605
    force_gimple_operand (*cond_expr, &gimplify_stmt_list, true, NULL_TREE);
2606
  gimple_seq_add_seq (cond_expr_stmt_list, gimplify_stmt_list);
2607
 
2608
  /* If we only needed the extra conditions and a new loop copy
2609
     bail out here.  */
2610
  if (!do_versioning)
2611
    return;
2612
 
2613
  initialize_original_copy_tables ();
2614
  loop_version (loop, *cond_expr, &condition_bb,
2615
                prob, prob, REG_BR_PROB_BASE - prob, true);
2616
  free_original_copy_tables();
2617
 
2618
  /* Loop versioning violates an assumption we try to maintain during
2619
     vectorization - that the loop exit block has a single predecessor.
2620
     After versioning, the exit block of both loop versions is the same
2621
     basic block (i.e. it has two predecessors). Just in order to simplify
2622
     following transformations in the vectorizer, we fix this situation
2623
     here by adding a new (empty) block on the exit-edge of the loop,
2624
     with the proper loop-exit phis to maintain loop-closed-form.  */
2625
 
2626
  merge_bb = single_exit (loop)->dest;
2627
  gcc_assert (EDGE_COUNT (merge_bb->preds) == 2);
2628
  new_exit_bb = split_edge (single_exit (loop));
2629
  new_exit_e = single_exit (loop);
2630
  e = EDGE_SUCC (new_exit_bb, 0);
2631
 
2632
  for (gsi = gsi_start_phis (merge_bb); !gsi_end_p (gsi); gsi_next (&gsi))
2633
    {
2634
      orig_phi = gsi_stmt (gsi);
2635
      new_phi = create_phi_node (SSA_NAME_VAR (PHI_RESULT (orig_phi)),
2636
                                  new_exit_bb);
2637
      arg = PHI_ARG_DEF_FROM_EDGE (orig_phi, e);
2638
      add_phi_arg (new_phi, arg, new_exit_e,
2639
                   gimple_phi_arg_location_from_edge (orig_phi, e));
2640
      adjust_phi_and_debug_stmts (orig_phi, e, PHI_RESULT (new_phi));
2641
    }
2642
 
2643
  /* End loop-exit-fixes after versioning.  */
2644
 
2645
  update_ssa (TODO_update_ssa);
2646
  if (*cond_expr_stmt_list)
2647
    {
2648
      cond_exp_gsi = gsi_last_bb (condition_bb);
2649
      gsi_insert_seq_before (&cond_exp_gsi, *cond_expr_stmt_list,
2650
                             GSI_SAME_STMT);
2651
      *cond_expr_stmt_list = NULL;
2652
    }
2653
  *cond_expr = NULL_TREE;
2654
}
2655
 

powered by: WebSVN 2.1.0

© copyright 1999-2024 OpenCores.org, equivalent to Oliscience, all rights reserved. OpenCores®, registered trademark.