diff --git a/lnet/include/linux/kp30.h b/lnet/include/linux/kp30.h
index 63eaee76295cc109af679e8bf9d4d33bf537d1a2..12d9f56d3bfc935162aab492311f24dda0ab0125 100644
--- a/lnet/include/linux/kp30.h
+++ b/lnet/include/linux/kp30.h
@@ -282,9 +282,8 @@ do {                                                                          \
 
 #define PORTAL_ALLOC_GFP(ptr, size, mask)                                 \
 do {                                                                      \
-        LASSERT (!in_interrupt() ||                                       \
-                 (size <= PORTAL_VMALLOC_SIZE &&                          \
-                  (mask & __GFP_WAIT) == 0));                             \
+        LASSERT(!in_interrupt() ||                                        \
+               (size <= PORTAL_VMALLOC_SIZE && mask == GFP_ATOMIC));      \
         if ((size) > PORTAL_VMALLOC_SIZE)                                 \
                 (ptr) = vmalloc(size);                                    \
         else                                                              \
diff --git a/lustre/portals/include/linux/kp30.h b/lustre/portals/include/linux/kp30.h
index 63eaee76295cc109af679e8bf9d4d33bf537d1a2..12d9f56d3bfc935162aab492311f24dda0ab0125 100644
--- a/lustre/portals/include/linux/kp30.h
+++ b/lustre/portals/include/linux/kp30.h
@@ -282,9 +282,8 @@ do {                                                                          \
 
 #define PORTAL_ALLOC_GFP(ptr, size, mask)                                 \
 do {                                                                      \
-        LASSERT (!in_interrupt() ||                                       \
-                 (size <= PORTAL_VMALLOC_SIZE &&                          \
-                  (mask & __GFP_WAIT) == 0));                             \
+        LASSERT(!in_interrupt() ||                                        \
+               (size <= PORTAL_VMALLOC_SIZE && mask == GFP_ATOMIC));      \
         if ((size) > PORTAL_VMALLOC_SIZE)                                 \
                 (ptr) = vmalloc(size);                                    \
         else                                                              \