[PATCH 2/2] drm/amdgpu: Add address check for uniras

[Jinzhou Su]

Add address check for uniras

Signed-off-by: Jinzhou Su <jinzhou...@amd.com>
---
 drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c
index 97eece7d8504..3271bbf75cd4 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c
@@ -237,8 +237,13 @@ static int amdgpu_check_address_validity(struct 
amdgpu_device *adev,
            (address >= RAS_UMC_INJECT_ADDR_LIMIT))
                return -EFAULT;
 
-       count = amdgpu_umc_lookup_bad_pages_in_a_row(adev,
+       if (amdgpu_uniras_enabled(adev))
+               count = amdgpu_ras_mgr_lookup_bad_pages_in_a_row(adev, address,
+                       page_pfns, ARRAY_SIZE(page_pfns));
+       else
+               count = amdgpu_umc_lookup_bad_pages_in_a_row(adev,
                                address, page_pfns, ARRAY_SIZE(page_pfns));
+
        if (count <= 0)
                return -EPERM;
 
-- 
2.43.0