1 files changed, 2 insertions, 6 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_virt.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_virt.c
index cca794a1f8ba..ff887639bfa3 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_virt.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_virt.c
@@ -70,9 +70,7 @@ uint32_t amdgpu_virt_kiq_rreg(struct amdgpu_device *adev, uint32_t reg)
         if (r < 1 && (adev->in_gpu_reset || in_interrupt()))
                 goto failed_kiq_read;
 
-        if (in_interrupt())
-                might_sleep();
-
+        might_sleep();
         while (r < 1 && cnt++ < MAX_KIQ_REG_TRY) {
                 msleep(MAX_KIQ_REG_BAILOUT_INTERVAL);
                 r = amdgpu_fence_wait_polling(ring, seq, MAX_KIQ_REG_WAIT);
@@ -118,9 +116,7 @@ void amdgpu_virt_kiq_wreg(struct amdgpu_device *adev, uint32_t reg, uint32_t v)
         if (r < 1 && (adev->in_gpu_reset || in_interrupt()))
                 goto failed_kiq_write;
 
-        if (in_interrupt())
-                might_sleep();
-
+        might_sleep();
         while (r < 1 && cnt++ < MAX_KIQ_REG_TRY) {
 
                 msleep(MAX_KIQ_REG_BAILOUT_INTERVAL);