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

restored nonptx atomic_*_local

Browse Source
This commit is contained in:
Evghenii
2014-01-28 15:56:30 +01:00
parent 585afa09e5
commit 5a6b650d8b
2 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples_ptx/radixSort/radixSort.ispc
View File

@@ -30,7 +30,13 @@ void countPass(
if (reduce_equal(key, &skey) == true) if (reduce_equal(key, &skey) == true)
counts[skey] += reduce_add(1); counts[skey] += reduce_add(1);
else else
{
#ifdef __NVPTX__
atomic_add_global(&counts[key], 1);
#else
atomic_add_local(&counts[key], 1); atomic_add_local(&counts[key], 1);
#endif
}
} }
#else #else
#endif #endif

9
stdlib.ispc
View File

@@ -2088,11 +2088,20 @@ static inline TYPE atomic_##NAME##_local(uniform TYPE * uniform ptr, TYPE value)
} \ } \
static inline TYPE atomic_##NAME##_local(uniform TYPE * p, TYPE value) { \ static inline TYPE atomic_##NAME##_local(uniform TYPE * p, TYPE value) { \
TYPE ret; \ TYPE ret; \
if (__is_nvptx_target) { \
foreach_active (i) { \ foreach_active (i) { \
uniform TYPE * uniform ptr = (uniform TYPE * uniform)extract((int64)p, i); \ uniform TYPE * uniform ptr = (uniform TYPE * uniform)extract((int64)p, i); \
ret = insert(ret, i, *ptr); \ ret = insert(ret, i, *ptr); \
*ptr = OPFUNC(*ptr, extract(value, i)); \ *ptr = OPFUNC(*ptr, extract(value, i)); \
} \ } \
} else { \
uniform TYPE * uniform ptrs[programCount]; \
ptrs[programIndex] = p; \
foreach_active (i) { \
ret = insert(ret, i, *ptrs[i]); \
*ptrs[i] = OPFUNC(*ptrs[i], extract(value, i)); \
} \
} \
return ret; \ return ret; \
} }