SWDEV-271496 - Added missing params check

Change-Id: I17bbf5ed445826e7380df22fa5c6ec1675e208d6
Этот коммит содержится в:
Sarbojit Sarkar
2021-02-05 05:40:41 -05:00
коммит произвёл Sarbojit Sarkar
родитель 18234e1d24
Коммит b7fd0bc902
2 изменённых файлов: 8 добавлений и 6 удалений
+5 -5
Просмотреть файл
@@ -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);
};
+3 -1
Просмотреть файл
@@ -414,7 +414,9 @@ hipError_t hipDeviceGetPCIBusId ( char* pciBusId, int len, int device ) {
hipError_t hipDeviceGetSharedMemConfig ( hipSharedMemConfig * pConfig ) {
HIP_INIT_API(hipDeviceGetSharedMemConfig, pConfig);
if (pConfig == nullptr) {
return HIP_RETURN(hipErrorInvalidValue);
}
*pConfig = hipSharedMemBankSizeFourByte;
HIP_RETURN(hipSuccess);