1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/infiniband/hw/cxgb4/mem.c b/drivers/infiniband/hw/cxgb4/mem.c
index f9ca072a99ed..ec7a2988a703 100644
--- a/drivers/infiniband/hw/cxgb4/mem.c
+++ b/drivers/infiniband/hw/cxgb4/mem.c
@@ -259,8 +259,12 @@ static int write_tpt_entry(struct c4iw_rdev *rdev, u32 reset_tpt_entry,
 
         if ((!reset_tpt_entry) && (*stag == T4_STAG_UNSET)) {
                 stag_idx = c4iw_get_resource(&rdev->resource.tpt_table);
-                if (!stag_idx)
+                if (!stag_idx) {
+                        mutex_lock(&rdev->stats.lock);
+                        rdev->stats.stag.fail++;
+                        mutex_unlock(&rdev->stats.lock);
                         return -ENOMEM;
+                }
                 mutex_lock(&rdev->stats.lock);
                 rdev->stats.stag.cur += 32;
                 if (rdev->stats.stag.cur > rdev->stats.stag.max)