Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2024-12-27 02:09:54 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

GPGPU: Allow region statements

Browse Source 
llvm-svn: 281305
This commit is contained in:
Tobias Grosser 2016-09-13 08:42:10 +00:00
parent 05453991fe
commit a82c4b5df8
2 changed files with 87 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
polly/lib/CodeGen/PPCGCodeGeneration.cpp
View File

@ -834,7 +834,7 @@ void GPUNodeBuilder::createScopStmt(isl_ast_expr *Expr,
if (Stmt->isBlockStmt())
BlockGen.copyStmt(*Stmt, LTS, Indexes);
else
assert(0 && "Region statement not supported\n");
RegionGen.copyStmt(*Stmt, LTS, Indexes);
}
void GPUNodeBuilder::createKernelSync() {

86
polly/test/GPGPU/region-stmt.ll Normal file
View File

@ -0,0 +1,86 @@
; RUN: opt %loadPolly -polly-codegen-ppcg -polly-acc-dump-code \
; RUN: -disable-output < %s | \
; RUN: FileCheck -check-prefix=CODE %s
; RUN: opt %loadPolly -polly-codegen-ppcg -S < %s | \
; RUN: FileCheck %s -check-prefix=IR
; CODE: Code
; CODE-NEXT: ====
; CODE-NEXT: # host
; CODE-NEXT: {
; CODE-NEXT: cudaCheckReturn(cudaMemcpy(dev_MemRef_A, MemRef_A, (128) * sizeof(float), cudaMemcpyHostToDevice));
; CODE-NEXT: cudaCheckReturn(cudaMemcpy(dev_MemRef_B, MemRef_B, (128) * sizeof(float), cudaMemcpyHostToDevice));
; CODE-NEXT: {
; CODE-NEXT: dim3 k0_dimBlock(32);
; CODE-NEXT: dim3 k0_dimGrid(4);
; CODE-NEXT: kernel0 <<<k0_dimGrid, k0_dimBlock>>> (dev_MemRef_A, dev_MemRef_B);
; CODE-NEXT: cudaCheckKernel();
; CODE-NEXT: }
; CODE: cudaCheckReturn(cudaMemcpy(MemRef_B, dev_MemRef_B, (128) * sizeof(float), cudaMemcpyDeviceToHost));
; CODE-NEXT: }
; CODE: # kernel0
; CODE-NEXT: Stmt_for_body__TO__if_end(32 * b0 + t0);
; IR: @polly_initContext
; KERNEL-IR: kernel_0
; REQUIRES: pollyacc
; void foo(float A[], float B[]) {
; for (long i = 0; i < 128; i++)
; if (A[i] == 42)
; B[i] += 2 * i;
; else
; B[i] += 4 * i;
; }
;
source_filename = "/tmp/test.c"
target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
define void @foo(float* %A, float* %B) {
entry:
br label %for.cond
for.cond: ; preds = %for.inc, %entry
%i.0 = phi i64 [ 0, %entry ], [ %inc, %for.inc ]
%exitcond = icmp ne i64 %i.0, 128
br i1 %exitcond, label %for.body, label %for.end
for.body: ; preds = %for.cond
%arrayidx = getelementptr inbounds float, float* %A, i64 %i.0
%tmp = load float, float* %arrayidx, align 4
%cmp1 = fcmp oeq float %tmp, 4.200000e+01
br i1 %cmp1, label %if.then, label %if.else
if.then: ; preds = %for.body
%mul = shl nsw i64 %i.0, 1
%conv = sitofp i64 %mul to float
%arrayidx2 = getelementptr inbounds float, float* %B, i64 %i.0
%tmp1 = load float, float* %arrayidx2, align 4
%add = fadd float %tmp1, %conv
store float %add, float* %arrayidx2, align 4
br label %if.end
if.else: ; preds = %for.body
%mul3 = shl nsw i64 %i.0, 2
%conv4 = sitofp i64 %mul3 to float
%arrayidx5 = getelementptr inbounds float, float* %B, i64 %i.0
%tmp2 = load float, float* %arrayidx5, align 4
%add6 = fadd float %tmp2, %conv4
store float %add6, float* %arrayidx5, align 4
br label %if.end
if.end: ; preds = %if.else, %if.then
br label %for.inc
for.inc: ; preds = %if.end
%inc = add nuw nsw i64 %i.0, 1
br label %for.cond
for.end: ; preds = %for.cond
ret void
}