AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

local changes for hipnccl

Browse Source 
Change-Id: I05a1f0381ce2914a800f573342cc954eb5ff82d9


[ROCm/clr commit: 808e555247]
This commit is contained in:
pensun
2016-12-01 13:51:58 -06:00
parent 448c6b3119
commit aaecc3fe04
3 changed files with 68 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/hipamd/src/hip_device.cpp
+11 -3
View File
@@ -337,13 +337,21 @@ hipError_t hipDeviceGetName(char *name,int len,hipDevice_t device)
return ihipLogStatus(e);
}
hipError_t hipDeviceGetPCIBusId (char *pciBusId,int len,hipDevice_t device)
// hipError_t hipDeviceGetPCIBusId (char *pciBusId,int len,hipDevice_t device)
// {
// HIP_INIT_API(pciBusId, len, device);
// hipError_t e = hipSuccess;
// int deviceId= device->_deviceId;
// e = ihipDeviceGetAttribute((int*)pciBusId, hipDeviceAttributePciBusId, deviceId);
// return ihipLogStatus(e);
// }
hipError_t hipDeviceGetPCIBusId (char *pciBusId,int len,int device)
{
HIP_INIT_API(pciBusId, len, device);
hipError_t e = hipSuccess;
int deviceId= device->_deviceId;
int tempPciBusId = 0;
e = ihipDeviceGetAttribute( &tempPciBusId, hipDeviceAttributePciBusId, deviceId);
e = ihipDeviceGetAttribute( &tempPciBusId, hipDeviceAttributePciBusId, device);
if( e == hipSuccess) {
std::string tempPciStr = std::to_string(tempPciBusId);
memcpy( pciBusId , tempPciStr.c_str() , tempPciStr.length() );