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: Ohad Sharabi <osharabi@habana.ai>
Reviewed-by: Oded Gabbay <ogabbay@kernel.org>
Signed-off-by: Oded Gabbay <ogabbay@kernel.org>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

diff --git a/drivers/misc/habanalabs/common/memory.c b/drivers/misc/habanalabs/common/memory.c
index 326c2179628f7cf4805ba0dbb88c8fb42a8f7ea7..6face45c57e35adc007395712d599cca8570fb0d 100644 (file)
--- a/drivers/misc/habanalabs/common/memory.c
+++ b/drivers/misc/habanalabs/common/memory.c
@@ -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 */
-       rc = hl_mmu_prefetch_cache_range(hdev, *vm_type, ctx->asid, ret_vaddr,
-                                               phys_pg_pack->total_size);
-       if (rc)
-               goto map_err;
+       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);
 

diff --git a/include/uapi/misc/habanalabs.h b/include/uapi/misc/habanalabs.h
index ca2af5f980567bbf5264df7d06b4e7eb6e806de2..3576bf2b4841be34f7aa3c3ba946bc4b6eaea2e7 100644 (file)
--- a/include/uapi/misc/habanalabs.h
+++ b/include/uapi/misc/habanalabs.h
@@ -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