From 81c71aeb67590758342c859902ff0bb7f32c0a7f Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Tue, 20 Sep 2022 20:06:17 -0700
Subject: [PATCH] Revert changes to gfx1030 (#622)

[ROCm/rccl commit: d9216af48bda519a958fc667a9c63b993b4dfc39]
---
 projects/rccl/src/collectives/device/prims_simple.h | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/projects/rccl/src/collectives/device/prims_simple.h b/projects/rccl/src/collectives/device/prims_simple.h
index e268d4f306..6ea594ce02 100644
--- a/projects/rccl/src/collectives/device/prims_simple.h
+++ b/projects/rccl/src/collectives/device/prims_simple.h
@@ -319,7 +319,11 @@ private:
 
         }
         barrier(); // This barrier has a counterpart in following loop
+#if defined(__gfx1030__)
+       if (Send && (flags & RolePostSend) && index == 0) __threadfence_system();
+#else
 	if ((MaxSend == 0 || MaxRecv == 0) && Send && (flags & RolePostSend) && index == 0) __threadfence_system();
+#endif
 	__syncwarp();
         postPeer<Recv, Send>();
         offset += sliceSize;
@@ -339,7 +343,11 @@ private:
         waitPeer<DirectRecv, DirectSend, Recv, Send, Src, Dst>(0, 0, 0, 0);
       }
       barrier(); // Has couterpart in preceding worker-only loop.
+#if defined(__gfx1030__)
+      if (Send && (flags & RolePostSend) && sliceSize > 0 && index == 0) __threadfence_system();
+#else
       if ((MaxSend == 0 || MaxRecv == 0) && Send && (flags & RolePostSend) && sliceSize > 0 && index == 0) __threadfence_system();
+#endif
       __syncwarp();
       postPeer<Recv, Send>();
       offset += sliceSize;