[Tools/Replayer] Fix prohibited calls during capture mode (#1938)
[ROCm/rccl commit: b703ffdfa4]
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
c5cdee4fa5
Коммит
4c7afea115
@@ -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
|
||||
|
||||
Ссылка в новой задаче
Block a user