proxmox-mirrors/mirror_ubuntu-kernels
1
0
Fork 0
mirror of https://git.proxmox.com/git/mirror_ubuntu-kernels.git synced 2025-12-23 13:32:36 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

habanalabs: handle HBM MMU when capturing page fault data

Browse Source 
In case of HBM MMU page fault, capture its relevant mappings.

Signed-off-by: Dani Liberman <dliberman@habana.ai>
Reviewed-by: Oded Gabbay <ogabbay@kernel.org>
Signed-off-by: Oded Gabbay <ogabbay@kernel.org>
This commit is contained in:
Dani Liberman 2022-09-29 10:21:28 +03:00 committed by Oded Gabbay
parent 1eebb25929
commit 17f3f42af2
1 changed files with 21 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
drivers/misc/habanalabs/common/device.c
View File

@ -2273,15 +2273,20 @@ void hl_capture_razwi(struct hl_device *hdev, u64 addr, u16 *engine_id, u16 num_
num_of_engines * sizeof(u16)); num_of_engines * sizeof(u16));
hdev->captured_err_info.razwi.flags = flags; hdev->captured_err_info.razwi.flags = flags;
} }
static void hl_capture_user_mappings(struct hl_device *hdev) static void hl_capture_user_mappings(struct hl_device *hdev, bool is_pmmu)
{ {
struct page_fault_info *pgf_info = &hdev->captured_err_info.pgf_info; struct page_fault_info *pgf_info = &hdev->captured_err_info.pgf_info;
struct hl_vm_phys_pg_pack *phys_pg_pack = NULL;
struct hl_vm_hash_node *hnode; struct hl_vm_hash_node *hnode;
struct hl_userptr *userptr; struct hl_userptr *userptr;
enum vm_type *vm_type;
struct hl_ctx *ctx; struct hl_ctx *ctx;
u32 map_idx = 0; u32 map_idx = 0;
int i; int i;
/* Reset previous session count*/
pgf_info->num_of_user_mappings = 0;
ctx = hl_get_compute_ctx(hdev); ctx = hl_get_compute_ctx(hdev);
if (!ctx) { if (!ctx) {
dev_err(hdev->dev, "Can't get user context for user mappings\n"); dev_err(hdev->dev, "Can't get user context for user mappings\n");
@ -2300,17 +2305,25 @@ static void hl_capture_user_mappings(struct hl_device *hdev)
*/ */
vfree(pgf_info->user_mappings); vfree(pgf_info->user_mappings);
pgf_info->user_mappings = pgf_info->user_mappings =
vmalloc(pgf_info->num_of_user_mappings * sizeof(struct hl_user_mapping)); vzalloc(pgf_info->num_of_user_mappings * sizeof(struct hl_user_mapping));
if (!pgf_info->user_mappings) { if (!pgf_info->user_mappings) {
pgf_info->num_of_user_mappings = 0; pgf_info->num_of_user_mappings = 0;
goto finish; goto finish;
} }
hash_for_each(ctx->mem_hash, i, hnode, node) { hash_for_each(ctx->mem_hash, i, hnode, node) {
vm_type = hnode->ptr;
if ((*vm_type == VM_TYPE_USERPTR) && (is_pmmu)) {
userptr = hnode->ptr; userptr = hnode->ptr;
pgf_info->user_mappings[map_idx].dev_va = hnode->vaddr; pgf_info->user_mappings[map_idx].dev_va = hnode->vaddr;
pgf_info->user_mappings[map_idx].size = userptr->size; pgf_info->user_mappings[map_idx].size = userptr->size;
map_idx++; map_idx++;
} else if ((*vm_type == VM_TYPE_PHYS_PACK) && (!is_pmmu)) {
phys_pg_pack = hnode->ptr;
pgf_info->user_mappings[map_idx].dev_va = hnode->vaddr;
pgf_info->user_mappings[map_idx].size = phys_pg_pack->total_size;
map_idx++;
}
} }
finish: finish:
mutex_unlock(&ctx->mem_hash_lock); mutex_unlock(&ctx->mem_hash_lock);
@ -2326,5 +2339,5 @@ void hl_capture_page_fault(struct hl_device *hdev, u64 addr, u16 eng_id, bool is
hdev->captured_err_info.pgf_info.pgf.timestamp = ktime_to_ns(ktime_get()); hdev->captured_err_info.pgf_info.pgf.timestamp = ktime_to_ns(ktime_get());
hdev->captured_err_info.pgf_info.pgf.addr = addr; hdev->captured_err_info.pgf_info.pgf.addr = addr;
hdev->captured_err_info.pgf_info.pgf.engine_id = eng_id; hdev->captured_err_info.pgf_info.pgf.engine_id = eng_id;
hl_capture_user_mappings(hdev); hl_capture_user_mappings(hdev, is_pmmu);
} }