From: Tomer Tayar <ttayar@habana.ai>
Date: Sun, 4 Aug 2019 07:03:41 +0000 (+0000)
Subject: habanalabs: fix DRAM usage accounting on context tear down
X-Git-Tag: v5.3-rc5~6^2~2^2~4
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=c8113756ba27298d6e95403c087dc5881b419a99;p=thirdparty%2Fkernel%2Flinux.git

habanalabs: fix DRAM usage accounting on context tear down

The patch fix the DRAM usage accounting by adding a missing update of
the DRAM memory consumption, when a context is being torn down without an
organized release of the allocated memory.

Signed-off-by: Tomer Tayar <ttayar@habana.ai>
Reviewed-by: Oded Gabbay <oded.gabbay@gmail.com>
Signed-off-by: Oded Gabbay <oded.gabbay@gmail.com>
---

diff --git a/drivers/misc/habanalabs/memory.c b/drivers/misc/habanalabs/memory.c
index 42d237cae1dc5..365fb0cb8dfff 100644
--- a/drivers/misc/habanalabs/memory.c
+++ b/drivers/misc/habanalabs/memory.c
@@ -1629,6 +1629,8 @@ void hl_vm_ctx_fini(struct hl_ctx *ctx)
 			dev_dbg(hdev->dev,
 				"page list 0x%p of asid %d is still alive\n",
 				phys_pg_list, ctx->asid);
+			atomic64_sub(phys_pg_list->total_size,
+					&hdev->dram_used_mem);
 			free_phys_pg_pack(hdev, phys_pg_list);
 			idr_remove(&vm->phys_pg_pack_handles, i);
 		}