1 files changed, 5 insertions, 2 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_sa.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_sa.c
index 8b88edb0434b..7d8f8f1e3f7f 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_sa.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_sa.c
@@ -321,8 +321,11 @@ int amdgpu_sa_bo_new(struct amdgpu_sa_manager *sa_manager,
         int i, r;
         signed long t;
 
-        BUG_ON(align > sa_manager->align);
-        BUG_ON(size > sa_manager->size);
+        if (WARN_ON_ONCE(align > sa_manager->align))
+                return -EINVAL;
+
+        if (WARN_ON_ONCE(size > sa_manager->size))
+                return -EINVAL;
 
         *sa_bo = kmalloc(sizeof(struct amdgpu_sa_bo), GFP_KERNEL);
         if ((*sa_bo) == NULL) {