added stream synch after hipMemset (#609)

This commit is contained in:
akolliasAMD
2022-08-30 16:18:37 -06:00
committed by GitHub
szülő c9f2fe1f65
commit 06bce9d0c9
5 fájl változott, egészen pontosan 5 új sor hozzáadva és 3 régi sor törölve
+1
Fájl megtekintése
@@ -93,6 +93,7 @@ static ncclResult_t ncclCudaCallocDebug(const char *filefunc, int line, T** ptr,
CUDACHECK(hipStreamDestroy(stream));
#else
CUDACHECK(hipMemset(*ptr, 0, nelem*sizeof(T)));
CUDACHECK(hipStreamSynchronize(NULL));
#endif
INFO(NCCL_ALLOC, "%s:%d Cuda Alloc Size %ld pointer %p", filefunc, line, nelem*sizeof(T), *ptr);
int dev;