habanalabs: if map page fails don't try to unmap it

The original code tried to unmap a page that was not mapped as part of
the map page error path.

Signed-off-by: Dafna Hirschfeld <dhirschfeld@habana.ai>
Reviewed-by: Oded Gabbay <ogabbay@kernel.org>
Signed-off-by: Oded Gabbay <ogabbay@kernel.org>

diff --git a/drivers/misc/habanalabs/common/mmu/mmu.c b/drivers/misc/habanalabs/common/mmu/mmu.c
index 60740de47b34ef66e215664fd4725436a03861e3..4153aec555949bf9308351c9fdc43e6ebde2ea88 100644 (file)
--- a/drivers/misc/habanalabs/common/mmu/mmu.c
+++ b/drivers/misc/habanalabs/common/mmu/mmu.c
@@ -403,6 +403,8 @@ int hl_mmu_map_contiguous(struct hl_ctx *ctx, u64 virt_addr,
                        dev_err(hdev->dev,
                                "Map failed for va 0x%llx to pa 0x%llx\n",
                                curr_va, curr_pa);
+                       /* last mapping failed so don't try to unmap it - reduce off by page_size */
+                       off -= page_size;
                        goto unmap;
                }
        }