added stream synch after hipMemset (#609)

[ROCm/rccl commit: 06bce9d0c9]
2022-08-30 16:18:37 -06:00
parent f18868f439
commit 2cd63dac42
5 changed files with 5 additions and 3 deletions
@@ -172,7 +172,7 @@ int main(int argc, char* argv[])
      HIPCHECK(hipMalloc(recvbuff+i, size*sizeof(float)));
      h_recvbuff[i] = (float*) malloc (size *sizeof(float));
      HIPCHECK(hipMemset(recvbuff[i], 0, size*sizeof(float)));
-
+      HIPCHECK(hipStreamSynchronize(NULL));
      HIPCHECK(hipStreamCreate(s+i));
  }

@@ -174,7 +174,7 @@ int main(int argc, char* argv[])
      HIPCHECK(hipMalloc(recvbuff+i, recvsize*sizeof(float)));
      h_recvbuff[i] = (float*) malloc (recvsize *sizeof(float));
      HIPCHECK(hipMemset(recvbuff[i], 0, recvsize*sizeof(float)));
-
+      HIPCHECK(hipStreamSynchronize(NULL));
      HIPCHECK(hipStreamCreate(s+i));
  }