Signed-off-by: Chunming Zhou <David1.Zhou@amd.com>
Reviewed-by: Christian König <christian.koenig@amd.com>
Reviewed-by: Junwei Zhang <Jerry.Zhang@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>
 
 static void amdgpu_get_block_size(struct amdgpu_device *adev)
 {
+       /* from AI, asic starts to support multiple level VMPT */
+       if (adev->family >= AMDGPU_FAMILY_AI) {
+               if (amdgpu_vm_block_size != 9)
+                       dev_warn(adev->dev, "Multi-VMPT limits block size to"
+                                "one page!\n");
+               amdgpu_vm_block_size = 9;
+               return;
+       }
        /* defines number of bits in page table versus page directory,
         * a page is 4KB so we have 12 bits offset, minimum 9 bits in the
         * page table and the remaining bits are in the page directory */