aaron/ispc
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ISPC sync is not added

Browse Source
This commit is contained in:
Evghenii
2013-11-12 17:02:31 +01:00
parent dbde936c3c
commit 1d91a626f2
2 changed files with 31 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
builtins/target-nvptx64.ll
View File

@@ -24,8 +24,9 @@ define i32 @__tid_x() nounwind readnone alwaysinline
}
define i32 @__warpsize() nounwind readnone alwaysinline
{
%tid = call i32 @llvm.nvvm.read.ptx.sreg.warpsize()
ret i32 %tid
;; %tid = call i32 @llvm.nvvm.read.ptx.sreg.warpsize()
;; ret i32 %tid
ret i32 32
}
@@ -116,14 +117,26 @@ define i8* @ISPCAlloc(i8**, i64, i32) nounwind alwaysinline
}
define void @ISPCLaunch(i8**, i8* %func_ptr, i8** %func_args, i32 %ntx, i32 %nty, i32 %ntz) nounwind alwaysinline
{
entry:
%func_i64 = ptrtoint i8* %func_ptr to i64
%args_i64 = ptrtoint i8** %func_args to i64
;; nbx = (%ntx-1)/(blocksize/warpsize) + 1 for blocksize=128 & warpsize=32
%sub = add nsw i32 %ntx, -1
%div = sdiv i32 %sub, 4
%nbx = add nsw i32 %div, 1
%ntxm1 = add nsw i32 %ntx, -1
%ntxm1d4 = sdiv i32 %ntxm1, 4
%nbx = add nsw i32 %ntxm1d4, 1
%res = call i32 asm sideeffect "{
;; only 1 lane must launch the kernel !!!
%call = tail call i32 @__tid_x()
%call1 = tail call i32 @__warpsize()
%sub = add nsw i32 %call1, -1
%and = and i32 %sub, %call
;; if (laneIdx == 0)
%cmp = icmp eq i32 %and, 0
br i1 %cmp, label %if.then, label %if.end
if.then: ; preds = %entry
%res_tmp = call i32 asm sideeffect "{
.reg .s32 %r<8>;
.reg .s64 %rd<3>;
.param .b64 param0;
@@ -158,6 +171,11 @@ define void @ISPCLaunch(i8**, i8* %func_ptr, i8** %func_args, i32 %ntx, i32 %nty
}
",
"=r, l,l, r,r,r, r,r,r, r,l"(i64 %func_i64,i64 %args_i64, i32 %nbx,i32 %nty,i32 %ntz, i32 128,i32 1,i32 1, i32 0,i64 0);
br label %if.end
if.end: ; preds = %if.then, %entry
%res = phi i32 [ %res_tmp, %if.then ], [ undef, %entry ]
ret void
}
declare i32 @cudaDeviceSynchronize() nounwind

9
examples_cuda/aobench/ao1.ispc
View File

@@ -286,7 +286,11 @@ static inline void ao_tile(
#define TILEX 64
#define TILEY 4
/* task will generate " __global__ " only */
/* unless task/export is specified all functions
* are generated as mangled "__device__" functions
*/
/* task will generate mangled "__global__" function only */
void task ao_task(uniform int width, uniform int height,
uniform int nsubsamples, uniform float image[])
{
@@ -302,11 +306,12 @@ void task ao_task(uniform int width, uniform int height,
}
/* export will generate "extern "C" __global__" an "__device__" */
/* export will generate unmangled "extern "C" __global__" and mangled "__device__" */
export void ao_ispc_tasks(uniform int w, uniform int h, uniform int nsubsamples,
uniform float image[])
{
const uniform int ntilex = (w+TILEX-1)/TILEX;
const uniform int ntiley = (h+TILEY-1)/TILEY;
launch[ntilex,ntiley] ao_task(w, h, nsubsamples, image);
sync;
}