From f2daa2d97ec16766bc36fc2b6218d0acb47885ed Mon Sep 17 00:00:00 2001 From: Yuri Nudelman Date: Mon, 2 May 2022 13:41:11 +0300 Subject: [PATCH] habanalabs: add topic to memory manager buffer Currently, buffers from multiple flows pass through the same infra. This way, in logs, we are unable to distinguish between buffers that came from separate flows. To address this problem, add a "topic" to buffer behavior descriptor - a string identifier that will be used to identify in logs the flow this buffer relates to. Signed-off-by: Yuri Nudelman Reviewed-by: Oded Gabbay Signed-off-by: Oded Gabbay Signed-off-by: Greg Kroah-Hartman --- .../misc/habanalabs/common/command_buffer.c | 1 + drivers/misc/habanalabs/common/habanalabs.h | 2 ++ drivers/misc/habanalabs/common/memory.c | 1 + drivers/misc/habanalabs/common/memory_mgr.c | 23 +++++++++++-------- 4 files changed, 18 insertions(+), 9 deletions(-) diff --git a/drivers/misc/habanalabs/common/command_buffer.c b/drivers/misc/habanalabs/common/command_buffer.c index fd9ef32ea6a02..1fac72c38c87a 100644 --- a/drivers/misc/habanalabs/common/command_buffer.c +++ b/drivers/misc/habanalabs/common/command_buffer.c @@ -319,6 +319,7 @@ static int hl_cb_mmap(struct hl_mmap_mem_buf *buf, } static struct hl_mmap_mem_buf_behavior cb_behavior = { + .topic = "CB", .mem_id = HL_MMAP_TYPE_CB, .alloc = hl_cb_mmap_mem_alloc, .release = hl_cb_mmap_mem_release, diff --git a/drivers/misc/habanalabs/common/habanalabs.h b/drivers/misc/habanalabs/common/habanalabs.h index 59150caa98a2d..918e8a04acaba 100644 --- a/drivers/misc/habanalabs/common/habanalabs.h +++ b/drivers/misc/habanalabs/common/habanalabs.h @@ -731,6 +731,7 @@ struct hl_mem_mgr { /** * struct hl_mmap_mem_buf_behavior - describes unified memory manager buffer behavior + * @topic: string identifier used for logging * @mem_id: memory type identifier, embedded in the handle and used to identify * the memory type by handle. * @alloc: callback executed on buffer allocation, shall allocate the memory, @@ -739,6 +740,7 @@ struct hl_mem_mgr { * @release: callback executed on release, must free the resources used by the buffer */ struct hl_mmap_mem_buf_behavior { + const char *topic; u64 mem_id; int (*alloc)(struct hl_mmap_mem_buf *buf, gfp_t gfp, void *args); diff --git a/drivers/misc/habanalabs/common/memory.c b/drivers/misc/habanalabs/common/memory.c index e7a0c44c487db..ecf3c094242ae 100644 --- a/drivers/misc/habanalabs/common/memory.c +++ b/drivers/misc/habanalabs/common/memory.c @@ -2141,6 +2141,7 @@ free_mem: } static struct hl_mmap_mem_buf_behavior hl_ts_behavior = { + .topic = "TS", .mem_id = HL_MMAP_TYPE_TS_BUFF, .mmap = hl_ts_mmap, .alloc = hl_ts_alloc_buf, diff --git a/drivers/misc/habanalabs/common/memory_mgr.c b/drivers/misc/habanalabs/common/memory_mgr.c index 9f3ab6cf25d3e..0ddfebe3a9ef6 100644 --- a/drivers/misc/habanalabs/common/memory_mgr.c +++ b/drivers/misc/habanalabs/common/memory_mgr.c @@ -162,7 +162,8 @@ hl_mmap_mem_buf_alloc(struct hl_mem_mgr *mmg, spin_unlock(&mmg->lock); if (rc < 0) { dev_err(mmg->dev, - "Failed to allocate IDR for a new buffer, rc=%d\n", rc); + "%s: Failed to allocate IDR for a new buffer, rc=%d\n", + behavior->topic, rc); goto free_buf; } @@ -173,8 +174,8 @@ hl_mmap_mem_buf_alloc(struct hl_mem_mgr *mmg, rc = buf->behavior->alloc(buf, gfp, args); if (rc) { - dev_err(mmg->dev, "Failure in buffer alloc callback %d\n", - rc); + dev_err(mmg->dev, "%s: Failure in buffer alloc callback %d\n", + behavior->topic, rc); goto remove_idr; } @@ -253,8 +254,8 @@ int hl_mem_mgr_mmap(struct hl_mem_mgr *mmg, struct vm_area_struct *vma, user_mem_size = vma->vm_end - vma->vm_start; if (user_mem_size != ALIGN(buf->mappable_size, PAGE_SIZE)) { dev_err(mmg->dev, - "Memory mmap failed, mmap VM size 0x%llx != 0x%llx allocated physical mem size\n", - user_mem_size, buf->mappable_size); + "%s: Memory mmap failed, mmap VM size 0x%llx != 0x%llx allocated physical mem size\n", + buf->behavior->topic, user_mem_size, buf->mappable_size); rc = -EINVAL; goto put_mem; } @@ -266,8 +267,8 @@ int hl_mem_mgr_mmap(struct hl_mem_mgr *mmg, struct vm_area_struct *vma, if (!access_ok((void __user *)(uintptr_t)vma->vm_start, user_mem_size)) { #endif - dev_err(mmg->dev, "user pointer is invalid - 0x%lx\n", - vma->vm_start); + dev_err(mmg->dev, "%s: User pointer is invalid - 0x%lx\n", + buf->behavior->topic, vma->vm_start); rc = -EINVAL; goto put_mem; @@ -275,7 +276,8 @@ int hl_mem_mgr_mmap(struct hl_mem_mgr *mmg, struct vm_area_struct *vma, if (atomic_cmpxchg(&buf->mmap, 0, 1)) { dev_err(mmg->dev, - "Memory mmap failed, already mmaped to user\n"); + "%s, Memory mmap failed, already mmaped to user\n", + buf->behavior->topic); rc = -EINVAL; goto put_mem; } @@ -328,14 +330,17 @@ void hl_mem_mgr_fini(struct hl_mem_mgr *mmg) { struct hl_mmap_mem_buf *buf; struct idr *idp; + const char *topic; u32 id; idp = &mmg->handles; idr_for_each_entry(idp, buf, id) { + topic = buf->behavior->topic; if (hl_mmap_mem_buf_put(buf) != 1) dev_err(mmg->dev, - "Buff handle %u for CTX is still alive\n", id); + "%s: Buff handle %u for CTX is still alive\n", + topic, id); } /* TODO: can it happen that some buffer is still in use at this point? */ -- 2.39.5