diff --git a/python/sglang/srt/mem_cache/memory_pool_host.py b/python/sglang/srt/mem_cache/memory_pool_host.py index 3132d9844..f45c1d616 100644 --- a/python/sglang/srt/mem_cache/memory_pool_host.py +++ b/python/sglang/srt/mem_cache/memory_pool_host.py @@ -147,6 +147,7 @@ ALLOC_MEMORY_FUNCS = defaultdict( lambda: alloc_with_host_register, { "npu": alloc_with_pin_memory, + "musa": alloc_with_pin_memory, }, )