SWDEV-379395 - Use getQueue as it handles null stream and flags can be combination.

Change-Id: If37854a9fc0fc57acb30e652953a7f283831cad8


[ROCm/clr commit: 01c3299425]
Tento commit je obsažen v:
Jaydeep Patel
2023-01-31 13:22:22 +00:00
odevzdal Jaydeepkumar Patel
rodič c6da0aa014
revize 5981bd5c22
+3 -2
Zobrazit soubor
@@ -516,7 +516,8 @@ hipError_t ihipModuleLaunchCooperativeKernelMultiDevice(hipFunctionLaunchParams*
return hipErrorInvalidValue;
}
if ((flags < 0) || (flags > hipCooperativeLaunchMultiDeviceNoPostSync)) {
if (flags > (hipCooperativeLaunchMultiDeviceNoPostSync +
hipCooperativeLaunchMultiDeviceNoPreSync)) {
return hipErrorInvalidValue;
}
@@ -730,7 +731,7 @@ hipError_t ihipLaunchCooperativeKernelMultiDevice(hipLaunchParams* launchParamsL
return hipErrorInvalidValue;
}
amd::HostQueue* queue = reinterpret_cast<hip::Stream*>(launch.stream)->asHostQueue();
amd::HostQueue* queue = hip::getQueue(launch.stream);
hipFunction_t func = nullptr;
// The order of devices in the launch may not match the order in the global array
for (size_t dev = 0; dev < g_devices.size(); ++dev) {