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
5 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
6 target triple = "x86_64-pc_linux"
10 ;void foo(float *A, float *B, int *trigger) {
12 ; for (int i=0; i<10000; i++) {
13 ; if (trigger[i] < 100) {
14 ; A[i] = B[i] + trigger[i];
20 ;AVX2: llvm.masked.load.v8f32
21 ;AVX2: llvm.masked.store.v8f32
22 ;AVX512: llvm.masked.load.v16f32
23 ;AVX512: llvm.masked.store.v16f32
24 ;AVX1-NOT: llvm.masked
26 ; Function Attrs: nounwind uwtable
27 define void @foo(float* %A, float* %B, i32* %trigger) {
29 %A.addr = alloca float*, align 8
30 %B.addr = alloca float*, align 8
31 %trigger.addr = alloca i32*, align 8
32 %i = alloca i32, align 4
33 store float* %A, float** %A.addr, align 8
34 store float* %B, float** %B.addr, align 8
35 store i32* %trigger, i32** %trigger.addr, align 8
36 store i32 0, i32* %i, align 4
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
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
53 if.then: ; preds = %for.body
54 %4 = load i32* %i, align 4
55 %idxprom2 = sext i32 %4 to i64
56 %5 = load float** %B.addr, align 8
57 %arrayidx3 = getelementptr inbounds float* %5, i64 %idxprom2
58 %6 = load float* %arrayidx3, align 4
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 float
65 %add = fadd float %6, %conv
66 %10 = load i32* %i, align 4
67 %idxprom6 = sext i32 %10 to i64
68 %11 = load float** %A.addr, align 8
69 %arrayidx7 = getelementptr inbounds float* %11, i64 %idxprom6
70 store float %add, float* %arrayidx7, align 4
73 if.end: ; preds = %if.then, %for.body
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
82 for.end: ; preds = %for.cond