feat: add dependencies install script.

Author: Kang-Meng

setup.py

@@ -445,7 +445,7 @@ def check_and_install_pre_commit():
             print("Run 'pre-commit install' failed. Please install pre-commit: pip install pre-commit")
             exit(0)
 
-def run_git_command(command, cwd=None, check=True):
+def run_shell_command(command, cwd=None, check=True):
     try:
         subprocess.run(command, cwd=cwd, check=check, shell=True, capture_output=True, text=True)
         return True
@@ -492,15 +492,15 @@ def apply_patch_safely(patch_file_path, repo_path):
 
     if has_uncommitted_changes(repo_path):
         print(f"⚠️ Uncommitted changes detected. Running `git reset --hard` for {repo_path}")
-        if not run_git_command("git reset --hard", cwd=repo_path):
+        if not run_shell_command("git reset --hard", cwd=repo_path):
             print("❌ Failed to reset changes!")
             return False
 
     print(f"🛠️ Apply patch: {patch_file_path}")
-    apply_success = run_git_command(f"git apply --check {patch_file_path}", cwd=repo_path, check=False)
+    apply_success = run_shell_command(f"git apply --check {patch_file_path}", cwd=repo_path, check=False)
 
     if apply_success:
-        if not run_git_command(f"git apply {patch_file_path}", cwd=repo_path):
+        if not run_shell_command(f"git apply {patch_file_path}", cwd=repo_path):
             print("❌ apply patch fail!")
             apply_success = False
 
@@ -512,7 +512,7 @@ def apply_patch_safely(patch_file_path, repo_path):
         print(f"  cd {repo_path} && git apply {patch_file_path}")
         return False
 
-def apply_patch():
+def pre_build():
     if os.path.exists("third_party/custom_patch"):
         script_path = os.path.dirname(os.path.abspath(__file__))
         mooncake_repo_path = os.path.join(script_path, "third_party/Mooncake")
@@ -521,6 +521,9 @@ def apply_patch():
         cpprestsdk_repo_path = os.path.join(script_path, "third_party/cpprestsdk")
         if not apply_patch_safely("../custom_patch/cpprestsdk.patch", cpprestsdk_repo_path):
             exit(0)
+        if not run_shell_command("sh third_party/dependencies.sh", cwd=script_path):
+            print("❌ Failed to reset changes!")
+            exit(0)
 
 if __name__ == "__main__":
     device = 'a2'  # default
@@ -537,9 +540,10 @@ def apply_patch():
             del sys.argv[idx]
             del sys.argv[idx]
     if '--dry_run' not in sys.argv:
-        apply_patch()
+        pre_build()
     else:
         sys.argv.remove("--dry_run") 
+    
     if '--install-xllm-kernels' in sys.argv:
         idx = sys.argv.index('--install-xllm-kernels')
         if idx + 1 < len(sys.argv):

third_party/dependencies.sh

@@ -0,0 +1,96 @@
+#!/bin/bash
+
+# Color definitions
+GREEN="\033[0;32m"
+BLUE="\033[0;34m"
+YELLOW="\033[0;33m"
+RED="\033[0;31m"
+NC="\033[0m" # No Color
+
+# Configuration
+REPO_ROOT=`pwd`
+
+# Function to print section headers
+print_section() {
+    echo -e "\n${BLUE}=== $1 ===${NC}"
+}
+
+# Function to print success messages
+print_success() {
+    echo -e "${GREEN}✓ $1${NC}"
+}
+
+# Function to print error messages and exit
+print_error() {
+    echo -e "${RED}✗ ERROR: $1${NC}"
+    exit 1
+}
+
+# Function to check command success
+check_success() {
+    if [ $? -ne 0 ]; then
+        print_error "$1"
+    fi
+}
+
+if [ $(id -u) -ne 0 ]; then
+	print_error "Require root permission, try sudo ./dependencies.sh"
+fi
+
+
+# Install yalantinglibs
+print_section "Installing yalantinglibs"
+
+# Check if thirdparties directory exists
+if [ ! -d "${REPO_ROOT}/third_party/Mooncake/thirdparties" ]; then
+    mkdir -p "${REPO_ROOT}/third_party/Mooncake/thirdparties"
+    check_success "Failed to create Mooncake/thirdparties directory"
+fi
+
+# Change to thirdparties directory
+cd "${REPO_ROOT}/third_party/Mooncake/thirdparties"
+check_success "Failed to change to Mooncake/thirdparties directory"
+
+# Check if yalantinglibs is already installed
+if [ -d "yalantinglibs" ]; then
+    echo -e "${YELLOW}yalantinglibs directory already exists. Removing for fresh install...${NC}"
+    rm -rf yalantinglibs
+    check_success "Failed to remove existing yalantinglibs directory"
+fi
+
+# Clone yalantinglibs
+echo "Cloning yalantinglibs from https://github.com/alibaba/yalantinglibs.git"
+git clone https://github.com/alibaba/yalantinglibs.git
+check_success "Failed to clone yalantinglibs"
+
+# Build and install yalantinglibs
+cd yalantinglibs
+check_success "Failed to change to yalantinglibs directory"
+
+# Checkout version 0.5.5
+echo "Checking out yalantinglibs version 0.5.5..."
+git checkout 0.5.5
+check_success "Failed to checkout yalantinglibs version 0.5.5"
+
+mkdir -p build
+check_success "Failed to create build directory"
+
+cd build
+check_success "Failed to change to build directory"
+
+echo "Configuring yalantinglibs..."
+cmake .. -DBUILD_EXAMPLES=OFF -DBUILD_BENCHMARK=OFF -DBUILD_UNIT_TESTS=OFF -DYLT_ENABLE_IBV=ON 
+check_success "Failed to configure yalantinglibs"
+
+echo "Building yalantinglibs (using $(nproc) cores)..."
+cmake --build . -j$(nproc)
+check_success "Failed to build yalantinglibs"
+
+echo "Installing yalantinglibs..."
+cmake --install .
+check_success "Failed to install yalantinglibs"
+
+sed -i '54s/target_link_libraries(${ylt_target_name} -libverbs)/target_link_libraries(${ylt_target_name} INTERFACE -libverbs)/' /usr/local/lib/cmake/yalantinglibs/config.cmake
+
+print_success "yalantinglibs installed successfully"
+