Added hipDeviceGetByPCIBusId in hip/hcc path

Change-Id: I3cca0dc533d0281689d8a407c7da16ca1ba6a3a8
Αυτή η υποβολή περιλαμβάνεται σε:
Rahul Garg
2016-11-01 10:57:48 +05:30
γονέας e1812a1319
υποβολή 81c91f5b0b
2 αρχεία άλλαξαν με 30 προσθήκες και 0 διαγραφές
@@ -1591,6 +1591,16 @@ hipError_t hipDeviceGetName(char *name,int len,hipDevice_t device);
*/
hipError_t hipDeviceGetPCIBusId (int *pciBusId,int len,hipDevice_t device);
/**
* @brief Returns a handle to a compute device.
* @param [out] device handle
* @param [in] PCI Bus ID
*
* @returns #hipSuccess, #hipErrorInavlidDevice, #hipErrorInvalidValue
*/
hipError_t hipDeviceGetByPCIBusId ( int* device,const int* pciBusId );
/**
* @brief Returns the total amount of memory on the device.
* @param [out] bytes
@@ -340,8 +340,28 @@ hipError_t hipDeviceTotalMem (size_t *bytes,hipDevice_t device)
return ihipLogStatus(e);
}
hipError_t hipDeviceGetByPCIBusId (int* device, const int* pciBusId )
{
HIP_INIT_API(device,pciBusId);
hipDeviceProp_t tempProp;
int deviceCount;
hipError_t e = hipErrorInvalidValue;
hipGetDeviceCount( &deviceCount );
*device = 0;
for (int i=0; i< deviceCount; i++) {
hipGetDeviceProperties( &tempProp, i );
if(tempProp.pciBusID == *pciBusId) {
*device =i;
e = hipSuccess;
break;
}
}
return ihipLogStatus(e);
}
hipError_t hipChooseDevice( int* device, const hipDeviceProp_t* prop )
{
HIP_INIT_API(device,prop);
hipDeviceProp_t tempProp;
int deviceCount;
int inPropCount=0;