From 2a044e382308158746499161ba38e6bec6bb9538 Mon Sep 17 00:00:00 2001 From: Aditya Atluri Date: Mon, 21 Mar 2016 10:32:30 -0500 Subject: [PATCH] fixed memory free apis [ROCm/hip commit: 96a1899df75dd16b293e9f49dc878acf566bc0d8] --- .../hip/include/hcc_detail/hip_runtime_api.h | 4 +- projects/hip/include/hip_runtime_api.h | 2 + projects/hip/src/hip_hcc.cpp | 27 ++++++++-- projects/hip/tests/src/CMakeLists.txt | 4 ++ projects/hip/tests/src/hipMemoryAllocate.cpp | 51 +++++++++++++++++++ 5 files changed, 81 insertions(+), 7 deletions(-) create mode 100644 projects/hip/tests/src/hipMemoryAllocate.cpp diff --git a/projects/hip/include/hcc_detail/hip_runtime_api.h b/projects/hip/include/hcc_detail/hip_runtime_api.h index ef5ad4927c..4c854e72f1 100644 --- a/projects/hip/include/hcc_detail/hip_runtime_api.h +++ b/projects/hip/include/hcc_detail/hip_runtime_api.h @@ -688,10 +688,10 @@ hipError_t hipHostAlloc(void** ptr, size_t size, unsigned int flags) __attribute * * @param[out] dstPtr Device Pointer mapped to passed host pointer * @param[in] hstPtr Host Pointer allocated through hipHostAlloc - * @param[in] size Requested memory size + * @param[in] flags Flags to be passed for extension * @return Error code */ -hipError_t hipHostGetDevicePointer(void** devPtr, void* hstPtr, size_t size) ; +hipError_t hipHostGetDevicePointer(void** devPtr, void* hstPtr, unsigned int flags) ; /** * @brief Get flags associated with host pointer diff --git a/projects/hip/include/hip_runtime_api.h b/projects/hip/include/hip_runtime_api.h index c7d9f3fa8b..3a0a4b399a 100644 --- a/projects/hip/include/hip_runtime_api.h +++ b/projects/hip/include/hip_runtime_api.h @@ -149,6 +149,8 @@ typedef enum hipError_t { ,hipErrorInvalidResourceHandle ///< Resource handle (hipEvent_t or hipStream_t) invalid. ,hipErrorInvalidDevice ///< DeviceID must be in range 0...#compute-devices. ,hipErrorInvalidMemcpyDirection ///< Invalid memory copy direction + ,hipErrorInvalidDevicePointer ///< Invalid Device Pointer + ,hipErrorInitializationError ///< TODO comment from hipErrorInitializationError ,hipErrorNoDevice ///< Call to hipGetDeviceCount returned 0 devices ,hipErrorNotReady ///< Indicates that asynchronous operations enqueued earlier are not ready. This is not actually an error, but is used to distinguish from hipSuccess (which indicates completion). APIs that return this error include hipEventQuery and hipStreamQuery. diff --git a/projects/hip/src/hip_hcc.cpp b/projects/hip/src/hip_hcc.cpp index 7aa7ae4482..23481e27b3 100644 --- a/projects/hip/src/hip_hcc.cpp +++ b/projects/hip/src/hip_hcc.cpp @@ -2603,15 +2603,24 @@ hipError_t hipFree(void* ptr) // TODO - ensure this pointer was created by hipMalloc and not hipMallocHost std::call_once(hip_initialized, ihipInit); + hipError_t hipStatus = hipErrorInvalidDevicePointer; // Synchronize to ensure all work has finished. ihipGetTlsDefaultDevice()->waitAllStreams(); // ignores non-blocking streams, this waits for all activity to finish. if (ptr) { - hc::am_free(ptr); + hc::accelerator acc; + hc::AmPointerInfo amPointerInfo(NULL, NULL, 0, acc, 0, 0); + am_status_t status = hc::am_memtracker_getinfo(&amPointerInfo, ptr); + if(status == AM_SUCCESS){ + if(amPointerInfo._hostPointer == NULL){ + hc::am_free(ptr); + hipStatus = hipSuccess; + } + } } - return ihipLogStatus(hipSuccess); + return ihipLogStatus(hipStatus); } @@ -2620,12 +2629,20 @@ hipError_t hipHostFree(void* ptr) // TODO - ensure this pointer was created by hipMallocHost and not hipMalloc std::call_once(hip_initialized, ihipInit); + hipError_t hipStatus = hipErrorInvalidDevicePointer; if (ptr) { - tprintf (DB_MEM, " %s: %p\n", __func__, ptr); - hc::am_free(ptr); + hc::accelerator acc; + hc::AmPointerInfo amPointerInfo(NULL, NULL, 0, acc, 0, 0); + am_status_t status = hc::am_memtracker_getinfo(&amPointerInfo, ptr); + if(status == AM_SUCCESS){ + if(amPointerInfo._hostPointer == ptr){ + hc::am_free(ptr); + hipStatus = hipSuccess; + } + } } - return ihipLogStatus(hipSuccess); + return ihipLogStatus(hipStatus); }; diff --git a/projects/hip/tests/src/CMakeLists.txt b/projects/hip/tests/src/CMakeLists.txt index bbbb3c52ed..ab9c41d545 100644 --- a/projects/hip/tests/src/CMakeLists.txt +++ b/projects/hip/tests/src/CMakeLists.txt @@ -140,6 +140,8 @@ make_hip_executable (hipStreamL5 hipStreamL5.cpp) make_hip_executable (hipHostGetFlags hipHostGetFlags.cpp) make_hip_executable (hipHostRegister hipHostRegister.cpp) make_hip_executable (hipRandomMemcpyAsync hipRandomMemcpyAsync.cpp) +make_hip_executable (hipMemoryAllocate hipMemoryAllocate.cpp) + make_test(hip_ballot " " ) make_test(hip_anyall " " ) make_test(hip_popc " " ) @@ -170,4 +172,6 @@ make_test(hipStreamL5 " ") make_test(hipRandomMemcpyAsync " ") #make_test(hipAPIStreamEnable " ") #make_test(hipAPIStreamDisable " ") +make_test(hipMemoryAllocate " ") + make_hipify_test(specialFunc.cu ) diff --git a/projects/hip/tests/src/hipMemoryAllocate.cpp b/projects/hip/tests/src/hipMemoryAllocate.cpp new file mode 100644 index 0000000000..30da822eaf --- /dev/null +++ b/projects/hip/tests/src/hipMemoryAllocate.cpp @@ -0,0 +1,51 @@ +/* +Copyright (c) 2015-2016 Advanced Micro Devices, Inc. All rights reserved. +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: +The above copyright notice and this permission notice shall be included in +all copies or substantial portions of the Software. +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANNTY OF ANY KIND, EXPRESS OR +IMPLIED, INNCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANNY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER INN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR INN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +THE SOFTWARE. +*/ +#include"test_common.h" + +#define SIZE 1024*1024*256 + +int main(){ + float *Ad, *B, *Bd, *Bm, *C, *Cd; + B = (float*)malloc(SIZE); + hipMalloc((void**)&Ad, SIZE); + hipHostMalloc((void**)&B, SIZE); + hipHostMalloc((void**)&Bd, SIZE, hipHostMallocDefault); + hipHostMalloc((void**)&Bm, SIZE, hipHostMallocMapped); + hipHostMalloc((void**)&C, SIZE, hipHostMallocMapped); + hipHostGetDevicePointer((void**)&Cd, C, SIZE); + + HIPASSERT(hipFree(Ad) == hipSuccess); + HIPASSERT(hipHostFree(Ad) == hipErrorInvalidDevicePointer); + + HIPASSERT(hipFree(B) == hipErrorInvalidDevicePointer); + HIPASSERT(hipFree(Bd) == hipErrorInvalidDevicePointer); + HIPASSERT(hipFree(Bm) == hipErrorInvalidDevicePointer); + HIPASSERT(hipHostFree(Bd) == hipSuccess); + HIPASSERT(hipHostFree(Bm) == hipSuccess); + + HIPASSERT(hipFree(C) == hipErrorInvalidDevicePointer); + HIPASSERT(hipFree(Cd) == hipErrorInvalidDevicePointer); + HIPASSERT(hipHostFree(C) == hipSuccess); + HIPASSERT(hipHostFree(Cd) == hipErrorInvalidDevicePointer); + HIPASSERT(hipFree(Cd) == hipErrorInvalidDevicePointer); + + HIPASSERT(hipFree(NULL) == hipErrorInvalidDevicePointer); + HIPASSERT(hipHostFree(NULL) == hipErrorInvalidDevicePointer); + passed(); +}