Added a small change to the way pointer arguments are set in the kernel code generation. The way the pointer is retrieved now, specifically requests global address space to be annotated. This is necessary, if the IR should be run through NVPTX to generate OpenCL compatible PTX. The changes do not affect the PTX Strings generated for the CUDA target (nvptx64-nvidia-cuda), but are necessary for OpenCL (nvptx64-nvidia-nvcl). Additionally, the data layout has been updated to what the NVPTX Backend requests/recommends. Contributed-by: Philipp Schaad Reviewers: Meinersbur, grosser, bollu Reviewed By: grosser, bollu Subscribers: jlebar, pollydev, llvm-commits, nemanjai, yaxunl, Anastasia Tags: #polly Differential Revision: https://reviews.llvm.org/D32215 llvm-svn: 301299
84 lines
3.2 KiB
LLVM
84 lines
3.2 KiB
LLVM
; RUN: opt %loadPolly -polly-codegen-ppcg -polly-acc-dump-code \
|
|
; RUN: -polly-acc-use-shared \
|
|
; RUN: -disable-output < %s | \
|
|
; RUN: FileCheck -check-prefix=CODE %s
|
|
|
|
; RUN: opt %loadPolly -polly-codegen-ppcg \
|
|
; RUN: -polly-acc-use-shared \
|
|
; RUN: -disable-output -polly-acc-dump-kernel-ir < %s | \
|
|
; RUN: FileCheck -check-prefix=KERNEL %s
|
|
|
|
; REQUIRES: pollyacc
|
|
|
|
; void add(float *A) {
|
|
; for (long i = 0; i < 32; i++)
|
|
; for (long j = 0; j < 10; j++)
|
|
; A[i] += 1;
|
|
; }
|
|
|
|
; CODE: # kernel0
|
|
; CODE: {
|
|
; CODE: read(t0);
|
|
; CODE: sync0();
|
|
; CODE: for (int c3 = 0; c3 <= 9; c3 += 1)
|
|
; CODE: Stmt_bb5(t0, c3);
|
|
; CODE: sync1();
|
|
; CODE: write(t0);
|
|
; CODE: }
|
|
|
|
; KERNEL: @shared_MemRef_A = internal addrspace(3) global [32 x float] zeroinitializer, align 4
|
|
|
|
; KERNEL: %polly.access.shared_MemRef_A = getelementptr float, float addrspace(3)* getelementptr inbounds ([32 x float], [32 x float] addrspace(3)* @shared_MemRef_A, i32 0, i32 0), i64 %t0
|
|
; KERNEL-NEXT: %polly.access.cast.MemRef_A = bitcast i8 addrspace(1)* %MemRef_A to float addrspace(1)*
|
|
; KERNEL-NEXT: %polly.access.MemRef_A = getelementptr float, float addrspace(1)* %polly.access.cast.MemRef_A, i64 %t0
|
|
; KERNEL-NEXT: %shared.read = load float, float addrspace(1)* %polly.access.MemRef_A
|
|
; KERNEL-NEXT: store float %shared.read, float addrspace(3)* %polly.access.shared_MemRef_A
|
|
|
|
; KERNEL: %polly.access.shared_MemRef_A3 = getelementptr float, float addrspace(3)* getelementptr inbounds ([32 x float], [32 x float] addrspace(3)* @shared_MemRef_A, i32 0, i32 0), i64 %t0
|
|
; KERNEL-NEXT: %polly.access.cast.MemRef_A4 = bitcast i8 addrspace(1)* %MemRef_A to float addrspace(1)*
|
|
; KERNEL-NEXT: %polly.access.MemRef_A5 = getelementptr float, float addrspace(1)* %polly.access.cast.MemRef_A4, i64 %t0
|
|
; KERNEL-NEXT: %shared.write = load float, float addrspace(3)* %polly.access.shared_MemRef_A3
|
|
; KERNEL-NEXT: store float %shared.write, float addrspace(1)* %polly.access.MemRef_A5
|
|
|
|
|
|
target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
|
|
|
|
define void @add(float* %A) {
|
|
bb:
|
|
br label %bb2
|
|
|
|
bb2: ; preds = %bb11, %bb
|
|
%i.0 = phi i64 [ 0, %bb ], [ %tmp12, %bb11 ]
|
|
%exitcond1 = icmp ne i64 %i.0, 32
|
|
br i1 %exitcond1, label %bb3, label %bb13
|
|
|
|
bb3: ; preds = %bb2
|
|
br label %bb4
|
|
|
|
bb4: ; preds = %bb8, %bb3
|
|
%j.0 = phi i64 [ 0, %bb3 ], [ %tmp9, %bb8 ]
|
|
%exitcond = icmp ne i64 %j.0, 10
|
|
br i1 %exitcond, label %bb5, label %bb10
|
|
|
|
bb5: ; preds = %bb4
|
|
%tmp = getelementptr inbounds float, float* %A, i64 %i.0
|
|
%tmp6 = load float, float* %tmp, align 4
|
|
%tmp7 = fadd float %tmp6, 1.000000e+00
|
|
store float %tmp7, float* %tmp, align 4
|
|
br label %bb8
|
|
|
|
bb8: ; preds = %bb5
|
|
%tmp9 = add nuw nsw i64 %j.0, 1
|
|
br label %bb4
|
|
|
|
bb10: ; preds = %bb4
|
|
br label %bb11
|
|
|
|
bb11: ; preds = %bb10
|
|
%tmp12 = add nuw nsw i64 %i.0, 1
|
|
br label %bb2
|
|
|
|
bb13: ; preds = %bb2
|
|
ret void
|
|
}
|