TTM used to track the "acc_size" of all BOs internally. We needed to
keep track of it in our memory reservation to avoid TTM running out
of memory in its own accounting. However, that "acc_size" accounting
has since been removed from TTM. Therefore we don't really need to
track it any more.

Signed-off-by: Alex Sierra <alex.sie...@amd.com>
---
 .../gpu/drm/amd/amdgpu/amdgpu_amdkfd_gpuvm.c  | 57 ++++++-------------
 1 file changed, 16 insertions(+), 41 deletions(-)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_amdkfd_gpuvm.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_amdkfd_gpuvm.c
index fada3b149361..e985cf9c7ec0 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_amdkfd_gpuvm.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_amdkfd_gpuvm.c
@@ -108,17 +108,8 @@ void amdgpu_amdkfd_reserve_system_mem(uint64_t size)
  * compromise that should work in most cases without reserving too
  * much memory for page tables unnecessarily (factor 16K, >> 14).
  */
-#define ESTIMATE_PT_SIZE(mem_size) ((mem_size) >> 14)
-
-static size_t amdgpu_amdkfd_acc_size(uint64_t size)
-{
-       size >>= PAGE_SHIFT;
-       size *= sizeof(dma_addr_t) + sizeof(void *);
 
-       return __roundup_pow_of_two(sizeof(struct amdgpu_bo)) +
-               __roundup_pow_of_two(sizeof(struct ttm_tt)) +
-               PAGE_ALIGN(size);
-}
+#define ESTIMATE_PT_SIZE(mem_size) ((mem_size) >> 14)
 
 /**
  * amdgpu_amdkfd_reserve_mem_limit() - Decrease available memory by size
@@ -136,28 +127,22 @@ static int amdgpu_amdkfd_reserve_mem_limit(struct 
amdgpu_device *adev,
 {
        uint64_t reserved_for_pt =
                ESTIMATE_PT_SIZE(amdgpu_amdkfd_total_mem_size);
-       size_t acc_size, system_mem_needed, ttm_mem_needed, vram_needed;
+       size_t system_mem_needed, ttm_mem_needed, vram_needed;
        int ret = 0;
 
-       acc_size = amdgpu_amdkfd_acc_size(size);
-
+       system_mem_needed = 0;
+       ttm_mem_needed = 0;
        vram_needed = 0;
        if (alloc_flag & KFD_IOC_ALLOC_MEM_FLAGS_GTT) {
-               system_mem_needed = acc_size + size;
-               ttm_mem_needed = acc_size + size;
+               system_mem_needed = size;
+               ttm_mem_needed = size;
        } else if (alloc_flag & KFD_IOC_ALLOC_MEM_FLAGS_VRAM) {
-               system_mem_needed = acc_size;
-               ttm_mem_needed = acc_size;
                vram_needed = size;
        } else if (alloc_flag & KFD_IOC_ALLOC_MEM_FLAGS_USERPTR) {
-               system_mem_needed = acc_size + size;
-               ttm_mem_needed = acc_size;
-       } else if (alloc_flag &
-                  (KFD_IOC_ALLOC_MEM_FLAGS_DOORBELL |
-                   KFD_IOC_ALLOC_MEM_FLAGS_MMIO_REMAP)) {
-               system_mem_needed = acc_size;
-               ttm_mem_needed = acc_size;
-       } else {
+               system_mem_needed = size;
+       } else if (!(alloc_flag &
+                               (KFD_IOC_ALLOC_MEM_FLAGS_DOORBELL |
+                                KFD_IOC_ALLOC_MEM_FLAGS_MMIO_REMAP))) {
                pr_err("%s: Invalid BO type %#x\n", __func__, alloc_flag);
                return -ENOMEM;
        }
@@ -193,28 +178,18 @@ static int amdgpu_amdkfd_reserve_mem_limit(struct 
amdgpu_device *adev,
 static void unreserve_mem_limit(struct amdgpu_device *adev,
                uint64_t size, u32 alloc_flag)
 {
-       size_t acc_size;
-
-       acc_size = amdgpu_amdkfd_acc_size(size);
-
        spin_lock(&kfd_mem_limit.mem_limit_lock);
 
        if (alloc_flag & KFD_IOC_ALLOC_MEM_FLAGS_GTT) {
-               kfd_mem_limit.system_mem_used -= (acc_size + size);
-               kfd_mem_limit.ttm_mem_used -= (acc_size + size);
+               kfd_mem_limit.system_mem_used -= size;
+               kfd_mem_limit.ttm_mem_used -= size;
        } else if (alloc_flag & KFD_IOC_ALLOC_MEM_FLAGS_VRAM) {
-               kfd_mem_limit.system_mem_used -= acc_size;
-               kfd_mem_limit.ttm_mem_used -= acc_size;
                adev->kfd.vram_used -= size;
        } else if (alloc_flag & KFD_IOC_ALLOC_MEM_FLAGS_USERPTR) {
-               kfd_mem_limit.system_mem_used -= (acc_size + size);
-               kfd_mem_limit.ttm_mem_used -= acc_size;
-       } else if (alloc_flag &
-                  (KFD_IOC_ALLOC_MEM_FLAGS_DOORBELL |
-                   KFD_IOC_ALLOC_MEM_FLAGS_MMIO_REMAP)) {
-               kfd_mem_limit.system_mem_used -= acc_size;
-               kfd_mem_limit.ttm_mem_used -= acc_size;
-       } else {
+               kfd_mem_limit.system_mem_used -= size;
+       } else if (!(alloc_flag &
+                               (KFD_IOC_ALLOC_MEM_FLAGS_DOORBELL |
+                                KFD_IOC_ALLOC_MEM_FLAGS_MMIO_REMAP))) {
                pr_err("%s: Invalid BO type %#x\n", __func__, alloc_flag);
                goto release;
        }
-- 
2.32.0

Reply via email to