[Tools/Replayer] Fix prohibited calls during capture mode (#1938)

Цей коміт міститься в:
Bertan Dogancay authored and GitHub committed 2025-10-29 12:19:32 -04:00
1 джерело 561ad2fe05
коміт b703ffdfa4
1 file changed
+4 -2
+4 -2
Переглянути файл
@@ -270,7 +270,7 @@ void Replayer::replay()
graphLife[call.graphID].counter--;
if (graphLife[call.graphID].starts.contains(lineNum))
{
HIP_CALL(hipStreamBeginCapture(streams[call.stream].first, hipStreamCaptureModeGlobal));
HIP_CALL(hipStreamBeginCapture(streams[call.stream].first, hipStreamCaptureModeRelaxed));
printf("[INFO ] Rank %d - Line %d : starting capture graph %llu\n", myRank, lineNum, call.graphID);
} else if (graphLife[call.graphID].stream != call.stream) {
printf("[WARNING ] \x1b[31mRank %d - Line %d : multi-stream graph may not replay original dependency accurately\x1b[0m\n", myRank, lineNum);
@@ -560,7 +560,9 @@ cleanup:
}
if (call.stream && lineNum == streams[call.stream].second)
{
HIP_CALL(hipStreamSynchronize(streams[call.stream].first)); // ?
if (call.graphCaptured != 1) {
HIP_CALL(hipStreamSynchronize(streams[call.stream].first)); // ?
}
HIP_CALL(hipStreamDestroy(streams[call.stream].first));
}
lineNum++; // change for a2av