H2H Async memcpy fix

In this change, the cpu memcpy will wait until all the commands in the current stream are done.
Note that, it only waits on current stream. But not on other streams.
This commit is contained in:
Aditya Avinash Atluri
2016-02-29 12:49:50 -06:00
والد 1c60874378
کامیت a845dccfef
+5
مشاهده پرونده
@@ -2405,6 +2405,11 @@ hipError_t hipMemcpyAsync(void* dst, const void* src, size_t sizeBytes, hipMemcp
} else if (kind == hipMemcpyHostToHost) { } else if (kind == hipMemcpyHostToHost) {
tprintf (TRACE_COPY2, "H2H copy with memcpy"); tprintf (TRACE_COPY2, "H2H copy with memcpy");
/* As this is a CPU op, we need to wait until all
the commands in current stream are finished.
*/
stream->wait();
memcpy(dst, src, sizeBytes); memcpy(dst, src, sizeBytes);
} else { } else {