Ben Sander
38c735fd1d
Add support for USE_AM_TRACKER=2 (HCC version). Add AM_ALLOC, AM_FREE indirection to ease swapping AM implementations.
158 lines
4.5 KiB
C++
158 lines
4.5 KiB
C++
#pragma once
|
|
|
|
#include <hc_am.hpp>
|
|
|
|
typedef int am_status_t;
|
|
#define AM_SUCCESS 0
|
|
// TODO - provide better mapping of HSA error conditions to HC error codes.
|
|
#define AM_ERROR_MISC -1 /** Misellaneous error */
|
|
|
|
// Flags for am_alloc API:
|
|
#define amHostPinned 0x1
|
|
|
|
|
|
namespace hc {
|
|
|
|
// Info for each pointer in the memtry tracker:
|
|
struct AmPointerInfo {
|
|
void * _hostPointer; ///< Host pointer. If host access is not allowed, NULL.
|
|
void * _devicePointer; ///< Device pointer.
|
|
size_t _sizeBytes; ///< Size of allocation.
|
|
hc::accelerator _acc; ///< Device / Accelerator to use.
|
|
bool _isInDeviceMem; ///< Memory is physically resident on a device (if false, memory is located on host)
|
|
bool _isAmManaged; ///< Memory was allocated by AM and should be freed when am_reset is called.
|
|
|
|
int _appId; ///< App-specific storage. (Used by HIP to store deviceID.)
|
|
unsigned _appAllocationFlags; ///< App-specific allocation flags. (Used by HIP to store allocation flags.)
|
|
|
|
AmPointerInfo() {};
|
|
|
|
AmPointerInfo(void *hostPointer, void *devicePointer, size_t sizeBytes, hc::accelerator acc, bool isInDeviceMem, bool isAmManaged) :
|
|
_hostPointer(hostPointer),
|
|
_devicePointer(devicePointer),
|
|
_sizeBytes(sizeBytes),
|
|
_acc(acc),
|
|
_isInDeviceMem(isInDeviceMem),
|
|
_isAmManaged(isAmManaged),
|
|
_appId(-1),
|
|
_appAllocationFlags(0) {};
|
|
};
|
|
}
|
|
|
|
|
|
|
|
namespace hc {
|
|
|
|
|
|
/**
|
|
* Allocate a block of @p size bytes of memory on the specified @p acc.
|
|
*
|
|
* The contents of the newly allocated block of memory are not initialized.
|
|
*
|
|
* If @p size == 0, 0 is returned.
|
|
*
|
|
* Flags must be 0.
|
|
*
|
|
* @return : On success, pointer to the newly allocated memory is returned.
|
|
* The pointer is typecast to the desired return type.
|
|
*
|
|
* If an error occurred trying to allocate the requested memory, 0 is returned.
|
|
*
|
|
* @see am_free, am_copy
|
|
*/
|
|
auto_voidp AM_alloc(size_t size, hc::accelerator acc, unsigned flags);
|
|
|
|
/**
|
|
* Free a block of memory previously allocated with am_alloc.
|
|
*
|
|
* @return AM_SUCCESS
|
|
* @see am_alloc, am_copy
|
|
*/
|
|
am_status_t AM_free(void* ptr);
|
|
|
|
|
|
/**
|
|
* Copy @p size bytes of memory from @p src to @ dst. The memory areas (src+size and dst+size) must not overlap.
|
|
*
|
|
* @return AM_SUCCESS on error or AM_ERROR_MISC if an error occurs.
|
|
* @see am_alloc, am_free
|
|
*/
|
|
am_status_t AM_copy(void* dst, const void* src, size_t size);
|
|
|
|
|
|
|
|
/**
|
|
* Return information about tracked pointer.
|
|
*
|
|
* AM tracks pointers when they are allocated or added to tracker with am_track_pointer.
|
|
* The tracker tracks the base pointer as well as the size of the allocation, and will
|
|
* find the information for a pointer anywhere in the tracked range.
|
|
*
|
|
* @returns AM_ERROR_MISC if pointer is not currently being tracked.
|
|
* @returns AM_SUCCESS if pointer is tracked and writes info to @p info.
|
|
*
|
|
* @see AM_memtracker_add,
|
|
*/
|
|
am_status_t am_memtracker_getinfo(hc::AmPointerInfo *info, const void *ptr);
|
|
|
|
|
|
/**
|
|
* Add a pointer to the memory tracker.
|
|
*
|
|
* @return AM_SUCCESS
|
|
* @see am_memtracker_getinfo
|
|
*/
|
|
am_status_t am_memtracker_add(void* ptr, size_t sizeBytes, hc::accelerator acc, bool isDeviceMem=false);
|
|
|
|
|
|
/*
|
|
* Update info for an existing pointer in the memory tracker.
|
|
*
|
|
* @returns AM_ERROR_MISC if pointer is not found in tracker.
|
|
* @returns AM_SUCCESS if pointer is not found in tracker.
|
|
*
|
|
* @see am_memtracker_getinfo, am_memtracker_add
|
|
*/
|
|
am_status_t am_memtracker_update(const void* ptr, int appId, unsigned allocationFlags);
|
|
|
|
|
|
/**
|
|
* Remove @ptr from the tracker structure.
|
|
*
|
|
* @p ptr may be anywhere in a tracked memory range.
|
|
*
|
|
* @returns AM_ERROR_MISC if pointer is not found in tracker.
|
|
* @returns AM_SUCCESS if pointer is not found in tracker.
|
|
*
|
|
* @see am_memtracker_getinfo, am_memtracker_add
|
|
*/
|
|
am_status_t am_memtracker_remove(void* ptr);
|
|
|
|
/**
|
|
* Remove all memory allocations associated with specified accelerator from the memory tracker.
|
|
*
|
|
* @returns Number of entries reset.
|
|
* @see am_memtracker_getinfo
|
|
*/
|
|
size_t am_memtracker_reset(hc::accelerator acc);
|
|
|
|
/**
|
|
* Print the entries in the memory tracker table.
|
|
*
|
|
* Intended primarily for debug purposes.
|
|
* @see am_memtracker_getinfo
|
|
**/
|
|
void am_memtracker_print();
|
|
|
|
|
|
/**
|
|
* Return total sizes of device, host, and user memory allocated by the application
|
|
*
|
|
* User memory is registered with am_tracker_add.
|
|
**/
|
|
void am_memtracker_sizeinfo(hc::accelerator acc, size_t *deviceMemSize, size_t *hostMemSize, size_t *userMemSize);
|
|
|
|
|
|
}; // namespace hc
|
|
|