From 651cc04ee10f7d8f4ea8c634fea414ccb879faad Mon Sep 17 00:00:00 2001
From: Evghenii <egaburov@dds.nl>
Date: Tue, 28 Jan 2014 21:29:20 +0100
Subject: [PATCH] +1

---
 examples_ptx/radixSort/radixSort.ispc | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/examples_ptx/radixSort/radixSort.ispc b/examples_ptx/radixSort/radixSort.ispc
index 6526a337..ff17f741 100644
--- a/examples_ptx/radixSort/radixSort.ispc
+++ b/examples_ptx/radixSort/radixSort.ispc
@@ -81,6 +81,7 @@ void sortPass(
     localCounts[i] = 0;
 
   const int unitScan = exclusive_scan_add(1);
+  const int unitSum  = exclusive_scan_add((int)(programIndex < programCount/2));
 
 
   foreach (i = 0 ... nloc)
@@ -102,11 +103,26 @@ void sortPass(
     else
     {
 #ifdef __NVPTX__
-      /* buf fix, somehow atomic w/o branching fails */
       if (programIndex < 16)
-        rel = atomic_add_global(&localCounts[key],1);
+      {
+        if (reduce_equal(key) == true)
+        {
+          rel = localCounts[key] + unitScan;
+          localCounts[key] = rel+1;
+        }
+        else
+          rel = atomic_add_global(&localCounts[key],1);
+      }
       else
-        rel = atomic_add_global(&localCounts[key],1);
+      {
+        if (reduce_equal(key) == true)
+        {
+          rel = localCounts[key] + unitScan - unitSum;
+          localCounts[key] = rel+1;
+        }
+        else
+          rel = atomic_add_global(&localCounts[key],1);
+      }
 #else
       rel = atomic_add_local(&localCounts[key],1);
 #endif