From cd44115728d14fcb9ff4656e15f7a7f9dfc75701 Mon Sep 17 00:00:00 2001
From: Brandon Potter <brandon.potter@amd.com>
Date: Fri, 27 Sep 2024 15:17:57 -0500
Subject: [PATCH] Poll the signal from one thread instead of all

[ROCm/rocshmem commit: 24b928a0075ca9425028d719b41262f14aa1135a]
---
 projects/rocshmem/src/memory/notifier.hpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/projects/rocshmem/src/memory/notifier.hpp b/projects/rocshmem/src/memory/notifier.hpp
index b72a256f15..946a9c291e 100644
--- a/projects/rocshmem/src/memory/notifier.hpp
+++ b/projects/rocshmem/src/memory/notifier.hpp
@@ -78,9 +78,12 @@ class Notifier {
       }
     }
 
-    while (detail::atomic::load<uint32_t, scope>(&signal_, orders_) != done) {
-      ;
+    if (executor) {
+      while (detail::atomic::load<uint32_t, scope>(&signal_, orders_) != done) {
+        ;
+      }
     }
+    __syncthreads();
   }
 
  private: