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

SWDEV-286346 - Implement Arena Memory Object for externally created memory.

Browse Source 
Change-Id: I8530602d89edf83ad367c52167e48a1559ee1e18


[ROCm/clr commit: 1c49d8816c]
This commit is contained in:
kjayapra-amd
2021-05-13 21:49:01 -04:00
committed by Karthik Jayaprakash
parent c521759b09
commit aae0d4ca51
7 changed files with 67 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/rocclr/device/rocm/rocmemory.cpp
+9 -3
View File
@@ -724,7 +724,7 @@ bool Buffer::create() {
flags_ |= HostMemoryDirectAccess;
}
if (owner()->getSvmPtr() == reinterpret_cast<void*>(1)) {
if (owner()->getSvmPtr() == reinterpret_cast<void*>(amd::Memory::MemoryType::kSvmMemoryPtr)) {
if (isFineGrain) {
if (memFlags & CL_MEM_ALLOC_HOST_PTR) {
if (dev().info().hmmSupported_) {
@@ -777,10 +777,16 @@ bool Buffer::create() {
owner()->setSvmPtr(deviceMemory_);
} else {
deviceMemory_ = owner()->getSvmPtr();
kind_ = MEMORY_KIND_PTRGIVEN;
if (owner()->getSvmPtr() == reinterpret_cast<void*>(amd::Memory::MemoryType
::kArenaMemoryPtr)) {
kind_ = MEMORY_KIND_ARENA;
} else {
kind_ = MEMORY_KIND_PTRGIVEN;
}
}
if ((deviceMemory_ != nullptr) && (dev().settings().apuSystem_ || !isFineGrain)) {
if ((deviceMemory_ != nullptr) && (dev().settings().apuSystem_ || !isFineGrain)
&& (kind_ != MEMORY_KIND_ARENA)) {
const_cast<Device&>(dev()).updateFreeMemory(size(), false);
}