test/Transforms/LoopVectorize/X86/mask3.ll

   1 ; RUN: opt < %s  -O3 -mcpu=corei7-avx -S | FileCheck %s -check-prefix=AVX1
   2 ; RUN: opt < %s  -O3 -mcpu=core-avx2 -S | FileCheck %s -check-prefix=AVX2
   3 ; RUN: opt < %s  -O3 -mcpu=knl -S | FileCheck %s -check-prefix=AVX512
   4
   5 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
   6 target triple = "x86_64-pc_linux"
   7
   8 ; The source code:
   9 ;
  10 ;void foo(double *A, double *B, int *trigger) {
  11 ;
  12 ;  for (int i=0; i<10000; i++) {
  13 ;    if (trigger[i] < 100) {
  14 ;          A[i] = B[i] + trigger[i];
  15 ;    }
  16 ;  }
  17 ;}
  18
  19
  20 ;AVX2: llvm.masked.load.v4f64
  21 ;AVX2: llvm.masked.store.v4f64
  22 ;AVX512: llvm.masked.load.v8f64
  23 ;AVX512: llvm.masked.store.v8f64
  24 ;AVX1-NOT: llvm.masked
  25
  26 ; Function Attrs: nounwind uwtable
  27 define void @foo(double* %A, double* %B, i32* %trigger) #0 {
  28 entry:
  29   %A.addr = alloca double*, align 8
  30   %B.addr = alloca double*, align 8
  31   %trigger.addr = alloca i32*, align 8
  32   %i = alloca i32, align 4
  33   store double* %A, double** %A.addr, align 8
  34   store double* %B, double** %B.addr, align 8
  35   store i32* %trigger, i32** %trigger.addr, align 8
  36   store i32 0, i32* %i, align 4
  37   br label %for.cond
  38
  39 for.cond:                                         ; preds = %for.inc, %entry
  40   %0 = load i32* %i, align 4
  41   %cmp = icmp slt i32 %0, 10000
  42   br i1 %cmp, label %for.body, label %for.end
  43
  44 for.body:                                         ; preds = %for.cond
  45   %1 = load i32* %i, align 4
  46   %idxprom = sext i32 %1 to i64
  47   %2 = load i32** %trigger.addr, align 8
  48   %arrayidx = getelementptr inbounds i32* %2, i64 %idxprom
  49   %3 = load i32* %arrayidx, align 4
  50   %cmp1 = icmp slt i32 %3, 100
  51   br i1 %cmp1, label %if.then, label %if.end
  52
  53 if.then:                                          ; preds = %for.body
  54   %4 = load i32* %i, align 4
  55   %idxprom2 = sext i32 %4 to i64
  56   %5 = load double** %B.addr, align 8
  57   %arrayidx3 = getelementptr inbounds double* %5, i64 %idxprom2
  58   %6 = load double* %arrayidx3, align 8
  59   %7 = load i32* %i, align 4
  60   %idxprom4 = sext i32 %7 to i64
  61   %8 = load i32** %trigger.addr, align 8
  62   %arrayidx5 = getelementptr inbounds i32* %8, i64 %idxprom4
  63   %9 = load i32* %arrayidx5, align 4
  64   %conv = sitofp i32 %9 to double
  65   %add = fadd double %6, %conv
  66   %10 = load i32* %i, align 4
  67   %idxprom6 = sext i32 %10 to i64
  68   %11 = load double** %A.addr, align 8
  69   %arrayidx7 = getelementptr inbounds double* %11, i64 %idxprom6
  70   store double %add, double* %arrayidx7, align 8
  71   br label %if.end
  72
  73 if.end:                                           ; preds = %if.then, %for.body
  74   br label %for.inc
  75
  76 for.inc:                                          ; preds = %if.end
  77   %12 = load i32* %i, align 4
  78   %inc = add nsw i32 %12, 1
  79   store i32 %inc, i32* %i, align 4
  80   br label %for.cond
  81
  82 for.end:                                          ; preds = %for.cond
  83   ret void
  84 }