added hipHostRegister support no multi-gpu

[ROCm/hip commit: 07d570468e]
Dieser Commit ist enthalten in:
Aditya Atluri
2016-04-11 10:28:16 -05:00
Ursprung c47b722b14
Commit ed3013cb6e
2 geänderte Dateien mit 19 neuen und 7 gelöschten Zeilen
+3 -2
Datei anzeigen
@@ -238,8 +238,9 @@ hipError_t hipHostRegister(void *hostPtr, size_t sizeBytes, unsigned int flags)
}
if(device){
if(flags == hipHostRegisterDefault){
hsa_status_t hsa_status = hsa_amd_memory_lock(hostPtr, sizeBytes, &device->_hsa_agent, 1, &srcPtr);
if(hsa_status == HSA_STATUS_SUCCESS){
am_status_t am_status = hc::am_memtracker_host_memory_lock(device->_acc, hostPtr, sizeBytes);
// hsa_status_t hsa_status = hsa_amd_memory_lock(hostPtr, sizeBytes, &device->_hsa_agent, 1, &srcPtr);
if(am_status == AM_SUCCESS){
hip_status = hipSuccess;
}else{
hip_status = hipErrorMemoryAllocation;