RDMA/hns: Add releasing resource operation in error branch

This patch adds releasing resource operation in error branch
of the function named hns_roce_table_get.

Signed-off-by: Wei Hu (Xavier) <xavier.huwei@huawei.com>
Signed-off-by: Doug Ledford <dledford@redhat.com>

diff --git a/drivers/infiniband/hw/hns/hns_roce_hem.c b/drivers/infiniband/hw/hns/hns_roce_hem.c
index 8a3e174f5b5fa97931113e84d5750befc2961ee4..125e26ebf2507d9733a66d2a4b2ce5d50b093cd6 100644 (file)
--- a/drivers/infiniband/hw/hns/hns_roce_hem.c
+++ b/drivers/infiniband/hw/hns/hns_roce_hem.c
@@ -563,6 +563,8 @@ int hns_roce_table_get(struct hns_roce_dev *hr_dev,
 
        /* Set HEM base address(128K/page, pa) to Hardware */
        if (hns_roce_set_hem(hr_dev, table, obj)) {
+               hns_roce_free_hem(hr_dev, table->hem[i]);
+               table->hem[i] = NULL;
                ret = -ENODEV;
                dev_err(dev, "set HEM base address to HW failed.\n");
                goto out;