Commit 17f3f42a authored by Dani Liberman's avatar Dani Liberman Committed by Oded Gabbay
Browse files

habanalabs: handle HBM MMU when capturing page fault data 



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

Signed-off-by: default avatarDani Liberman <dliberman@habana.ai>
Reviewed-by: default avatarOded Gabbay <ogabbay@kernel.org>
Signed-off-by: default avatarOded Gabbay <ogabbay@kernel.org>
parent 1eebb259

drivers/misc/habanalabs/common/device.c

+21 −8
Original line number Diff line number Diff line
@@ -2273,15 +2273,20 @@ void hl_capture_razwi(struct hl_device *hdev, u64 addr, u16 *engine_id, u16 num_ 
			num_of_engines * sizeof(u16));

	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 hl_vm_phys_pg_pack *phys_pg_pack = NULL;

	struct hl_vm_hash_node *hnode;

	struct hl_userptr *userptr;

	enum vm_type *vm_type;

	struct hl_ctx *ctx;

	u32 map_idx = 0;

	int i;



	/* Reset previous session count*/

	pgf_info->num_of_user_mappings = 0;



	ctx = hl_get_compute_ctx(hdev);

	if (!ctx) {

		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);

	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) {

		pgf_info->num_of_user_mappings = 0;

		goto finish;

	}



	hash_for_each(ctx->mem_hash, i, hnode, node) {

		vm_type = hnode->ptr;

		if ((*vm_type == VM_TYPE_USERPTR) && (is_pmmu)) {

			userptr = hnode->ptr;

			pgf_info->user_mappings[map_idx].dev_va = hnode->vaddr;

			pgf_info->user_mappings[map_idx].size = userptr->size;

			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:

	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.addr = addr;

	hdev->captured_err_info.pgf_info.pgf.engine_id = eng_id;

	hl_capture_user_mappings(hdev);

	hl_capture_user_mappings(hdev, is_pmmu);

}