From f3f11f4c7ddd2a0ef53f6ebc29196d1a57ff20d9 Mon Sep 17 00:00:00 2001
From: Satyanvesh Dittakavi <Satyanvesh.Dittakavi@amd.com>
Date: Wed, 2 Feb 2022 15:51:09 +0000
Subject: [PATCH] EXSWCPHIPT-13 - Fix Segfault on Nvidia Platform

Change-Id: Ib85bb0b0e91b7703afc4858c4a5ada0115ad68f4
---
 .../include/hip/nvidia_detail/nvidia_hip_runtime_api.h | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h b/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
index fdeb80f6bb..e6e9bd7a9b 100644
--- a/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
+++ b/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
@@ -1220,6 +1220,11 @@ inline static hipError_t hipSetDevice(int device) {
 }
 
 inline static hipError_t hipChooseDevice(int* device, const hipDeviceProp_t* prop) {
+
+    if (prop == NULL) {
+      return hipErrorInvalidValue;
+    }
+
     struct cudaDeviceProp cdprop;
     memset(&cdprop, 0x0, sizeof(struct cudaDeviceProp));
     cdprop.major = prop->major;
@@ -1532,6 +1537,11 @@ inline static hipError_t hipMemset3DAsync(hipPitchedPtr pitchedDevPtr, int  valu
 }
 
 inline static hipError_t hipGetDeviceProperties(hipDeviceProp_t* p_prop, int device) {
+
+    if (p_prop == NULL) {
+      return hipErrorInvalidValue;
+    }
+
     struct cudaDeviceProp cdprop;
     cudaError_t cerror;
     cerror = cudaGetDeviceProperties(&cdprop, device);