llvm/test/CodeGen/ARM/ParallelDSP/smlad0.ll

   1 ; RUN: opt -mtriple=armv8m.main-none-none-eabi -mattr=+dsp < %s -arm-parallel-dsp -S | FileCheck %s
   2 ;
   3 ; The Cortex-M0 does not support unaligned accesses:
   4 ; RUN: opt -mtriple=armv6m-none-none-eabi < %s -arm-parallel-dsp -S | FileCheck %s --check-prefix=CHECK-UNSUPPORTED
   5 ; RUN: opt -mtriple=armv6meb-none-eabi-eabi < %s -arm-parallel-dsp -S | FileCheck %s --check-prefix=CHECK-UNSUPPORTED
   6 ;
   7 ; Check DSP extension:
   8 ; RUN: opt -mtriple=armv8m.main-none-none-eabi -mattr=-dsp < %s -arm-parallel-dsp -S | FileCheck %s --check-prefix=CHECK-UNSUPPORTED
   9
  10 define dso_local i32 @OneReduction(i32 %arg, ptr nocapture readnone %arg1, ptr nocapture readonly %arg2, ptr nocapture readonly %arg3) {
  11 ;
  12 ; CHECK-LABEL: @OneReduction
  13 ; CHECK:  %mac1{{\.}}026 = phi i32 [ [[V8:%[0-9]+]], %for.body ], [ 0, %for.body.preheader ]
  14 ; CHECK:  [[V5:%[0-9]+]] = load i32, ptr %arrayidx, align 2
  15 ; CHECK:  [[V7:%[0-9]+]] = load i32, ptr %arrayidx3, align 2
  16 ; CHECK:  [[V8]] = call i32 @llvm.arm.smlad(i32 [[V7]], i32 [[V5]], i32 %mac1{{\.}}026)
  17 ; CHECK-NOT: call i32 @llvm.arm.smlad
  18 ;
  19 ; CHECK-UNSUPPORTED-NOT:  call i32 @llvm.arm.smlad
  20 ;
  21 entry:
  22   %cmp24 = icmp sgt i32 %arg, 0
  23   br i1 %cmp24, label %for.body.preheader, label %for.cond.cleanup
  24
  25 for.body.preheader:
  26   %.pre = load i16, ptr %arg3, align 2
  27   %.pre27 = load i16, ptr %arg2, align 2
  28   br label %for.body
  29
  30 for.cond.cleanup:
  31   %mac1.0.lcssa = phi i32 [ 0, %entry ], [ %add11, %for.body ]
  32   ret i32 %mac1.0.lcssa
  33
  34 for.body:
  35 ; One reduction statement here:
  36   %mac1.026 = phi i32 [ %add11, %for.body ], [ 0, %for.body.preheader ]
  37
  38   %i.025 = phi i32 [ %add, %for.body ], [ 0, %for.body.preheader ]
  39   %arrayidx = getelementptr inbounds i16, ptr %arg3, i32 %i.025
  40   %0 = load i16, ptr %arrayidx, align 2
  41   %add = add nuw nsw i32 %i.025, 1
  42   %arrayidx1 = getelementptr inbounds i16, ptr %arg3, i32 %add
  43   %1 = load i16, ptr %arrayidx1, align 2
  44   %arrayidx3 = getelementptr inbounds i16, ptr %arg2, i32 %i.025
  45   %2 = load i16, ptr %arrayidx3, align 2
  46   %conv = sext i16 %2 to i32
  47   %conv4 = sext i16 %0 to i32
  48   %mul = mul nsw i32 %conv, %conv4
  49   %arrayidx6 = getelementptr inbounds i16, ptr %arg2, i32 %add
  50   %3 = load i16, ptr %arrayidx6, align 2
  51   %conv7 = sext i16 %3 to i32
  52   %conv8 = sext i16 %1 to i32
  53   %mul9 = mul nsw i32 %conv7, %conv8
  54   %add10 = add i32 %mul, %mac1.026
  55
  56 ; Here the Mul is the LHS, and the Add the RHS.
  57   %add11 = add i32 %mul9, %add10
  58
  59   %exitcond = icmp ne i32 %add, %arg
  60   br i1 %exitcond, label %for.body, label %for.cond.cleanup
  61 }
  62
  63 define dso_local arm_aapcs_vfpcc i32 @TwoReductions(i32 %arg, ptr nocapture readnone %arg1, ptr nocapture readonly %arg2, ptr nocapture readonly %arg3) {
  64 ;
  65 ; CHECK-LABEL: @TwoReductions
  66 ;
  67 ; CHECK:  %mac1{{\.}}058 = phi i32 [ [[V10:%[0-9]+]], %for.body ], [ 0, %for.body.preheader ]
  68 ; CHECK:  %mac2{{\.}}057 = phi i32 [ [[V17:%[0-9]+]], %for.body ], [ 0, %for.body.preheader ]
  69 ; CHECK:  [[V10]] = call i32 @llvm.arm.smlad(i32 %{{.*}}, i32 %{{.*}}, i32 %mac1{{\.}}058)
  70 ; CHECK:  [[V17]] = call i32 @llvm.arm.smlad(i32 %{{.*}}, i32 %{{.*}}, i32 %mac2{{\.}}057)
  71 ; CHECK-NOT: call i32 @llvm.arm.smlad
  72 ;
  73 entry:
  74   %cmp55 = icmp sgt i32 %arg, 0
  75   br i1 %cmp55, label %for.body.preheader, label %for.cond.cleanup
  76
  77 for.cond.cleanup:
  78   %mac2.0.lcssa = phi i32 [ 0, %entry ], [ %add28, %for.body ]
  79   %mac1.0.lcssa = phi i32 [ 0, %entry ], [ %add16, %for.body ]
  80   %add30 = add nsw i32 %mac1.0.lcssa, %mac2.0.lcssa
  81   ret i32 %add30
  82
  83 for.body.preheader:
  84   br label %for.body
  85
  86 for.body:
  87 ; And two reduction statements here:
  88   %mac1.058 = phi i32 [ %add16, %for.body ], [ 0, %for.body.preheader ]
  89   %mac2.057 = phi i32 [ %add28, %for.body ], [ 0, %for.body.preheader ]
  90
  91   %i.056 = phi i32 [ %add29, %for.body ], [ 0, %for.body.preheader ]
  92   %arrayidx = getelementptr inbounds i16, ptr %arg3, i32 %i.056
  93   %0 = load i16, ptr %arrayidx, align 2
  94   %add1 = or i32 %i.056, 1
  95   %arrayidx2 = getelementptr inbounds i16, ptr %arg3, i32 %add1
  96   %1 = load i16, ptr %arrayidx2, align 2
  97   %add3 = or i32 %i.056, 2
  98   %arrayidx4 = getelementptr inbounds i16, ptr %arg3, i32 %add3
  99   %2 = load i16, ptr %arrayidx4, align 2
 100
 101   %add5 = or i32 %i.056, 3
 102   %arrayidx6 = getelementptr inbounds i16, ptr %arg3, i32 %add5
 103   %3 = load i16, ptr %arrayidx6, align 2
 104   %arrayidx8 = getelementptr inbounds i16, ptr %arg2, i32 %i.056
 105   %4 = load i16, ptr %arrayidx8, align 2
 106   %conv = sext i16 %4 to i32
 107   %conv9 = sext i16 %0 to i32
 108   %mul = mul nsw i32 %conv, %conv9
 109   %arrayidx11 = getelementptr inbounds i16, ptr %arg2, i32 %add1
 110   %5 = load i16, ptr %arrayidx11, align 2
 111   %conv12 = sext i16 %5 to i32
 112   %conv13 = sext i16 %1 to i32
 113   %mul14 = mul nsw i32 %conv12, %conv13
 114   %add15 = add i32 %mul, %mac1.058
 115   %add16 = add i32 %add15, %mul14
 116   %arrayidx18 = getelementptr inbounds i16, ptr %arg2, i32 %add3
 117   %6 = load i16, ptr %arrayidx18, align 2
 118   %conv19 = sext i16 %6 to i32
 119   %conv20 = sext i16 %2 to i32
 120   %mul21 = mul nsw i32 %conv19, %conv20
 121   %arrayidx23 = getelementptr inbounds i16, ptr %arg2, i32 %add5
 122   %7 = load i16, ptr %arrayidx23, align 2
 123   %conv24 = sext i16 %7 to i32
 124   %conv25 = sext i16 %3 to i32
 125   %mul26 = mul nsw i32 %conv24, %conv25
 126   %add27 = add i32 %mul21, %mac2.057
 127   %add28 = add i32 %add27, %mul26
 128   %add29 = add nuw nsw i32 %i.056, 4
 129   %cmp = icmp slt i32 %add29, %arg
 130   br i1 %cmp, label %for.body, label %for.cond.cleanup
 131 }
 132
 133 define i32 @one_zext(i32 %arg, ptr nocapture readnone %arg1, ptr nocapture readonly %arg2, ptr nocapture readonly %arg3) {
 134 ; CHECK-LABEL: @one_zext
 135 ; CHECK-NOT: call i32 @llvm.arm.smlad
 136 entry:
 137   %cmp24 = icmp sgt i32 %arg, 0
 138   br i1 %cmp24, label %for.body.preheader, label %for.cond.cleanup
 139
 140 for.body.preheader:
 141   %.pre = load i16, ptr %arg3, align 2
 142   %.pre27 = load i16, ptr %arg2, align 2
 143   br label %for.body
 144
 145 for.cond.cleanup:
 146   %mac1.0.lcssa = phi i32 [ 0, %entry ], [ %add11, %for.body ]
 147   ret i32 %mac1.0.lcssa
 148
 149 for.body:
 150   %mac1.026 = phi i32 [ %add11, %for.body ], [ 0, %for.body.preheader ]
 151   %i.025 = phi i32 [ %add, %for.body ], [ 0, %for.body.preheader ]
 152   %arrayidx = getelementptr inbounds i16, ptr %arg3, i32 %i.025
 153   %0 = load i16, ptr %arrayidx, align 2
 154   %add = add nuw nsw i32 %i.025, 1
 155   %arrayidx1 = getelementptr inbounds i16, ptr %arg3, i32 %add
 156   %1 = load i16, ptr %arrayidx1, align 2
 157   %arrayidx3 = getelementptr inbounds i16, ptr %arg2, i32 %i.025
 158   %2 = load i16, ptr %arrayidx3, align 2
 159   %conv = sext i16 %2 to i32
 160   %conv4 = zext i16 %0 to i32
 161   %mul = mul nsw i32 %conv, %conv4
 162   %arrayidx6 = getelementptr inbounds i16, ptr %arg2, i32 %add
 163   %3 = load i16, ptr %arrayidx6, align 2
 164   %conv7 = sext i16 %3 to i32
 165   %conv8 = zext i16 %1 to i32
 166   %mul9 = mul nsw i32 %conv7, %conv8
 167   %add10 = add i32 %mul, %mac1.026
 168   %add11 = add i32 %mul9, %add10
 169   %exitcond = icmp ne i32 %add, %arg
 170   br i1 %exitcond, label %for.body, label %for.cond.cleanup
 171 }
 172
 173 define i32 @two_zext(i32 %arg, ptr nocapture readnone %arg1, ptr nocapture readonly %arg2, ptr nocapture readonly %arg3) {
 174 ; CHECK-LABEL: @two_zext
 175 ; CHECK-NOT: call i32 @llvm.arm.smlad
 176 entry:
 177   %cmp24 = icmp sgt i32 %arg, 0
 178   br i1 %cmp24, label %for.body.preheader, label %for.cond.cleanup
 179
 180 for.body.preheader:
 181   %.pre = load i16, ptr %arg3, align 2
 182   %.pre27 = load i16, ptr %arg2, align 2
 183   br label %for.body
 184
 185 for.cond.cleanup:
 186   %mac1.0.lcssa = phi i32 [ 0, %entry ], [ %add11, %for.body ]
 187   ret i32 %mac1.0.lcssa
 188
 189 for.body:
 190   %mac1.026 = phi i32 [ %add11, %for.body ], [ 0, %for.body.preheader ]
 191   %i.025 = phi i32 [ %add, %for.body ], [ 0, %for.body.preheader ]
 192   %arrayidx = getelementptr inbounds i16, ptr %arg3, i32 %i.025
 193   %0 = load i16, ptr %arrayidx, align 2
 194   %add = add nuw nsw i32 %i.025, 1
 195   %arrayidx1 = getelementptr inbounds i16, ptr %arg3, i32 %add
 196   %1 = load i16, ptr %arrayidx1, align 2
 197   %arrayidx3 = getelementptr inbounds i16, ptr %arg2, i32 %i.025
 198   %2 = load i16, ptr %arrayidx3, align 2
 199   %conv = zext i16 %2 to i32
 200   %conv4 = zext i16 %0 to i32
 201   %mul = mul nsw i32 %conv, %conv4
 202   %arrayidx6 = getelementptr inbounds i16, ptr %arg2, i32 %add
 203   %3 = load i16, ptr %arrayidx6, align 2
 204   %conv7 = zext i16 %3 to i32
 205   %conv8 = zext i16 %1 to i32
 206   %mul9 = mul nsw i32 %conv7, %conv8
 207   %add10 = add i32 %mul, %mac1.026
 208   %add11 = add i32 %mul9, %add10
 209   %exitcond = icmp ne i32 %add, %arg
 210   br i1 %exitcond, label %for.body, label %for.cond.cleanup
 211 }
 212