llvm/test/Transforms/LoopVectorize/gep_with_bitcast.ll

   1 ; RUN: opt -S -loop-vectorize -instcombine -force-vector-width=4  < %s | FileCheck %s
   2
   3 target datalayout = "e-m:e-i64:64-i128:128-n32:64-S128"
   4
   5 ; Vectorization of loop with bitcast between GEP and load
   6 ; Simplified source code:
   7 ;void foo (double** __restrict__  in, bool * __restrict__ res) {
   8 ;
   9 ;  for (int i = 0; i < 4096; ++i)
  10 ;    res[i] = ((unsigned long long)in[i] == 0);
  11 ;}
  12
  13 ; CHECK-LABEL: @foo
  14 ; CHECK: vector.body
  15 ; CHECK:  %[[IV:.+]] = phi i64 [ 0, %vector.ph ], [ %index.next, %vector.body ]
  16 ; CHECK:  %[[v0:.+]] = getelementptr inbounds double*, double** %in, i64 %[[IV]]
  17 ; CHECK:  %[[v1:.+]] = bitcast double** %[[v0]] to <4 x i64>*
  18 ; CHECK:  %wide.load = load <4 x i64>, <4 x i64>* %[[v1]], align 8
  19 ; CHECK:  icmp eq <4 x i64> %wide.load, zeroinitializer
  20 ; CHECK:  br i1
  21
  22 define void @foo(double** noalias nocapture readonly %in, double** noalias nocapture readnone %out, i8* noalias nocapture %res) #0 {
  23 entry:
  24   br label %for.body
  25
  26 for.body:
  27   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  28   %arrayidx = getelementptr inbounds double*, double** %in, i64 %indvars.iv
  29   %tmp53 = bitcast double** %arrayidx to i64*
  30   %tmp54 = load i64, i64* %tmp53, align 8
  31   %cmp1 = icmp eq i64 %tmp54, 0
  32   %arrayidx3 = getelementptr inbounds i8, i8* %res, i64 %indvars.iv
  33   %frombool = zext i1 %cmp1 to i8
  34   store i8 %frombool, i8* %arrayidx3, align 1
  35   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  36   %exitcond = icmp eq i64 %indvars.iv.next, 4096
  37   br i1 %exitcond, label %for.end, label %for.body
  38
  39 for.end:
  40   ret void
  41 }