[MUSA][19/N] Support HiCache with pin_memory allocator (#23361)

Signed-off-by: yafeng.li <yafeng.li@mthreads.com>
2026-04-21 19:45:53 -07:00
parent 7c399f3c82
commit bf5e71dcec
1 changed files with 1 additions and 0 deletions
@@ -147,6 +147,7 @@ ALLOC_MEMORY_FUNCS = defaultdict(
    lambda: alloc_with_host_register,
    {
        "npu": alloc_with_pin_memory,
+        "musa": alloc_with_pin_memory,
    },
 )