SWDEV-271496 - Added missing params check

Change-Id: I17bbf5ed445826e7380df22fa5c6ec1675e208d6


[ROCm/clr commit: b7fd0bc902]
This commit is contained in:
Sarbojit Sarkar
2021-02-05 05:40:41 -05:00
zatwierdzone przez Sarbojit Sarkar
rodzic 125de2dbe6
commit 3c1920f76c
2 zmienionych plików z 8 dodań i 6 usunięć
@@ -40,12 +40,12 @@ amd::HostQueue* Device::NullStream(bool skip_alloc) {
hipError_t hipDeviceGet(hipDevice_t *device, int deviceId) {
HIP_INIT_API(hipDeviceGet, device, deviceId);
if (device != nullptr) {
*device = deviceId;
} else {
HIP_RETURN(hipErrorInvalidValue);
if (deviceId < 0 ||
static_cast<size_t>(deviceId) >= g_devices.size() ||
device == nullptr) {
HIP_RETURN(hipErrorInvalidDevice);
}
*device = deviceId;
HIP_RETURN(hipSuccess);
};