AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[SWDEV-456769][SWDEV-456770] - Fix for Memcpy2D Async and Positive Sync tests

Browse Source 
Change-Id: I586d7f75d26ada1609156054e921f05b878e8503
This commit is contained in:
Rahul Manocha
2024-04-15 16:24:44 -07:00
committed by Rakesh Roy
parent d391820895
commit e835edcbab
3 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
catch/unit/memory/memcpy2d_tests_common.hh
+2 -1
View File
@@ -90,7 +90,7 @@ void Memcpy2DDeviceToDeviceShell(F memcpy_func, const hipStream_t kernel_stream
}
LinearAllocGuard2D<int> src_alloc(cols * src_cols_mult, rows);
HIP_CHECK(hipSetDevice(src_device));
HIP_CHECK(hipSetDevice(dst_device));
LinearAllocGuard2D<int> dst_alloc(cols, rows);
HIP_CHECK(hipSetDevice(src_device));
LinearAllocGuard<int> host_alloc(LinearAllocs::hipHostMalloc, dst_alloc.width() * rows);
@@ -181,6 +181,7 @@ void MemcpySyncBehaviorCheck(F memcpy_func, const bool should_sync,
LaunchDelayKernel(std::chrono::milliseconds{300}, kernel_stream);
HIP_CHECK(memcpy_func());
if (should_sync) {
HIP_CHECK(hipStreamSynchronize(kernel_stream));
HIP_CHECK(hipStreamQuery(kernel_stream));
} else {
HIP_CHECK_ERROR(hipStreamQuery(kernel_stream), hipErrorNotReady);