Correctly import compat layer from llama.cpp

Signed-off-by: fxzjshm <fxzjshm@163.com>

ktransformers/ktransformers_ext/CMakeLists.txt

@@ -211,6 +211,18 @@ endif()
 list(APPEND CMAKE_PREFIX_PATH  ${ROCM_PATH})
 list(APPEND CMAKE_PREFIX_PATH "${ROCM_PATH}/lib64/cmake")
 
+if (NOT EXISTS $ENV{MUSA_PATH})
+    if (NOT EXISTS /opt/musa)
+        set(MUSA_PATH /usr/local/musa)
+    else()
+        set(MUSA_PATH /opt/musa)
+    endif()
+else()
+    set(MUSA_PATH $ENV{MUSA_PATH})
+endif()
+
+list(APPEND CMAKE_MODULE_PATH "${MUSA_PATH}/cmake")
+
 add_compile_options("$<$<COMPILE_LANGUAGE:CXX>:${ARCH_FLAGS}>")
 add_compile_options("$<$<COMPILE_LANGUAGE:C>:${ARCH_FLAGS}>")
 
@@ -223,14 +235,14 @@ if (WIN32)
 elseif (UNIX)
     find_package(CUDA)
     find_package(HIP)
-    find_package(MUSA)
+    find_package(MUSAToolkit)
     if(CUDA_FOUND)
         include_directories("${CUDA_INCLUDE_DIRS}")
     endif()
     if(HIP_FOUND)
         include_directories("${HIP_INCLUDE_DIRS}")
     endif()
-    if(MUSA_FOUND)
+    if(MUSAToolkit_FOUND)
         include_directories("${MUSA_INCLUDE_DIRS}")
     endif()
 endif()
@@ -261,7 +273,7 @@ elseif(UNIX)
         target_link_libraries(${PROJECT_NAME} PRIVATE "${ROCM_PATH}/lib/libamdhip64.so")
         message(STATUS "Building for HIP")
     endif()
-    if(MUSA_FOUND)
+    if(MUSAToolkit_FOUND)
         add_compile_definitions(USE_MUSA=1)
         message(STATUS "Building for MUSA")
     endif()