1 files changed, 4 insertions, 3 deletions

diff --git a/lib/scatterlist.c b/lib/scatterlist.c
index 0d475d8167bf..e6dcd3b0d974 100644
--- a/lib/scatterlist.c
+++ b/lib/scatterlist.c
@@ -9,6 +9,7 @@
 #include <linux/module.h>
 #include <linux/scatterlist.h>
 #include <linux/highmem.h>
+#include <linux/interrupt.h>
 
 /**
  * sg_next - return the next scatterlist entry in a list
@@ -399,7 +400,7 @@ void sg_miter_stop(struct sg_mapping_iter *miter)
                         flush_kernel_dcache_page(miter->page);
 
                 if (miter->__flags & SG_MITER_ATOMIC) {
-                        WARN_ON(!irqs_disabled());
+                        WARN_ON_NONRT(!irqs_disabled());
                         kunmap_atomic(miter->addr, KM_BIO_SRC_IRQ);
                 } else
                         kunmap(miter->page);
@@ -439,7 +440,7 @@ static size_t sg_copy_buffer(struct scatterlist *sgl, unsigned int nents,
 
         sg_miter_start(&miter, sgl, nents, sg_flags);
 
-        local_irq_save(flags);
+        local_irq_save_nort(flags);
 
         while (sg_miter_next(&miter) && offset < buflen) {
                 unsigned int len;
@@ -456,7 +457,7 @@ static size_t sg_copy_buffer(struct scatterlist *sgl, unsigned int nents,
 
         sg_miter_stop(&miter);
 
-        local_irq_restore(flags);
+        local_irq_restore_nort(flags);
         return offset;
 }