Add USE_HCC_LOCK_API.
Default off, if set will compile assuming HCC lock APIs exist.
Šī revīzija ir iekļauta:
@@ -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:
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
Atsaukties uz šo jaunā problēmā
Block a user