Add USE_HCC_LOCK_API.

Default off, if set will compile assuming HCC lock APIs exist.
Šī revīzija ir iekļauta:
Ben Sander
2016-04-18 23:50:52 -05:00
vecāks 22d5738f82
revīzija 3bcefdaa2d
2 mainīti faili ar 11 papildinājumiem un 3 dzēšanām
+3 -3
Parādīt failu
@@ -39,10 +39,10 @@ THE SOFTWARE.
// >= 3 : use hc::am_memtracker_update_peers(...)
#define USE_PEER_TO_PEER 2
// Use new lock API in HCC:
#define USE_HCC_LOCK 0
//#define INLINE static inline
// Use new am_memory_host_lock APIs:
#define USE_HCC_LOCK_API 0
//---
// Environment variables:
+8
Parādīt failu
@@ -253,7 +253,11 @@ hipError_t hipHostRegister(void *hostPtr, size_t sizeBytes, unsigned int flags)
for(int i=0;i<g_deviceCnt;i++){
vecAcc.push_back(g_devices[i]._acc);
}
#if USE_HCC_LOCK_API
am_status = hc::am_memory_host_lock(device->_acc, hostPtr, sizeBytes, &vecAcc[0], vecAcc.size());
#else
am_status = AM_ERROR_MISC;
#endif
if(am_status == AM_SUCCESS){
hip_status = hipSuccess;
}else{
@@ -276,7 +280,11 @@ hipError_t hipHostUnregister(void *hostPtr)
if(hostPtr == NULL){
hip_status = hipErrorInvalidValue;
}else{
#if USE_HCC_LOCK_API
am_status_t am_status = hc::am_memory_host_unlock(device->_acc, hostPtr);
#else
am_status_t am_status = AM_ERROR_MISC;
#endif
if(am_status != AM_SUCCESS){
hip_status = hipErrorHostMemoryNotRegistered;
}