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

+improved support for dp

Browse Source
This commit is contained in:
Evghenii
2014-02-02 18:51:17 +01:00
parent 796942e6fa
commit 92e69c8197
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples_ptx/nbody/nbody.cu
View File

@@ -118,7 +118,7 @@ void computeForces(
const real dy = jposy - iposy;
const real dz = jposz - iposz;
const real r2 = dx*dx + dy*dy + dz*dz;
const real rinv = r2 > 0.0 ? rsqrt((float)r2) : 0;
const real rinv = r2 > 0.0 ? 1.0/sqrt(r2) : 0;
const real mrinv = -jmass * rinv;
const real mrinv3 = mrinv * rinv*rinv;
iaccx += mrinv3 * dx;

6
examples_ptx/nbody/nbody.ispc
View File

@@ -41,6 +41,7 @@ void computeForces(
const uniform int blockDim = (nbodies + taskCount - 1)/taskCount;
const uniform int blockBeg = blockIdx * blockDim;
const uniform int blockEnd = min(blockBeg + blockDim, nbodies);
uniform real shmem[4][programCount];
//real gpotLoc = 0;
foreach (i = blockBeg ... blockEnd)
@@ -63,7 +64,7 @@ void computeForces(
const real dy = jposy - iposy;
const real dz = jposz - iposz;
const real r2 = dx*dx + dy*dy + dz*dz;
const real rinv = r2> 0.0d ? rsqrt((float)r2) : 0;
const real rinv = r2> 0.0d ? 1.0/sqrt(r2) : 0;
const real mrinv = -jmass * rinv;
const real mrinv3 = mrinv * rinv*rinv;
@@ -76,7 +77,6 @@ void computeForces(
for (uniform int j = 0; j < nbodies; j += programCount)
{
#if 1
uniform real shmem[4][programCount];
shmem[0][programIndex] = posx[j+programIndex];
shmem[1][programIndex] = posy[j+programIndex];
shmem[2][programIndex] = posz[j+programIndex];
@@ -104,7 +104,7 @@ void computeForces(
const real dy = jposy - iposy;
const real dz = jposz - iposz;
const real r2 = dx*dx + dy*dy + dz*dz;
const real rinv = r2 > 0.0d ? rsqrt((float)r2) : 0;
const real rinv = r2 > 0.0d ? 1.0/sqrt(r2) : 0;
const real mrinv = -jmass * rinv;
const real mrinv3 = mrinv * rinv*rinv;