polly/test/GPGPU/parametric-loop-bound.ll

   1 ; RUN: opt %loadPolly -polly-codegen-ppcg -polly-acc-dump-code \
   2 ; RUN: -disable-output < %s | \
   3 ; RUN: FileCheck -check-prefix=CODE %s
   4
   5 ; RUN: opt %loadPolly -polly-codegen-ppcg \
   6 ; RUN: -S < %s | \
   7 ; RUN: FileCheck -check-prefix=IR %s
   8
   9 ; REQUIRES: pollyacc
  10
  11 ;    void foo(long A[], long n) {
  12 ;      for (long i = 0; i < n; i++)
  13 ;        A[i] += 100;
  14 ;    }
  15
  16 ; CODE: if (n >= 1) {
  17 ; CODE:        cudaCheckReturn(cudaMemcpy(dev_MemRef_A, MemRef_A, (n) * sizeof(i64), cudaMemcpyHostToDevice));
  18 ; CODE-NEXT:   {
  19 ; CODE-NEXT:     dim3 k0_dimBlock(32);
  20 ; CODE-NEXT:     dim3 k0_dimGrid(n >= 1048545 ? 32768 : (n + 31) / 32);
  21 ; CODE-NEXT:     kernel0 <<<k0_dimGrid, k0_dimBlock>>> (dev_MemRef_A, n);
  22 ; CODE-NEXT:     cudaCheckKernel();
  23 ; CODE-NEXT:   }
  24
  25 ; CODE:        cudaCheckReturn(cudaMemcpy(MemRef_A, dev_MemRef_A, (n) * sizeof(i64), cudaMemcpyDeviceToHost));
  26 ; CODE-NEXT:   cudaCheckReturn(cudaFree(dev_MemRef_A));
  27 ; CODE-NEXT: }
  28
  29 ; CODE: # kernel0
  30 ; CODE-NEXT: for (int c0 = 0; c0 <= (n - 32 * b0 - 1) / 1048576; c0 += 1)
  31 ; CODE-NEXT:   if (n >= 32 * b0 + t0 + 1048576 * c0 + 1)
  32 ; CODE-NEXT:     Stmt_bb2(32 * b0 + t0 + 1048576 * c0);
  33
  34 ; IR: store i64 %n, i64* %polly_launch_0_param_1
  35 ; IR-NEXT: [[REGA:%.+]] = getelementptr [2 x i8*], [2 x i8*]* %polly_launch_0_params, i64 0, i64 1
  36 ; IR-NEXT: [[REGB:%.+]] = bitcast i64* %polly_launch_0_param_1 to i8*
  37 ; IR-NEXT: store i8* [[REGB]], i8** [[REGA]]
  38
  39 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
  40
  41 define void @foo(i64* %A, i64 %n) {
  42 bb:
  43   br label %bb1
  44
  45 bb1:                                              ; preds = %bb6, %bb
  46   %i.0 = phi i64 [ 0, %bb ], [ %tmp7, %bb6 ]
  47   %tmp = icmp slt i64 %i.0, %n
  48   br i1 %tmp, label %bb2, label %bb8
  49
  50 bb2:                                              ; preds = %bb1
  51   %tmp3 = getelementptr inbounds i64, i64* %A, i64 %i.0
  52   %tmp4 = load i64, i64* %tmp3, align 8
  53   %tmp5 = add nsw i64 %tmp4, 100
  54   store i64 %tmp5, i64* %tmp3, align 8
  55   br label %bb6
  56
  57 bb6:                                              ; preds = %bb2
  58   %tmp7 = add nuw nsw i64 %i.0, 1
  59   br label %bb1
  60
  61 bb8:                                              ; preds = %bb1
  62   ret void
  63 }