Performing __threadfence_system() with only first thread

[ROCm/rccl commit: 1392dd2997]
2019-10-11 09:16:19 -07:00
parent d257970ad1
commit cf597ff257
1 changed files with 7 additions and 4 deletions
@@ -189,10 +189,13 @@ class ncclPrimitives {
        }
      }
      exitIfAbortBarrier(abort, abortCount);
-      if (tid == 0) FOR_SEND(postSendSize, realSize*sizeof(T));
-      if (SEND) __threadfence_system();
-      if (tid == 0) FOR_SEND(postSend);
-      if (tid == 0) FOR_RECV(postRecv);
+      if (tid == 0)
+      {
+        FOR_SEND(postSendSize, realSize*sizeof(T));
+        __threadfence_system();
+        FOR_SEND(postSend);
+        FOR_RECV(postRecv);
+      }

      for (int i=0; i<RECV*NRECV+SRC; i++) srcs[i] += sliceSize;
      for (int i=0; i<SEND*NSEND+DST; i++) dsts[i] += sliceSize;