SWDEV-447525 - maxSize param for Mem pool props.
Change-Id: Ib743611a66921aeaa449b6622b933cc8b9190d87
[ROCm/clr commit: b72d8da1bd]
This commit is contained in:
gecommit door
Jaydeepkumar Patel
bovenliggende
54d326dfa1
commit
844765d7e0
@@ -171,6 +171,9 @@ void* MemoryPool::AllocateMemory(size_t size, Stream* stream, void* dptr) {
|
||||
void* dev_ptr = nullptr;
|
||||
amd::Memory* memory = free_heap_.FindMemory(size, stream, Opportunistic(), dptr);
|
||||
if (memory == nullptr) {
|
||||
if (Properties().maxSize != 0 && (max_total_size_ + size) > Properties().maxSize) {
|
||||
return nullptr;
|
||||
}
|
||||
amd::Context* context = device_->asContext();
|
||||
const auto& dev_info = context->devices()[0]->info();
|
||||
if (dev_info.maxMemAllocSize_ < size) {
|
||||
|
||||
@@ -215,6 +215,7 @@ class MemoryPool : public amd::ReferenceCountedObject {
|
||||
.handleTypes = hipMemHandleTypeNone,
|
||||
.location = {.type = hipMemLocationTypeDevice, .id = device_->deviceId()},
|
||||
.win32SecurityAttributes = nullptr,
|
||||
.maxSize = 0;
|
||||
.reserved = {}};
|
||||
}
|
||||
state_.interprocess_ = properties_.handleTypes != hipMemHandleTypeNone;
|
||||
|
||||
Verwijs in nieuw issue
Block a user