Use different active wait scheme.

2541812e · Adam Osewski · cfbef574 · 2541812e
Commit 2541812e authored May 14, 2024 by Adam Osewski
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 8 deletions

include/ck/utility/work_scheduling.hpp include/ck/utility/work_scheduling.hpp +4 -8

No files found.
--- a/include/ck/utility/work_scheduling.hpp
+++ b/include/ck/utility/work_scheduling.hpp
@@ -105,15 +105,11 @@ class StridedReductionTileLoop
        if(neighbour_count > 0)
        {
-            index_t flag_sum = 0;
+            // Check if this workgroup's flag is also set (i = 0)
-            do
+            for(index_t i = 0; i <= neighbour_count; ++i)
            {
-                flag_sum = 0;
+                finished_block_flags_.wait_eq(GetWorkgroupFlagIdx() + i, 1);
-                for(index_t i = 1; i <= neighbour_count; ++i)
+            }
-                {
-                    flag_sum += finished_block_flags_.ld(GetWorkgroupFlagIdx() + i);
-                }
-            } while(flag_sum != neighbour_count);
        }
        return neighbour_count;