drm/amdgpu: increase fragmentation size for Vega10 v2

The fragment bits work differently for Vega10 compared to previous generations.

Increase the fragment size to 2MB for now to better handle that.

v2: handle the hardware setup as well

Signed-off-by: Christian König <christian.koenig@amd.com>
Reviewed-and-tested-by: Felix Kuehling <Felix.Kuehling@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>

1 files changed, 3 insertions, 2 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
index b017b54e45ba..088c8504dbe7 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
@@ -1381,8 +1381,9 @@ static int amdgpu_vm_frag_ptes(struct amdgpu_pte_update_params	*params,
          */
 
         /* SI and newer are optimized for 64KB */
-        uint64_t frag_flags = AMDGPU_PTE_FRAG(AMDGPU_LOG2_PAGES_PER_FRAG);
-        uint64_t frag_align = 1 << AMDGPU_LOG2_PAGES_PER_FRAG;
+        unsigned pages_per_frag = AMDGPU_LOG2_PAGES_PER_FRAG(params->adev);
+        uint64_t frag_flags = AMDGPU_PTE_FRAG(pages_per_frag);
+        uint64_t frag_align = 1 << pages_per_frag;
 
         uint64_t frag_start = ALIGN(start, frag_align);
         uint64_t frag_end = end & ~(frag_align - 1);