1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/infiniband/hw/mthca/mthca_cmd.c b/drivers/infiniband/hw/mthca/mthca_cmd.c
index 968d1519761c..71314460b11e 100644
--- a/drivers/infiniband/hw/mthca/mthca_cmd.c
+++ b/drivers/infiniband/hw/mthca/mthca_cmd.c
@@ -1051,7 +1051,11 @@ int mthca_QUERY_DEV_LIM(struct mthca_dev *dev,
         MTHCA_GET(field, outbox, QUERY_DEV_LIM_MAX_EQ_OFFSET);
         dev_lim->max_eqs = 1 << (field & 0x7);
         MTHCA_GET(field, outbox, QUERY_DEV_LIM_RSVD_MTT_OFFSET);
-        dev_lim->reserved_mtts = 1 << (field >> 4);
+        if (mthca_is_memfree(dev))
+                dev_lim->reserved_mtts = ALIGN((1 << (field >> 4)) * sizeof(u64),
+                                               MTHCA_MTT_SEG_SIZE) / MTHCA_MTT_SEG_SIZE;
+        else
+                dev_lim->reserved_mtts = 1 << (field >> 4);
         MTHCA_GET(field, outbox, QUERY_DEV_LIM_MAX_MRW_SZ_OFFSET);
         dev_lim->max_mrw_sz = 1 << field;
         MTHCA_GET(field, outbox, QUERY_DEV_LIM_RSVD_MRW_OFFSET);