SWDEV-313905 - Fix segfault

Change-Id: I7b7bbe80a169014b5d4a8de134351b9f8bf11735
This commit is contained in:
Sarbojit Sarkar
2022-01-07 09:41:24 +00:00
zatwierdzone przez Sarbojit Sarkar
rodzic c4f6037ae8
commit 017603703b
2 zmienionych plików z 5 dodań i 1 usunięć
+4
Wyświetl plik
@@ -518,6 +518,10 @@ hipError_t hipLaunchCooperativeKernel(const void* f,
HIP_INIT_API(hipLaunchCooperativeKernel, f, gridDim, blockDim,
sharedMemBytes, hStream);
if (!hip::isValid(hStream)) {
HIP_RETURN(hipErrorInvalidValue);
}
hipFunction_t func = nullptr;
int deviceId = hip::Stream::DeviceId(hStream);
HIP_RETURN_ONFAIL(PlatformState::instance().getStatFunc(&func, f, deviceId));