Match hipLaunchCooperativeKernelMultiDevice behavior with nvidia (#385)

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

[ROCm/clr commit: 63c4020e49]
Cette révision appartient à :
Stojiljkovic, VladanaetAnusha GodavarthySurya a créé et GitHub a révisé 2025-05-22 20:52:37 +05:30
1 Parent 25084759b8
révision 9af11abb42
1 fichier modifié
+14
+14
Voir le fichier
@@ -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;