Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 11 additions & 11 deletions src/gpgmm/d3d12/HeapD3D12.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -81,17 +81,17 @@ namespace gpgmm::d3d12 {
if (FAILED(residencyManager->EnsureInBudget(descriptor.SizeInBytes,
descriptor.MemorySegmentGroup))) {
DXGI_QUERY_VIDEO_MEMORY_INFO currentVideoInfo = {};
ReturnIfFailed(residencyManager->QueryVideoMemoryInfo(descriptor.MemorySegmentGroup,
&currentVideoInfo));

gpgmm::ErrorLog(MessageId::kBudgetExceeded)
<< "Unable to create heap because not enough budget exists ("
<< GPGMM_BYTES_TO_MB(descriptor.SizeInBytes) << " vs "
<< GPGMM_BYTES_TO_MB((currentVideoInfo.Budget > currentVideoInfo.CurrentUsage)
? currentVideoInfo.Budget -
currentVideoInfo.CurrentUsage
: 0)
<< " MBs) and HEAP_FLAG_ALWAYS_IN_BUDGET was specified.";
if (SUCCEEDED(residencyManager->QueryVideoMemoryInfo(descriptor.MemorySegmentGroup,
&currentVideoInfo))) {
gpgmm::ErrorLog(MessageId::kBudgetExceeded)
<< "Unable to create heap because not enough budget exists ("
<< GPGMM_BYTES_TO_MB(descriptor.SizeInBytes) << " vs "
<< GPGMM_BYTES_TO_MB(
(currentVideoInfo.Budget > currentVideoInfo.CurrentUsage)
? currentVideoInfo.Budget - currentVideoInfo.CurrentUsage
: 0)
<< " MBs) and HEAP_FLAG_ALWAYS_IN_BUDGET was specified.";
}

return E_OUTOFMEMORY;
}
Expand Down
24 changes: 14 additions & 10 deletions src/gpgmm/d3d12/ResidencyManagerD3D12.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -479,38 +479,42 @@ namespace gpgmm::d3d12 {
static_cast<uint64_t>(queryVideoMemoryInfoOut.Budget * mMinPctOfBudgetToReserve),
pVideoMemoryInfo->AvailableForReservation);

const uint64_t oldUsage = pVideoMemoryInfo->CurrentUsage;
const uint64_t previousUsage = pVideoMemoryInfo->CurrentUsage;
pVideoMemoryInfo->CurrentUsage =
queryVideoMemoryInfoOut.CurrentUsage - pVideoMemoryInfo->CurrentReservation;

if (oldUsage > pVideoMemoryInfo->CurrentUsage) {
if (previousUsage > pVideoMemoryInfo->CurrentUsage &&
GPGMM_BYTES_TO_MB(previousUsage - pVideoMemoryInfo->CurrentUsage) > 0) {
gpgmm::DebugLog() << GetMemorySegmentName(memorySegmentGroup, mIsUMA)
<< " GPU memory usage went down by "
<< GPGMM_BYTES_TO_MB(oldUsage - pVideoMemoryInfo->CurrentUsage)
<< GPGMM_BYTES_TO_MB(previousUsage - pVideoMemoryInfo->CurrentUsage)
<< " MBs.";
} else if (oldUsage < pVideoMemoryInfo->CurrentUsage) {
} else if (previousUsage < pVideoMemoryInfo->CurrentUsage &&
GPGMM_BYTES_TO_MB(pVideoMemoryInfo->CurrentUsage - previousUsage) > 0) {
gpgmm::DebugLog() << GetMemorySegmentName(memorySegmentGroup, mIsUMA)
<< " GPU memory usage went up by "
<< GPGMM_BYTES_TO_MB(pVideoMemoryInfo->CurrentUsage - oldUsage)
<< GPGMM_BYTES_TO_MB(pVideoMemoryInfo->CurrentUsage - previousUsage)
<< " MBs.";
}

// If we're restricting the budget, leave the budget as is.
if (!mIsBudgetRestricted) {
const uint64_t oldBudget = pVideoMemoryInfo->Budget;
const uint64_t previousBudget = pVideoMemoryInfo->Budget;
pVideoMemoryInfo->Budget = static_cast<uint64_t>(
(queryVideoMemoryInfoOut.Budget - pVideoMemoryInfo->CurrentReservation) *
mMaxPctOfVideoMemoryToBudget);

if (oldBudget > pVideoMemoryInfo->Budget) {
if (previousBudget > pVideoMemoryInfo->Budget &&
GPGMM_BYTES_TO_MB(previousBudget - pVideoMemoryInfo->Budget) > 0) {
gpgmm::DebugLog() << GetMemorySegmentName(memorySegmentGroup, mIsUMA)
<< " GPU memory budget went down by "
<< GPGMM_BYTES_TO_MB(oldBudget - pVideoMemoryInfo->Budget)
<< GPGMM_BYTES_TO_MB(previousBudget - pVideoMemoryInfo->Budget)
<< " MBs.";
} else if (oldBudget < pVideoMemoryInfo->Budget) {
} else if (previousBudget < pVideoMemoryInfo->Budget &&
GPGMM_BYTES_TO_MB(pVideoMemoryInfo->Budget - previousBudget) > 0) {
gpgmm::DebugLog() << GetMemorySegmentName(memorySegmentGroup, mIsUMA)
<< " GPU memory budget went up by "
<< GPGMM_BYTES_TO_MB(pVideoMemoryInfo->Budget - oldBudget)
<< GPGMM_BYTES_TO_MB(pVideoMemoryInfo->Budget - previousBudget)
<< " MBs.";
}
}
Expand Down