polly/test/CodeGen/MemAccess/create_arrays_heap.ll

   1 ; RUN: opt %loadPolly -polly-stmt-granularity=bb -polly-print-scops -polly-print-import-jscop -polly-import-jscop-postfix=transformed -disable-output < %s | FileCheck %s
   2 ; RUN: opt %loadPolly -polly-stmt-granularity=bb -polly-import-jscop -polly-import-jscop-postfix=transformed -polly-codegen -S < %s | FileCheck %s --check-prefix=CODEGEN
   3 ;
   4 ; #define Ni 1056
   5 ; #define Nj 1056
   6 ; #define Nk 1024
   7 ;
   8 ; void create_arrays_heap(double beta, double A[Ni][Nk], double B[Ni][Nj]) {
   9 ;   int i,j,k;
  10 ;
  11 ;   for (i = 0; i < Ni; i++) {
  12 ;     for (j = 0; j < Nj; j++) {
  13 ;       for (k = 0; k < Nk; ++k) {
  14 ;       B[i][j] = beta * A[i][k];
  15 ;       }
  16 ;     }
  17 ;   }
  18 ; }
  19 ;
  20 ; Check if the info from the JSON file has been analysed without errors.
  21 ; CHECK: Arrays {
  22 ; CHECK: double MemRef_A[*][1024]; // Element size 8
  23 ; CHECK: double MemRef_beta; // Element size 8
  24 ; CHECK: double MemRef_B[*][1056]; // Element size 8
  25 ; CHECK: double D[270336]; // Element size 8
  26 ; CHECK: double E[270336][200000]; // Element size 8
  27 ; CHECK: i64 F[270336]; // Element size 8
  28 ;
  29 ; Check if there are the 3 expected malloc calls with the right parameters at polly.start.
  30 ;       %D : size(D) = product_all_dimensions*sizeof(type) = 270336*8 = 2162688 cast to double*
  31 ;       %E : size(E) = 270336*200000*8 = 432537600000 cast to double*
  32 ;       %F : size(F) = 270336*8 = 2162688 cast to i64*
  33 ; CODEGEN: polly.start:
  34 ; CODEGEN: %D = tail call ptr @malloc(i64 2162688)
  35 ; CODEGEN: %E = tail call ptr @malloc(i64 432537600000)
  36 ; CODEGEN: %F = tail call ptr @malloc(i64 2162688)
  37 ;
  38 ; Check if there are the 3 expected malloc calls with the right parameters at polly.exiting.
  39 ;       Cast to i8* before freeing because malloc give us a i8 and free is waiting for a i8*
  40 ; CODEGEN: polly.exiting:
  41 ; CODEGEN: tail call void @free(ptr %D)
  42 ; CODEGEN: tail call void @free(ptr %E)
  43 ; CODEGEN: tail call void @free(ptr %F)
  44 ;
  45 ; Check if the new access for array E is present.
  46 ; CODEGEN: %polly.access.mul.{{.*}} = mul nsw i64 %polly.indvar, 200000
  47 ; CODEGEN: %polly.access.add.{{.*}} = add nsw i64 %polly.access.mul.{{.*}}, %
  48 ; CODEGEN: %polly.access.{{.*}} = getelementptr double, ptr %E, i64 %polly.access.add.{{.*}}
  49 ;
  50 ; ModuleID = 'create_arrays_heap.ll'
  51 ;
  52 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
  53 target triple = "x86_64-unknown-linux-gnu"
  54
  55 ; Function Attrs: nounwind uwtable
  56 define void @create_arrays_heap(double %beta, ptr nocapture readonly %A, ptr nocapture %B) local_unnamed_addr {
  57 entry:
  58   br label %for.cond1.preheader
  59
  60 for.cond1.preheader:                              ; preds = %for.inc16, %entry
  61   %indvars.iv35 = phi i64 [ 0, %entry ], [ %indvars.iv.next36, %for.inc16 ]
  62   br label %for.cond4.preheader
  63
  64 for.cond4.preheader:                              ; preds = %for.inc13, %for.cond1.preheader
  65   %indvars.iv32 = phi i64 [ 0, %for.cond1.preheader ], [ %indvars.iv.next33, %for.inc13 ]
  66   %arrayidx12 = getelementptr inbounds [1056 x double], ptr %B, i64 %indvars.iv35, i64 %indvars.iv32
  67   br label %for.body6
  68
  69 for.body6:                                        ; preds = %for.body6, %for.cond4.preheader
  70   %indvars.iv = phi i64 [ 0, %for.cond4.preheader ], [ %indvars.iv.next.3, %for.body6 ]
  71   %arrayidx8 = getelementptr inbounds [1024 x double], ptr %A, i64 %indvars.iv35, i64 %indvars.iv
  72   %0 = load double, ptr %arrayidx8, align 8
  73   %mul = fmul double %0, %beta
  74   store double %mul, ptr %arrayidx12, align 8
  75   %indvars.iv.next = or disjoint i64 %indvars.iv, 1
  76   %arrayidx8.1 = getelementptr inbounds [1024 x double], ptr %A, i64 %indvars.iv35, i64 %indvars.iv.next
  77   %1 = load double, ptr %arrayidx8.1, align 8
  78   %mul.1 = fmul double %1, %beta
  79   store double %mul.1, ptr %arrayidx12, align 8
  80   %indvars.iv.next.1 = or disjoint i64 %indvars.iv, 2
  81   %arrayidx8.2 = getelementptr inbounds [1024 x double], ptr %A, i64 %indvars.iv35, i64 %indvars.iv.next.1
  82   %2 = load double, ptr %arrayidx8.2, align 8
  83   %mul.2 = fmul double %2, %beta
  84   store double %mul.2, ptr %arrayidx12, align 8
  85   %indvars.iv.next.2 = or disjoint i64 %indvars.iv, 3
  86   %arrayidx8.3 = getelementptr inbounds [1024 x double], ptr %A, i64 %indvars.iv35, i64 %indvars.iv.next.2
  87   %3 = load double, ptr %arrayidx8.3, align 8
  88   %mul.3 = fmul double %3, %beta
  89   store double %mul.3, ptr %arrayidx12, align 8
  90   %indvars.iv.next.3 = add nsw i64 %indvars.iv, 4
  91   %exitcond.3 = icmp eq i64 %indvars.iv.next.3, 1024
  92   br i1 %exitcond.3, label %for.inc13, label %for.body6
  93
  94 for.inc13:                                        ; preds = %for.body6
  95   %indvars.iv.next33 = add nuw nsw i64 %indvars.iv32, 1
  96   %exitcond34 = icmp eq i64 %indvars.iv.next33, 1056
  97   br i1 %exitcond34, label %for.inc16, label %for.cond4.preheader
  98
  99 for.inc16:                                        ; preds = %for.inc13
 100   %indvars.iv.next36 = add nuw nsw i64 %indvars.iv35, 1
 101   %exitcond37 = icmp eq i64 %indvars.iv.next36, 1056
 102   br i1 %exitcond37, label %for.end18, label %for.cond1.preheader
 103
 104 for.end18:                                        ; preds = %for.inc16
 105   ret void
 106 }
 107