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

small improvement

Browse Source
This commit is contained in:
Evghenii
2014-01-29 13:49:35 +01:00
parent d4b46b1295
commit 6099492579
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples_ptx/radixSort/radixSort.cu
View File

@@ -103,11 +103,11 @@ void sortPass(
#endif #endif
int scatter;
for (int i = programIndex; i < nloc; i += programCount) for (int i = programIndex; i < nloc; i += programCount)
if (i < nloc) if (i < nloc)
{ {
const int key = mask & ((unsigned int)keys[i] >> bit); const int key = mask & ((unsigned int)keys[i] >> bit);
int scatter;
/* not a vector friendly loop */ /* not a vector friendly loop */
#pragma unroll 1 /* needed, otherwise compiler unroll and optimizes the result :S */ #pragma unroll 1 /* needed, otherwise compiler unroll and optimizes the result :S */
for (int iv = 0; iv < programCount; iv++) for (int iv = 0; iv < programCount; iv++)

3
examples_ptx/radixSort/radixSort.ispc
View File

@@ -196,6 +196,9 @@ export void radixSort_alloc(const uniform int n)
{ {
assert(memoryPool == NULL); assert(memoryPool == NULL);
numBlocks = num_cores()*4; numBlocks = num_cores()*4;
#ifdef __NVPTX__
numBlocks = 13*32*4; //num_cores()*4;
#endif
nSharedCounts = NUMDIGITS*numBlocks; nSharedCounts = NUMDIGITS*numBlocks;
nCountsGlobal = NUMDIGITS; nCountsGlobal = NUMDIGITS;
nExcScan = NUMDIGITS*numBlocks; nExcScan = NUMDIGITS*numBlocks;