Commit 5d1a0de2 authored by Ohad Sharabi's avatar Ohad Sharabi Committed by Greg Kroah-Hartman
Browse files

habanalabs: add prefetch flag to the MAP operation 



This patch let the user decide whether the translations done in the
page tables will be fetched directly to the STLB right after the map.

We want to let the user control whether to perform prefetch upon map
operation.

To do so a memory flag was added, to be used in the MAP ioctl, called
HL_MEM_PREFETCH and if set- the mappings will be fetched directly to
the STLB after map operation.

Signed-off-by: default avatarOhad Sharabi <osharabi@habana.ai>
Reviewed-by: default avatarOded Gabbay <ogabbay@kernel.org>
Signed-off-by: default avatarOded Gabbay <ogabbay@kernel.org>
Signed-off-by: default avatarGreg Kroah-Hartman <gregkh@linuxfoundation.org>
parent 77c97a7e

drivers/misc/habanalabs/common/memory.c

+6 −5
Original line number Diff line number Diff line
@@ -1250,11 +1250,12 @@ static int map_device_va(struct hl_ctx *ctx, struct hl_mem_in *args, 
	if (rc)

		goto map_err;



	/* already prefetch the relevant translations to the cache */

	if (args->flags & HL_MEM_PREFETCH) {

		rc = hl_mmu_prefetch_cache_range(hdev, *vm_type, ctx->asid, ret_vaddr,

							phys_pg_pack->total_size);

		if (rc)

			goto map_err;

	}



	mutex_unlock(&ctx->mmu_lock);

include/uapi/misc/habanalabs.h

+1 −0
Original line number Diff line number Diff line
@@ -1134,6 +1134,7 @@ union hl_wait_cs_args { 
#define HL_MEM_SHARED		0x2

#define HL_MEM_USERPTR		0x4

#define HL_MEM_FORCE_HINT	0x8

#define HL_MEM_PREFETCH		0x40



/**

 * structure hl_mem_in - structure that handle input args for memory IOCTL