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

[ROCm/rccl commit: b703ffdfa4]
Этот коммит содержится в:
Bertan Dogancay
2025-10-29 12:19:32 -04:00
коммит произвёл GitHub
родитель c5cdee4fa5
Коммит 4c7afea115
+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