Match hipLaunchCooperativeKernelMultiDevice behavior with nvidia (#385)

Co-authored-by: Anusha GodavarthySurya <Anusha.GodavarthySurya@amd.com>

[ROCm/clr commit: 63c4020e49]
Αυτή η υποβολή περιλαμβάνεται σε:
Stojiljkovic, Vladana
2025-05-22 17:22:37 +02:00
υποβλήθηκε από GitHub
γονέας 25084759b8
υποβολή 9af11abb42
@@ -859,6 +859,20 @@ hipError_t ihipLaunchCooperativeKernelMultiDevice(hipLaunchParams* launchParamsL
if (!hip::isValid(launch.stream)) {
return hipErrorInvalidValue;
}
if (launch.stream == nullptr || launch.stream == hipStreamLegacy) {
return hipErrorInvalidResourceHandle;
}
// Not supported while stream is capturing
hip::Stream* s = reinterpret_cast<hip::Stream*>(launch.stream);
if (s->GetCaptureStatus() == hipStreamCaptureStatusActive) {
s->SetCaptureStatus(hipStreamCaptureStatusInvalidated);
return hipErrorStreamCaptureUnsupported;
}
if (s->GetCaptureStatus() == hipStreamCaptureStatusInvalidated) {
return hipErrorStreamCaptureInvalidated;
}
hip::Stream* hip_stream = hip::getStream(launch.stream);
hipFunction_t func = nullptr;