[RISCV] Fix mgather -> riscv.masked.strided.load combine not extending indices (...
[llvm-project.git] / polly / test / CodeGen / OpenMP / new_multidim_access.ll
blob1b2912ba9ef46268d7d9b639dee8487cfaa140d2
1 ; RUN: opt %loadPolly -polly-print-import-jscop \
2 ; RUN:                -disable-output < %s | FileCheck %s
4 ; RUN: opt %loadPolly -polly-import-jscop \
5 ; RUN:                -polly-codegen -S < %s \
6 ; RUN:                -polly-parallel \
7 ; RUN:                | FileCheck %s -check-prefix=IR
9 ;    void new_multidim_access(long n, long m, float A[][m]) {
10 ;      for (long i = 0; i < n; i++)
11 ;        for (long j = 0; j < 100; j++)
12 ;          A[i][2 * j] += i + j;
13 ;    }
15 ; CHECK:  ReadAccess :=       [Reduction Type: NONE] [Scalar: 0]
16 ; CHECK:         [n, m] -> { Stmt_bb4[i0, i1] -> MemRef_A[i0, 2i1] };
17 ; CHECK:    new: [n, m] -> { Stmt_bb4[i0, i1] -> MemRef_A[i0, 13 + i1] };
18 ; CHECK:  MustWriteAccess :=  [Reduction Type: NONE] [Scalar: 0]
19 ; CHECK:         [n, m] -> { Stmt_bb4[i0, i1] -> MemRef_A[i0, 2i1] };
20 ; CHECK:    new: [n, m] -> { Stmt_bb4[i0, i1] -> MemRef_A[i0, 43 + i1] };
22 ; IR: %polly.access.mul.polly.subfunc.arg.A = mul nsw i64 %polly.indvar, %polly.subfunc.arg.m
23 ; IR: %6 = add nsw i64 %polly.indvar4, 13
24 ; IR: %polly.access.add.polly.subfunc.arg.A = add nsw i64 %polly.access.mul.polly.subfunc.arg.A, %6
25 ; IR: %polly.access.polly.subfunc.arg.A = getelementptr float, ptr %polly.subfunc.arg.A, i64 %polly.access.add.polly.subfunc.arg.A
26 ; IR: %tmp10_p_scalar_ = load float, ptr %polly.access.polly.subfunc.arg.A, align 4, !alias.scope !0, !noalias !3, !llvm.access.group !4
28 ; IR: %polly.access.mul.polly.subfunc.arg.A7 = mul nsw i64 %polly.indvar, %polly.subfunc.arg.m
29 ; IR: %7 = add nsw i64 %polly.indvar4, 43
30 ; IR: %polly.access.add.polly.subfunc.arg.A8 = add nsw i64 %polly.access.mul.polly.subfunc.arg.A7, %7
31 ; IR: %polly.access.polly.subfunc.arg.A9 = getelementptr float, ptr %polly.subfunc.arg.A, i64 %polly.access.add.polly.subfunc.arg.A8
32 ; IR: store float %p_tmp11, ptr %polly.access.polly.subfunc.arg.A9, align 4, !alias.scope !0, !noalias !3, !llvm.access.group !4
33 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
35 define void @new_multidim_access(i64 %n, i64 %m, ptr %A) {
36 bb:
37   br label %bb1
39 bb1:                                              ; preds = %bb15, %bb
40   %i.0 = phi i64 [ 0, %bb ], [ %tmp16, %bb15 ]
41   %tmp = icmp slt i64 %i.0, %n
42   br i1 %tmp, label %bb2, label %bb17
44 bb2:                                              ; preds = %bb1
45   br label %bb3
47 bb3:                                              ; preds = %bb12, %bb2
48   %j.0 = phi i64 [ 0, %bb2 ], [ %tmp13, %bb12 ]
49   %exitcond = icmp ne i64 %j.0, 100
50   br i1 %exitcond, label %bb4, label %bb14
52 bb4:                                              ; preds = %bb3
53   %tmp5 = add nsw i64 %i.0, %j.0
54   %tmp6 = sitofp i64 %tmp5 to float
55   %tmp7 = shl nsw i64 %j.0, 1
56   %tmp8 = mul nsw i64 %i.0, %m
57   %.sum = add i64 %tmp8, %tmp7
58   %tmp9 = getelementptr inbounds float, ptr %A, i64 %.sum
59   %tmp10 = load float, ptr %tmp9, align 4
60   %tmp11 = fadd float %tmp10, %tmp6
61   store float %tmp11, ptr %tmp9, align 4
62   br label %bb12
64 bb12:                                             ; preds = %bb4
65   %tmp13 = add nsw i64 %j.0, 1
66   br label %bb3
68 bb14:                                             ; preds = %bb3
69   br label %bb15
71 bb15:                                             ; preds = %bb14
72   %tmp16 = add nsw i64 %i.0, 1
73   br label %bb1
75 bb17:                                             ; preds = %bb1
76   ret void