1 files changed, 3 insertions, 9 deletions

diff --git a/block/bio-integrity.c b/block/bio-integrity.c
index 9e241063a616..bc423f7b02da 100644
--- a/block/bio-integrity.c
+++ b/block/bio-integrity.c
@@ -70,8 +70,10 @@ struct bio_integrity_payload *bio_integrity_alloc(struct bio *bio,
                                           bs->bvec_integrity_pool);
                 if (!bip->bip_vec)
                         goto err;
+                bip->bip_max_vcnt = bvec_nr_vecs(idx);
         } else {
                 bip->bip_vec = bip->bip_inline_vecs;
+                bip->bip_max_vcnt = inline_vecs;
         }
 
         bip->bip_slab = idx;
@@ -114,14 +116,6 @@ void bio_integrity_free(struct bio *bio)
 }
 EXPORT_SYMBOL(bio_integrity_free);
 
-static inline unsigned int bip_integrity_vecs(struct bio_integrity_payload *bip)
-{
-        if (bip->bip_slab == BIO_POOL_NONE)
-                return BIP_INLINE_VECS;
-
-        return bvec_nr_vecs(bip->bip_slab);
-}
-
 /**
  * bio_integrity_add_page - Attach integrity metadata
  * @bio:        bio to update
@@ -137,7 +131,7 @@ int bio_integrity_add_page(struct bio *bio, struct page *page,
         struct bio_integrity_payload *bip = bio->bi_integrity;
         struct bio_vec *iv;
 
-        if (bip->bip_vcnt >= bip_integrity_vecs(bip)) {
+        if (bip->bip_vcnt >= bip->bip_max_vcnt) {
                 printk(KERN_ERR "%s: bip_vec full\n", __func__);
                 return 0;
         }