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/hip commit: eba2595611]
This commit is contained in:
pensun
2016-12-01 13:51:58 -06:00
parent 1c91897cb8
commit 2b557e949e
3 changed files with 68 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/hip/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() );
projects/hip/src/hip_memory.cpp
+15
View File
@@ -1034,3 +1034,18 @@ hipError_t hipMemGetAddressRange ( hipDeviceptr_t* pbase, size_t* psize, hipDevi
hipStatus = hipErrorInvalidDevicePointer;
return hipStatus;
}
//TODO: IPC implementaiton:
hipError_t hipIpcGetMemHandle(hipIpcMemHandle_t* handle, void* devPtr){
return hipSuccess;
}
hipError_t hipIpcCloseMemHandle(void *devPtr){
return hipSuccess;
}
hipError_t hipIpcOpenEventHandle(hipEvent_t* event, hipIpcEventHandle_t handle){
return hipSuccess;
}
hipError_t hipIpcOpenMemHandle(void** devPtr, hipIpcMemHandle_t handle, unsigned int flags){
return hipSuccess;
}