hipExtMallocWithFlags needs hcc workweek 19115 or higher

此提交包含在:
Maneesh Gupta authored and GitHub committed 2019-03-25 11:41:20 +05:30
1 父節點 73ec5d54b5
當前提交 cab119c8b2
1 file changed
+5
+5
查看文件
@@ -267,6 +267,8 @@ hipError_t hipMalloc(void** ptr, size_t sizeBytes) {
hipError_t hipExtMallocWithFlags(void** ptr, size_t sizeBytes, unsigned int flags) {
HIP_INIT_SPECIAL_API(hipExtMallocWithFlags, (TRACE_MEM), ptr, sizeBytes, flags);
HIP_SET_DEVICE();
#if (__hcc_workweek__ >= 19115)
hipError_t hip_status = hipSuccess;
auto ctx = ihipGetTlsDefaultCtx();
@@ -292,6 +294,9 @@ hipError_t hipExtMallocWithFlags(void** ptr, size_t sizeBytes, unsigned int flag
hip_status = hipErrorMemoryAllocation;
}
}
#else
hipError_t hip_status = hipErrorMemoryAllocation;
#endif
return ihipLogStatus(hip_status);
}