1 files changed, 5 insertions, 0 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_object.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_object.c
index 4d084eeb4e09..0a0f3cb89f92 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_object.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_object.c
@@ -520,6 +520,11 @@ int amdgpu_bo_pin_restricted(struct amdgpu_bo *bo, u32 domain,
                 return -EINVAL;
 
         if (bo->pin_count) {
+                uint32_t mem_type = bo->tbo.mem.mem_type;
+
+                if (domain != amdgpu_mem_type_to_domain(mem_type))
+                        return -EINVAL;
+
                 bo->pin_count++;
                 if (gpu_addr)
                         *gpu_addr = amdgpu_bo_gpu_offset(bo);