From 2c1bd4bd108f331ec7fd7b996b3d412d2852682d Mon Sep 17 00:00:00 2001
From: Rahul Garg <rahul.garg@amd.com>
Date: Fri, 16 Sep 2016 11:15:03 +0530
Subject: [PATCH] HIP/NVCC path for hipChooseDevice

Change-Id: I64690d7df1f5493e067c7ff5e3b39a10fcf323bb
---
 hipamd/include/nvcc_detail/hip_runtime_api.h | 23 ++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/hipamd/include/nvcc_detail/hip_runtime_api.h b/hipamd/include/nvcc_detail/hip_runtime_api.h
index 21ef239894..c90c7cbf29 100644
--- a/hipamd/include/nvcc_detail/hip_runtime_api.h
+++ b/hipamd/include/nvcc_detail/hip_runtime_api.h
@@ -214,6 +214,29 @@ inline static hipError_t hipSetDevice(int device) {
     return hipCUDAErrorTohipError(cudaSetDevice(device));
 }
 
+inline static hipError_t hipChooseDevice( int* device, const hipDeviceProp_t* prop )
+{
+    cudaDeviceProp cdprop;
+    memset(&cdprop,0x0,sizeof(cudaDeviceProp));
+    cdprop.major= prop->major;
+    cdprop.minor = prop->minor;
+    cdprop.totalGlobalMem = prop->totalGlobalMem ;
+    cdprop.sharedMemPerBlock = prop->sharedMemPerBlock;
+    cdprop.regsPerBlock = prop->regsPerBlock;
+    cdprop.warpSize = prop->warpSize ;
+    cdprop.maxThreadsPerBlock = prop->maxThreadsPerBlock ;
+    cdprop.clockRate = prop->clockRate;
+    cdprop.totalConstMem = prop->totalConstMem ;
+    cdprop.multiProcessorCount = prop->multiProcessorCount ;
+    cdprop.l2CacheSize = prop->l2CacheSize ;
+    cdprop.maxThreadsPerMultiProcessor = prop->maxThreadsPerMultiProcessor ;
+    cdprop.computeMode = prop->computeMode ;
+    cdprop.canMapHostMemory = prop->canMapHostMemory;
+    cdprop.memoryClockRate = prop->memoryClockRate;
+    cdprop.memoryBusWidth = prop->memoryBusWidth;
+    return hipCUDAErrorTohipError(cudaChooseDevice(device,&cdprop));
+}
+
 inline static hipError_t hipMemcpyHtoD(hipDeviceptr_t dst, 
                   void* src, size_t size)
 {