SWDEV-271496 - Added missing params check

Change-Id: I17bbf5ed445826e7380df22fa5c6ec1675e208d6
This commit is contained in:
Sarbojit Sarkar authored and Sarbojit Sarkar committed 2021-02-12 13:28:12 -05:00
1 parent 18234e1d24
commit b7fd0bc902
2 file changed
+8 -6

No files matched your search

+5 -5
View File
@@ -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);
};