96 lines
3.7 KiB
C++
96 lines
3.7 KiB
C++
/*
|
|
* Copyright (C) 2019-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/os_interface/windows/wddm_residency_allocations_container.h"
|
|
|
|
#include "shared/source/os_interface/windows/wddm/wddm.h"
|
|
#include "shared/source/os_interface/windows/wddm_allocation.h"
|
|
|
|
#include <algorithm>
|
|
|
|
namespace NEO {
|
|
WddmResidentAllocationsContainer::~WddmResidentAllocationsContainer() {
|
|
evictAllResources();
|
|
}
|
|
|
|
MemoryOperationsStatus WddmResidentAllocationsContainer::isAllocationResident(const D3DKMT_HANDLE &handle) {
|
|
auto lock = acquireLock(resourcesLock);
|
|
auto position = std::find(resourceHandles.begin(), resourceHandles.end(), handle);
|
|
return position != resourceHandles.end() ? MemoryOperationsStatus::SUCCESS : MemoryOperationsStatus::MEMORY_NOT_FOUND;
|
|
}
|
|
|
|
MemoryOperationsStatus WddmResidentAllocationsContainer::evictAllResources() {
|
|
decltype(resourceHandles) resourcesToEvict;
|
|
auto lock = acquireLock(resourcesLock);
|
|
resourceHandles.swap(resourcesToEvict);
|
|
if (resourcesToEvict.empty()) {
|
|
return MemoryOperationsStatus::MEMORY_NOT_FOUND;
|
|
}
|
|
uint64_t sizeToTrim = 0;
|
|
uint32_t evictedResources = static_cast<uint32_t>(resourcesToEvict.size());
|
|
bool success = wddm->evict(resourcesToEvict.data(), evictedResources, sizeToTrim);
|
|
return success ? MemoryOperationsStatus::SUCCESS : MemoryOperationsStatus::FAILED;
|
|
}
|
|
|
|
MemoryOperationsStatus WddmResidentAllocationsContainer::evictResource(const D3DKMT_HANDLE &handle) {
|
|
return evictResources(&handle, 1u);
|
|
}
|
|
|
|
MemoryOperationsStatus WddmResidentAllocationsContainer::evictResources(const D3DKMT_HANDLE *handles, const uint32_t count) {
|
|
auto lock = acquireLock(resourcesLock);
|
|
auto position = std::find(resourceHandles.begin(), resourceHandles.end(), handles[0]);
|
|
if (position == resourceHandles.end()) {
|
|
return MemoryOperationsStatus::MEMORY_NOT_FOUND;
|
|
}
|
|
auto distance = static_cast<size_t>(std::distance(resourceHandles.begin(), position));
|
|
UNRECOVERABLE_IF(distance + count > resourceHandles.size());
|
|
resourceHandles.erase(position, position + count);
|
|
uint64_t sizeToTrim = 0;
|
|
if (!wddm->evict(handles, count, sizeToTrim)) {
|
|
return MemoryOperationsStatus::FAILED;
|
|
}
|
|
return MemoryOperationsStatus::SUCCESS;
|
|
}
|
|
|
|
MemoryOperationsStatus WddmResidentAllocationsContainer::makeResidentResource(const D3DKMT_HANDLE &handle, size_t size) {
|
|
return makeResidentResources(&handle, 1u, size);
|
|
}
|
|
|
|
MemoryOperationsStatus WddmResidentAllocationsContainer::makeResidentResources(const D3DKMT_HANDLE *handles, const uint32_t count, size_t size) {
|
|
bool madeResident = false;
|
|
while (!(madeResident = wddm->makeResident(handles, count, false, nullptr, size))) {
|
|
if (evictAllResources() == MemoryOperationsStatus::SUCCESS) {
|
|
continue;
|
|
}
|
|
if (!wddm->makeResident(handles, count, false, nullptr, size)) {
|
|
DEBUG_BREAK_IF(true);
|
|
return MemoryOperationsStatus::OUT_OF_MEMORY;
|
|
};
|
|
break;
|
|
}
|
|
DEBUG_BREAK_IF(!madeResident);
|
|
auto lock = acquireLock(resourcesLock);
|
|
for (uint32_t i = 0; i < count; i++) {
|
|
resourceHandles.push_back(handles[i]);
|
|
}
|
|
lock.unlock();
|
|
wddm->waitOnPagingFenceFromCpu();
|
|
return madeResident ? MemoryOperationsStatus::SUCCESS : MemoryOperationsStatus::FAILED;
|
|
}
|
|
|
|
void WddmResidentAllocationsContainer::removeResource(const D3DKMT_HANDLE &handle) {
|
|
auto lock = acquireLock(resourcesLock);
|
|
auto position = std::find(resourceHandles.begin(), resourceHandles.end(), handle);
|
|
if (position == resourceHandles.end()) {
|
|
return;
|
|
}
|
|
*position = resourceHandles.back();
|
|
resourceHandles.pop_back();
|
|
}
|
|
|
|
} // namespace NEO
|