Match hipLaunchCooperativeKernelMultiDevice behavior with nvidia (#385)

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

[ROCm/clr commit: 63c4020e49]
这个提交包含在:
Stojiljkovic, Vladana和Anusha GodavarthySurya 编写, GitHub 于 2025-05-22 20:52:37 +05:30 提交
1 父节点 25084759b8
当前提交 9af11abb42
1 个文件已更改
+14
+14
查看文件
@@ -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;