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

SWDEV-436126 - Fix Unit_hipStreamBeginCapture_StreamSync_OngoingCapture_MThread global mode

Browse Source 
Change-Id: I766ff33485a49df98380cfc8a17bbd9db7548faa


[ROCm/hip-tests commit: 3641ca591c]
This commit is contained in:
Anusha GodavarthySurya
2024-03-28 06:14:19 +00:00
parent b2f3b86f5b
commit 8b89aab0d4
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/hip-tests/catch/unit/graph/hipStreamBeginCapture.cc
+6 -3
View File
@@ -1401,7 +1401,12 @@ static void captureStrmThread(hipGraph_t* graph, int* Ah, int* Ad, int* Bh, int*
std::thread t1(strmSyncThread, Ah, Ad, Bh, Bd, BLOCKSIZE, error);
t1.join();
myadd<<<GRIDSIZE, BLOCKSIZE, 0, stream[0]>>>(Ad, Bd);
HIP_CHECK(hipStreamEndCapture(stream[0], graph)); // End Capture
if (flag == hipStreamCaptureModeGlobal) {
HIP_CHECK_ERROR(hipStreamEndCapture(stream[0], graph),
hipErrorStreamCaptureInvalidated); // End Capture
} else {
HIP_CHECK(hipStreamEndCapture(stream[0], graph)); // End Capture
}
}
TEST_CASE("Unit_hipStreamBeginCapture_StreamSync_OngoingCapture_MThread") {
@@ -1445,13 +1450,11 @@ TEST_CASE("Unit_hipStreamBeginCapture_StreamSync_OngoingCapture_MThread") {
REQUIRE(error == hipErrorStreamCaptureUnsupported);
}
#endif
#if HT_AMD
SECTION("Capture Flag = hipStreamCaptureModeGlobal Multithreaded") {
captureStrmThread(&graph, Ah.host_ptr(), Ad.ptr(), Bh.host_ptr(), Bd.ptr(), BLOCKSIZE, GRIDSIZE,
hipStreamCaptureModeGlobal, &error);
REQUIRE(error == hipErrorStreamCaptureUnsupported);
}
#endif
SECTION("Capture Flag = hipStreamCaptureModeThreadLocal Multithreaded") {
captureStrmThread(&graph, Ah.host_ptr(), Ad.ptr(), Bh.host_ptr(), Bd.ptr(), BLOCKSIZE, GRIDSIZE,
hipStreamCaptureModeThreadLocal, &error);