AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor: Consolidate calls to hsaKmtAllocMemory

Browse Source 
Route all device-visible system memory allocations through system_allocator.

Change-Id: I5e90a1bf491e432678a6d8ab1f9f3770734cbda1


[ROCm/ROCR-Runtime commit: 74f5aca93d]
This commit is contained in:
Jay Cornwall
2016-08-22 20:19:21 -05:00
parent 7e2179da7b
commit d5ecfae62f
17 changed files with 89 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocr-runtime/runtime/hsa-runtime/core/common/shared.cpp
+2 -2
View File
@@ -43,6 +43,6 @@
#include "core/common/shared.h"
namespace core {
std::function<void*(size_t, size_t)> BaseShared::allocate_=nullptr;
std::function<void(void*)> BaseShared::free_=nullptr;
std::function<void*(size_t, size_t, uint32_t)> BaseShared::allocate_ = nullptr;
std::function<void(void*)> BaseShared::free_ = nullptr;
}
projects/rocr-runtime/runtime/hsa-runtime/core/common/shared.h
+3 -3
View File
@@ -55,14 +55,14 @@ namespace core {
class BaseShared {
public:
static void SetAllocateAndFree(
const std::function<void*(size_t, size_t)>& allocate,
const std::function<void*(size_t, size_t, uint32_t)>& allocate,
const std::function<void(void*)>& free) {
allocate_ = allocate;
free_ = free;
}
protected:
static std::function<void*(size_t, size_t)> allocate_;
static std::function<void*(size_t, size_t, uint32_t)> allocate_;
static std::function<void(void*)> free_;
};
@@ -78,7 +78,7 @@ class Shared : public BaseShared {
"Align is less than alignof(T)");
shared_object_ =
reinterpret_cast<T*>(allocate_(sizeof(T), Max(__alignof(T), Align)));
reinterpret_cast<T*>(allocate_(sizeof(T), Max(__alignof(T), Align), 0));
assert(shared_object_ != NULL && "Failed on allocating shared_object_");