llvm/test/Transforms/LoopLoadElim/unknown-stride-known-dep.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
   2 ; RUN: opt -passes=loop-load-elim -S %s | FileCheck %s
   3
   4 ; The test was originally written as part of the investigation of #96656.
   5 ; The bug has now been marked as invalid, and we keep the test to show
   6 ; LLE's operation on known dependence returned by LAA.
   7
   8 define void @unknown_stride_known_dependence(ptr %x, ptr %y, i1 %cond) {
   9 ; CHECK-LABEL: define void @unknown_stride_known_dependence(
  10 ; CHECK-SAME: ptr [[X:%.*]], ptr [[Y:%.*]], i1 [[COND:%.*]]) {
  11 ; CHECK-NEXT:  [[ENTRY:.*:]]
  12 ; CHECK-NEXT:    [[LOAD:%.*]] = load i32, ptr [[X]], align 4
  13 ; CHECK-NEXT:    br i1 [[COND]], label %[[NOLOOP_EXIT:.*]], label %[[LOOP_LVER_CHECK:.*]]
  14 ; CHECK:       [[LOOP_LVER_CHECK]]:
  15 ; CHECK-NEXT:    [[SEXT_X:%.*]] = sext i32 [[LOAD]] to i64
  16 ; CHECK-NEXT:    [[GEP_8:%.*]] = getelementptr i8, ptr [[Y]], i64 8
  17 ; CHECK-NEXT:    [[GEP_16:%.*]] = getelementptr i8, ptr [[Y]], i64 16
  18 ; CHECK-NEXT:    [[IDENT_CHECK:%.*]] = icmp ne i32 [[LOAD]], 1
  19 ; CHECK-NEXT:    br i1 [[IDENT_CHECK]], label %[[LOOP_PH_LVER_ORIG:.*]], label %[[LOOP_PH:.*]]
  20 ; CHECK:       [[LOOP_PH_LVER_ORIG]]:
  21 ; CHECK-NEXT:    br label %[[LOOP_LVER_ORIG:.*]]
  22 ; CHECK:       [[LOOP_LVER_ORIG]]:
  23 ; CHECK-NEXT:    [[IV_LVER_ORIG:%.*]] = phi i64 [ 0, %[[LOOP_PH_LVER_ORIG]] ], [ [[IV_NEXT_LVER_ORIG:%.*]], %[[LOOP_LVER_ORIG]] ]
  24 ; CHECK-NEXT:    [[MUL_LVER_ORIG:%.*]] = mul i64 [[IV_LVER_ORIG]], [[SEXT_X]]
  25 ; CHECK-NEXT:    [[GEP_8_MUL_LVER_ORIG:%.*]] = getelementptr double, ptr [[GEP_8]], i64 [[MUL_LVER_ORIG]]
  26 ; CHECK-NEXT:    [[LOAD_8_LVER_ORIG:%.*]] = load double, ptr [[GEP_8_MUL_LVER_ORIG]], align 8
  27 ; CHECK-NEXT:    [[GEP_16_MUL_LVER_ORIG:%.*]] = getelementptr double, ptr [[GEP_16]], i64 [[MUL_LVER_ORIG]]
  28 ; CHECK-NEXT:    store double [[LOAD_8_LVER_ORIG]], ptr [[GEP_16_MUL_LVER_ORIG]], align 8
  29 ; CHECK-NEXT:    [[IV_NEXT_LVER_ORIG]] = add i64 [[IV_LVER_ORIG]], 1
  30 ; CHECK-NEXT:    [[ICMP_LVER_ORIG:%.*]] = icmp eq i64 [[IV_LVER_ORIG]], 1
  31 ; CHECK-NEXT:    br i1 [[ICMP_LVER_ORIG]], label %[[EXIT_LOOPEXIT_LOOPEXIT:.*]], label %[[LOOP_LVER_ORIG]]
  32 ; CHECK:       [[LOOP_PH]]:
  33 ; CHECK-NEXT:    [[LOAD_INITIAL:%.*]] = load double, ptr [[GEP_8]], align 8
  34 ; CHECK-NEXT:    br label %[[LOOP:.*]]
  35 ; CHECK:       [[LOOP]]:
  36 ; CHECK-NEXT:    [[STORE_FORWARDED:%.*]] = phi double [ [[LOAD_INITIAL]], %[[LOOP_PH]] ], [ [[STORE_FORWARDED]], %[[LOOP]] ]
  37 ; CHECK-NEXT:    [[IV:%.*]] = phi i64 [ 0, %[[LOOP_PH]] ], [ [[IV_NEXT:%.*]], %[[LOOP]] ]
  38 ; CHECK-NEXT:    [[MUL:%.*]] = mul i64 [[IV]], [[SEXT_X]]
  39 ; CHECK-NEXT:    [[GEP_8_MUL:%.*]] = getelementptr double, ptr [[GEP_8]], i64 [[MUL]]
  40 ; CHECK-NEXT:    [[LOAD_8:%.*]] = load double, ptr [[GEP_8_MUL]], align 8
  41 ; CHECK-NEXT:    [[GEP_16_MUL:%.*]] = getelementptr double, ptr [[GEP_16]], i64 [[MUL]]
  42 ; CHECK-NEXT:    store double [[STORE_FORWARDED]], ptr [[GEP_16_MUL]], align 8
  43 ; CHECK-NEXT:    [[IV_NEXT]] = add i64 [[IV]], 1
  44 ; CHECK-NEXT:    [[ICMP:%.*]] = icmp eq i64 [[IV]], 1
  45 ; CHECK-NEXT:    br i1 [[ICMP]], label %[[EXIT_LOOPEXIT_LOOPEXIT1:.*]], label %[[LOOP]]
  46 ; CHECK:       [[NOLOOP_EXIT]]:
  47 ; CHECK-NEXT:    [[SEXT:%.*]] = sext i32 [[LOAD]] to i64
  48 ; CHECK-NEXT:    [[GEP_Y:%.*]] = getelementptr double, ptr [[Y]], i64 [[SEXT]]
  49 ; CHECK-NEXT:    [[LOAD_Y:%.*]] = load double, ptr [[GEP_Y]], align 8
  50 ; CHECK-NEXT:    store double [[LOAD_Y]], ptr [[X]], align 8
  51 ; CHECK-NEXT:    br label %[[EXIT:.*]]
  52 ; CHECK:       [[EXIT_LOOPEXIT_LOOPEXIT]]:
  53 ; CHECK-NEXT:    br label %[[EXIT_LOOPEXIT:.*]]
  54 ; CHECK:       [[EXIT_LOOPEXIT_LOOPEXIT1]]:
  55 ; CHECK-NEXT:    br label %[[EXIT_LOOPEXIT]]
  56 ; CHECK:       [[EXIT_LOOPEXIT]]:
  57 ; CHECK-NEXT:    br label %[[EXIT]]
  58 ; CHECK:       [[EXIT]]:
  59 ; CHECK-NEXT:    ret void
  60 ;
  61 entry:
  62   %load = load i32, ptr %x, align 4
  63   br i1 %cond, label %noloop.exit, label %loop.ph
  64
  65 loop.ph:                                              ; preds = %entry
  66   %sext.x = sext i32 %load to i64
  67   %gep.8 = getelementptr i8, ptr %y, i64 8
  68   %gep.16 = getelementptr i8, ptr %y, i64 16
  69   br label %loop
  70
  71 loop:                                                 ; preds = %loop, %loop.ph
  72   %iv = phi i64 [ 0, %loop.ph ], [ %iv.next, %loop ]
  73   %mul = mul i64 %iv, %sext.x
  74   %gep.8.mul = getelementptr double, ptr %gep.8, i64 %mul
  75   %load.8 = load double, ptr %gep.8.mul, align 8
  76   %gep.16.mul = getelementptr double, ptr %gep.16, i64 %mul
  77   store double %load.8, ptr %gep.16.mul
  78   %iv.next = add i64 %iv, 1
  79   %icmp = icmp eq i64 %iv, 1
  80   br i1 %icmp, label %exit, label %loop
  81
  82 noloop.exit:                                          ; preds = %loop.ph
  83   %sext = sext i32 %load to i64
  84   %gep.y = getelementptr double, ptr %y, i64 %sext
  85   %load.y = load double, ptr %gep.y
  86   store double %load.y, ptr %x
  87   br label %exit
  88
  89 exit:                                                 ; preds = %loop.body
  90   ret void
  91 }