From 6ec4fa821efd4b3e949821eb283836058b32d103 Mon Sep 17 00:00:00 2001 From: sunjqa1 Date: Thu, 26 Feb 2026 02:58:17 +0000 Subject: [PATCH 1/2] Add inference, training and communication test support --- scripts/common/install_deps.sh | 323 +++++++++++++++++++++++++++++++++ scripts/run_tests.sh | 25 +++ 2 files changed, 348 insertions(+) diff --git a/scripts/common/install_deps.sh b/scripts/common/install_deps.sh index 94a54f6..c5b1a5c 100755 --- a/scripts/common/install_deps.sh +++ b/scripts/common/install_deps.sh @@ -5,6 +5,9 @@ # Components: # operator - InfiniCore (operator testing) # hardware - CUDA memory benchmark (hardware testing) +# inference - Inference frameworks (vLLM, InfiniLM) +# training - Training frameworks (InfiniTrain, Megatron) +# comm - Communication tests (NCCL tests) # all - All components (default) if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then @@ -25,6 +28,9 @@ NC='\033[0m' # ======================================== export INFINI_ROOT="$HOME/.infini" export LD_LIBRARY_PATH="$INFINI_ROOT/lib:$LD_LIBRARY_PATH" +export NCCL_ROOT="$INFINI_ROOT/nccl" +export PATH="$NCCL_ROOT/bin:$PATH" +export LD_LIBRARY_PATH="$NCCL_ROOT/lib:$LD_LIBRARY_PATH" # Safe exit function that uses return when sourced, exit when executed directly safe_exit() { @@ -67,6 +73,18 @@ check_cuda() { fi } +# Check NCCL +check_nccl() { + echo -n " NCCL... " + if [ -d "$NCCL_ROOT" ] || ldconfig -p 2>/dev/null | grep -q libnccl; then + echo -e "${GREEN}[OK]${NC}" + return 0 + else + echo -e "${YELLOW}[WARNING]${NC} not found" + return 1 + fi +} + # Check InfiniCore check_infinicore() { echo -n " InfiniCore... " @@ -79,6 +97,75 @@ check_infinicore() { fi } +# Check vLLM +check_vllm() { + echo -n " vLLM... " + if check_python_package vllm; then + echo -e "${GREEN}[OK]${NC}" + return 0 + else + echo -e "${YELLOW}[WARNING]${NC} not installed" + return 1 + fi +} + +# Check InfiniLM +check_infinilm() { + echo -n " InfiniLM... " + if python -c "import infinilm" 2>/dev/null; then + local version=$(python -c "import infinilm; print(getattr(infinilm, '__version__', 'unknown'))" 2>/dev/null || echo "unknown") + echo -e "${GREEN}[OK]${NC} (version $version)" + return 0 + else + # Check if directory exists but not in PYTHONPATH + local INFINILM_PATH="${INFINILM_PATH:-$HOME/InfiniLM}" + if [ -d "$INFINILM_PATH/python/infinilm" ]; then + echo -e "${YELLOW}[WARNING]${NC} found at $INFINILM_PATH but not in PYTHONPATH" + echo "Run: export PYTHONPATH=$INFINILM_PATH/python:\$PYTHONPATH" + else + echo -e "${YELLOW}[WARNING]${NC} not installed" + fi + return 1 + fi +} + +# Check Megatron-LM +check_megatron() { + echo -n " Megatron-LM... " + if check_python_package megatron; then + echo -e "${GREEN}[OK]${NC}" + return 0 + else + echo -e "${YELLOW}[WARNING]${NC} not installed" + return 1 + fi +} + +# Check InfiniTrain +check_infinitrain() { + echo -n " InfiniTrain... " + if check_python_package infinitrain; then + echo -e "${GREEN}[OK]${NC}" + return 0 + else + echo -e "${YELLOW}[WARNING]${NC} not installed" + return 1 + fi +} + +# Check NCCL tests +check_nccl_tests() { + echo -n " NCCL tests... " + local NCCL_TESTS_PATH="$(cd "$(dirname "${BASH_SOURCE[0]}")/../../submodules/nccl-tests" && pwd)" + if [ -f "$NCCL_TESTS_PATH/build/all_reduce_perf" ]; then + echo -e "${GREEN}[OK]${NC}" + return 0 + else + echo -e "${YELLOW}[WARNING]${NC} not built" + return 1 + fi +} + # ======================================== # InfiniCore (Operator Testing) # ======================================== @@ -237,6 +324,212 @@ install_hardware() { fi } +# ======================================== +# NCCL Tests (Communication) +# ======================================== +install_comm() { + echo "" + echo "==========================================" + echo "NCCL Tests (Communication)" + echo "==========================================" + echo "" + + local SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" + local PROJECT_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)" + local NCCL_TESTS_PATH="$PROJECT_ROOT/submodules/nccl-tests" + + # 1. Check CUDA + echo -e "${BLUE}Checking dependencies...${NC}" + if ! check_cuda; then + echo -e "${RED}[ERROR] CUDA is required for NCCL tests${NC}" + return 1 + fi + echo "" + + # 2. Check if already built + echo -e "${BLUE}Checking NCCL tests...${NC}" + if [ -f "$NCCL_TESTS_PATH/build/all_reduce_perf" ]; then + echo -e "${GREEN}[OK] NCCL tests already built${NC}" + return 0 + fi + + echo -e "${YELLOW}[WARNING] NCCL tests not found, building...${NC}" + echo "" + + # 3. Build NCCL tests + if [ ! -d "$NCCL_TESTS_PATH" ]; then + echo -e "${RED}[ERROR] NCCL tests not found at: $NCCL_TESTS_PATH${NC}" + echo "Please ensure submodules are initialized:" + echo " git submodule update --init --recursive" + return 1 + fi + + cd "$NCCL_TESTS_PATH" || { + echo -e "${RED}[ERROR] Failed to change directory to $NCCL_TESTS_PATH${NC}" + return 1 + } + + echo -e "${BLUE}Building NCCL tests...${NC}" + if ! make -j; then + echo -e "${RED}[ERROR] Failed to build NCCL tests${NC}" + return 1 + fi + echo "" + + # 4. Verify + if [ -f "build/all_reduce_perf" ]; then + echo -e "${GREEN}[OK] NCCL tests built successfully!${NC}" + return 0 + else + echo -e "${RED}[ERROR] NCCL tests build failed${NC}" + return 1 + fi +} + +# ======================================== +# Inference Frameworks +# ======================================== +install_inference() { + echo "" + echo "==========================================" + echo "Inference Frameworks" + echo "==========================================" + echo "" + + local exit_code=0 + local SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" + local PROJECT_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)" + + # 1. Install vLLM + echo -e "${BLUE}Installing vLLM...${NC}" + if check_python_package vllm; then + echo -e "${GREEN}[OK] vLLM already installed${NC}" + else + if pip install vllm; then + echo -e "${GREEN}[OK] vLLM installed${NC}" + else + echo -e "${RED}[ERROR] Failed to install vLLM${NC}" + exit_code=1 + fi + fi + echo "" + + # 2. Install InfiniLM + echo -e "${BLUE}Installing InfiniLM...${NC}" + + # Check if already installed + if python -c "import infinilm" 2>/dev/null; then + echo -e "${GREEN}[OK] InfiniLM already installed${NC}" + return 0 + fi + + # Determine InfiniLM path + local INFINILM_PATH="" + + # Priority 1: Environment variable + if [ -n "$INFINILM_PATH" ] && [ -d "$INFINILM_PATH" ]; then + INFINILM_PATH="$INFINILM_PATH" + # Priority 2: Home directory + elif [ -d "$HOME/InfiniLM" ]; then + INFINILM_PATH="$HOME/InfiniLM" + # Priority 3: Parent directory + elif [ -d "$PROJECT_ROOT/../InfiniLM" ]; then + INFINILM_PATH="$(cd "$PROJECT_ROOT/../InfiniLM" && pwd)" + fi + + if [ -z "$INFINILM_PATH" ] || [ ! -d "$INFINILM_PATH" ]; then + echo -e "${YELLOW}[WARNING] InfiniLM directory not found${NC}" + echo "" + echo "Please set INFINILM_PATH environment variable:" + echo " export INFINILM_PATH=/home/sunjinge/InfiniLM" + echo "" + echo "Then run this script again" + return $exit_code + fi + + echo -e "${BLUE}Found InfiniLM at: $INFINILM_PATH${NC}" + + # Check Python package structure + local PYTHON_PACKAGE_PATH="$INFINILM_PATH/python" + + if [ -d "$PYTHON_PACKAGE_PATH" ]; then + echo -e "${BLUE}Found Python package at: $PYTHON_PACKAGE_PATH${NC}" + + # Add to PYTHONPATH for current session + export PYTHONPATH="$PYTHON_PACKAGE_PATH:$PYTHONPATH" + + # Add to .bashrc for future sessions + if ! grep -q "INFINILM_PATH" ~/.bashrc; then + echo "# InfiniLM configuration" >> ~/.bashrc + echo "export INFINILM_PATH=\"$INFINILM_PATH\"" >> ~/.bashrc + echo "export PYTHONPATH=\"$PYTHON_PACKAGE_PATH:\$PYTHONPATH\"" >> ~/.bashrc + echo -e "${GREEN}[OK] Added InfiniLM paths to ~/.bashrc${NC}" + fi + + # Check if we can import now + if python -c "import infinilm" 2>/dev/null; then + echo -e "${GREEN}[OK] InfiniLM successfully configured${NC}" + else + echo -e "${YELLOW}[WARNING] InfiniLM found but import failed${NC}" + echo "Current PYTHONPATH: $PYTHONPATH" + echo "You may need to install additional dependencies" + fi + else + echo -e "${YELLOW}[WARNING] No Python package found at $PYTHON_PACKAGE_PATH${NC}" + echo "Contents of $INFINILM_PATH:" + ls -la "$INFINILM_PATH" + fi + + return $exit_code +} + +# ======================================== +# Training Frameworks +# ======================================== +install_training() { + echo "" + echo "==========================================" + echo "Training Frameworks" + echo "==========================================" + echo "" + + local exit_code=0 + + # 1. Install Megatron-LM + echo -e "${BLUE}Installing Megatron-LM...${NC}" + local MEGATRON_PATH="${MEGATRON_PATH:-$PROJECT_ROOT/submodules/Megatron-LM}" + if [ -d "$MEGATRON_PATH" ]; then + cd "$MEGATRON_PATH" + if pip install -e .; then + echo -e "${GREEN}[OK] Megatron-LM installed${NC}" + else + echo -e "${RED}[ERROR] Failed to install Megatron-LM${NC}" + exit_code=1 + fi + else + echo -e "${YELLOW}[WARNING] Megatron-LM not found at $MEGATRON_PATH${NC}" + fi + echo "" + + # 2. Install InfiniTrain + echo -e "${BLUE}Installing InfiniTrain...${NC}" + local INFINITRAIN_PATH="${INFINITRAIN_PATH:-$PROJECT_ROOT/submodules/InfiniTrain}" + if [ -d "$INFINITRAIN_PATH" ]; then + cd "$INFINITRAIN_PATH" + if pip install -e .; then + echo -e "${GREEN}[OK] InfiniTrain installed${NC}" + else + echo -e "${RED}[ERROR] Failed to install InfiniTrain${NC}" + exit_code=1 + fi + else + echo -e "${YELLOW}[WARNING] InfiniTrain not found at $INFINITRAIN_PATH${NC}" + fi + echo "" + + return $exit_code +} + # ======================================== # Main # ======================================== @@ -246,15 +539,24 @@ show_usage() { echo "Components:" echo " operator - InfiniCore (operator testing)" echo " hardware - CUDA memory benchmark (hardware testing)" + echo " inference - Inference frameworks (vLLM, InfiniLM)" + echo " training - Training frameworks (Megatron-LM, InfiniTrain)" + echo " comm - Communication tests (NCCL tests)" echo " all - All components (default)" echo "" echo "Environment variables:" echo " INFINICORE_PATH - Path to InfiniCore source (required for operator)" + echo " INFINILM_PATH - Path to InfiniLM source" + echo " INFINITRAIN_PATH - Path to InfiniTrain source" + echo " MEGATRON_PATH - Path to Megatron-LM source" echo "" echo "Examples:" echo " export INFINICORE_PATH=\"\$HOME/workplace/InfiniCore\"" + echo " export INFINILM_PATH=\"\$HOME/InfiniLM\"" echo " $0 operator # Install InfiniCore" echo " $0 hardware # Build CUDA benchmark" + echo " $0 inference # Install inference frameworks" + echo " $0 comm # Build NCCL tests" echo " $0 all # Install everything" echo " $0 # Install everything (default)" } @@ -277,6 +579,18 @@ main() { install_hardware exit_code=$? ;; + inference) + install_inference + exit_code=$? + ;; + training) + install_training + exit_code=$? + ;; + comm) + install_comm + exit_code=$? + ;; all) install_infinicore exit_code=$? @@ -287,6 +601,15 @@ main() { install_hardware exit_code=$? + + install_inference + exit_code=$? + + install_training + exit_code=$? + + install_comm + exit_code=$? ;; --help|-h) show_usage diff --git a/scripts/run_tests.sh b/scripts/run_tests.sh index 8e6baa9..7fdbb9b 100755 --- a/scripts/run_tests.sh +++ b/scripts/run_tests.sh @@ -56,6 +56,7 @@ while [[ $# -gt 0 ]]; do echo " $0 --check all test.json # Check all dependencies" echo " $0 --check hardware test.json # Check hardware deps only" echo " $0 --check hardware,operator test.json" + echo " $0 --check inference,comm test.json # Check inference and comm deps" exit 0 ;; *) @@ -81,6 +82,7 @@ if [ ${#INPUT_PATHS[@]} -eq 0 ]; then echo " $0 --check all test.json # Check all dependencies" echo " $0 --check hardware test.json # Check hardware deps only" echo " $0 --check hardware,operator test.json" + echo " $0 --check inference,comm test.json # Check inference and comm deps" echo "" echo "Run '$0 --help' for more information" exit 1 @@ -106,10 +108,31 @@ check_all_deps() { echo "Checking operator dependencies:" check_infinicore ;; + inference) + echo "Checking inference dependencies:" + check_vllm + check_infinilm + ;; + training) + echo "Checking training dependencies:" + check_megatron + check_infinitrain + ;; + comm) + echo "Checking communication dependencies:" + check_nccl + check_nccl_tests + ;; all) echo "Checking all dependencies:" check_cuda + check_nccl check_infinicore + check_vllm + check_infinilm + check_megatron + check_infinitrain + check_nccl_tests ;; *) echo "[WARNING] Unknown dependency type: $dtype" @@ -140,6 +163,8 @@ main() { # Display environment echo "Environment: INFINI_ROOT=$INFINI_ROOT" + echo "NCCL_ROOT=$NCCL_ROOT" + echo "CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES:-not set}" echo "" # Run tests From bb343451445fbc52a235f75e5b6219872965b20b Mon Sep 17 00:00:00 2001 From: sunjqa1 Date: Sat, 28 Feb 2026 09:06:18 +0000 Subject: [PATCH 2/2] feat(scripts):remove nccl_root and Impove training module handling --- scripts/common/install_deps.sh | 120 ++++++++++++++++----------------- scripts/run_tests.sh | 10 +-- 2 files changed, 64 insertions(+), 66 deletions(-) diff --git a/scripts/common/install_deps.sh b/scripts/common/install_deps.sh index c5b1a5c..58391fe 100755 --- a/scripts/common/install_deps.sh +++ b/scripts/common/install_deps.sh @@ -26,11 +26,11 @@ NC='\033[0m' # ======================================== # Environment Variables (always exported) # ======================================== +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" +PROJECT_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)" + export INFINI_ROOT="$HOME/.infini" export LD_LIBRARY_PATH="$INFINI_ROOT/lib:$LD_LIBRARY_PATH" -export NCCL_ROOT="$INFINI_ROOT/nccl" -export PATH="$NCCL_ROOT/bin:$PATH" -export LD_LIBRARY_PATH="$NCCL_ROOT/lib:$LD_LIBRARY_PATH" # Safe exit function that uses return when sourced, exit when executed directly safe_exit() { @@ -73,18 +73,6 @@ check_cuda() { fi } -# Check NCCL -check_nccl() { - echo -n " NCCL... " - if [ -d "$NCCL_ROOT" ] || ldconfig -p 2>/dev/null | grep -q libnccl; then - echo -e "${GREEN}[OK]${NC}" - return 0 - else - echo -e "${YELLOW}[WARNING]${NC} not found" - return 1 - fi -} - # Check InfiniCore check_infinicore() { echo -n " InfiniCore... " @@ -129,26 +117,29 @@ check_infinilm() { fi } -# Check Megatron-LM +# TODO: Megatron-LM check - currently assumes submodule usage check_megatron() { echo -n " Megatron-LM... " if check_python_package megatron; then - echo -e "${GREEN}[OK]${NC}" + echo -e "${GREEN}[OK]${NC} (installed as package)" + return 0 + elif [ -d "$PROJECT_ROOT/submodules/Megatron-LM" ]; then + echo -e "${GREEN}[OK]${NC} (found as submodule)" return 0 else - echo -e "${YELLOW}[WARNING]${NC} not installed" + echo -e "${YELLOW}[WARNING]${NC} not found (expected as submodule or Python package)" return 1 fi } -# Check InfiniTrain +# TODO: InfiniTrain check - currently not available as Python package check_infinitrain() { echo -n " InfiniTrain... " - if check_python_package infinitrain; then - echo -e "${GREEN}[OK]${NC}" + if [ -d "$PROJECT_ROOT/submodules/InfiniTrain" ]; then + echo -e "${GREEN}[OK]${NC} (found as submodule)" return 0 else - echo -e "${YELLOW}[WARNING]${NC} not installed" + echo -e "${YELLOW}[WARNING]${NC} not found (expected as submodule)" return 1 fi } @@ -397,10 +388,16 @@ install_inference() { echo "" local exit_code=0 - local SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" - local PROJECT_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)" - # 1. Install vLLM + # 1. Check InfiniCore dependency (required for InfiniLM) + echo -e "${BLUE}Checking InfiniCore (required for InfiniLM)...${NC}" + if ! check_infinicore; then + echo -e "${YELLOW}[WARNING] InfiniCore not found, InfiniLM may not work${NC}" + echo "Consider running: $0 operator" + fi + echo "" + + # 2. Install vLLM echo -e "${BLUE}Installing vLLM...${NC}" if check_python_package vllm; then echo -e "${GREEN}[OK] vLLM already installed${NC}" @@ -414,17 +411,17 @@ install_inference() { fi echo "" - # 2. Install InfiniLM + # 3. Install InfiniLM echo -e "${BLUE}Installing InfiniLM...${NC}" # Check if already installed if python -c "import infinilm" 2>/dev/null; then echo -e "${GREEN}[OK] InfiniLM already installed${NC}" - return 0 + return $exit_code fi # Determine InfiniLM path - local INFINILM_PATH="" + local INFINILM_PATH="${INFINILM_PATH:-}" # Priority 1: Environment variable if [ -n "$INFINILM_PATH" ] && [ -d "$INFINILM_PATH" ]; then @@ -441,7 +438,7 @@ install_inference() { echo -e "${YELLOW}[WARNING] InfiniLM directory not found${NC}" echo "" echo "Please set INFINILM_PATH environment variable:" - echo " export INFINILM_PATH=/home/sunjinge/InfiniLM" + echo "export INFINILM_PATH=/home/sunjinge/InfiniLM" echo "" echo "Then run this script again" return $exit_code @@ -459,12 +456,11 @@ install_inference() { export PYTHONPATH="$PYTHON_PACKAGE_PATH:$PYTHONPATH" # Add to .bashrc for future sessions - if ! grep -q "INFINILM_PATH" ~/.bashrc; then - echo "# InfiniLM configuration" >> ~/.bashrc - echo "export INFINILM_PATH=\"$INFINILM_PATH\"" >> ~/.bashrc - echo "export PYTHONPATH=\"$PYTHON_PACKAGE_PATH:\$PYTHONPATH\"" >> ~/.bashrc - echo -e "${GREEN}[OK] Added InfiniLM paths to ~/.bashrc${NC}" - fi + echo "" + echo -e "${YELLOW}To make InfiniLM permanently available, add these lines to your ~/.bashrc:${NC}" + echo " export INFINILM_PATH=\"$INFINILM_PATH\"" + echo " export PYTHONPATH=\"$PYTHON_PACKAGE_PATH:\$PYTHONPATH\"" + echo "" # Check if we can import now if python -c "import infinilm" 2>/dev/null; then @@ -495,36 +491,38 @@ install_training() { local exit_code=0 - # 1. Install Megatron-LM - echo -e "${BLUE}Installing Megatron-LM...${NC}" - local MEGATRON_PATH="${MEGATRON_PATH:-$PROJECT_ROOT/submodules/Megatron-LM}" - if [ -d "$MEGATRON_PATH" ]; then - cd "$MEGATRON_PATH" - if pip install -e .; then - echo -e "${GREEN}[OK] Megatron-LM installed${NC}" - else - echo -e "${RED}[ERROR] Failed to install Megatron-LM${NC}" - exit_code=1 - fi - else - echo -e "${YELLOW}[WARNING] Megatron-LM not found at $MEGATRON_PATH${NC}" + # TODO: Megatron-LM installation - currently assumes submodule usage + echo -e "${BLUE}Setting up Megatron-LM...${NC}" + + local MEGATRON_PATH="" + + if [ -n "$MEGATRON_PATH" ] && [ -d "$MEGATRON_PATH" ]; then + MEGATRON_PATH="$MEGATRON_PATH" + + elif [ -d "$HOME/Megatron-LM" ]; then + MEGATRON_PATH="$HOME/Megatron-LM" + + elif [ -d "$PROJECT_ROOT/submodules/Megatron-LM" ]; then + MEGATRON_PATH="$PROJECT_ROOT/submodules/Megatron-LM" + fi - echo "" - # 2. Install InfiniTrain - echo -e "${BLUE}Installing InfiniTrain...${NC}" - local INFINITRAIN_PATH="${INFINITRAIN_PATH:-$PROJECT_ROOT/submodules/InfiniTrain}" - if [ -d "$INFINITRAIN_PATH" ]; then - cd "$INFINITRAIN_PATH" - if pip install -e .; then - echo -e "${GREEN}[OK] InfiniTrain installed${NC}" - else - echo -e "${RED}[ERROR] Failed to install InfiniTrain${NC}" - exit_code=1 - fi + if [ -z "$MEGATRON_PATH" ]; then + echo -e "${YELLOW}[WARNING] Megatron-LM not found${NC}" + echo "" + echo "Supported locations:" + echo " \$MEGATRON_PATH" + echo " \$HOME/Megatron-LM" + echo " submodules/Megatron-LM" else - echo -e "${YELLOW}[WARNING] InfiniTrain not found at $INFINITRAIN_PATH${NC}" + echo -e "${GREEN}[OK] Megatron-LM found at $MEGATRON_PATH${NC}" fi + + echo "" + + # TODO: InfiniTrain setup - to be implemented + echo -e "${BLUE}Setting up InfiniTrain...${NC}" + echo -e "${YELLOW}[INFO] InfiniTrain setup not yet implemented${NC}" echo "" return $exit_code diff --git a/scripts/run_tests.sh b/scripts/run_tests.sh index 7fdbb9b..54ae7f6 100755 --- a/scripts/run_tests.sh +++ b/scripts/run_tests.sh @@ -20,8 +20,11 @@ set -e SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" -source "${SCRIPT_DIR}/common/install_deps.sh" -source "${SCRIPT_DIR}/common/prepare_env.sh" +COMMON_DIR="${SCRIPT_DIR}/common" + +# Source common scripts +source "${COMMON_DIR}/install_deps.sh" +source "${COMMON_DIR}/prepare_env.sh" # Default values CHECK_DEPS="false" @@ -120,13 +123,11 @@ check_all_deps() { ;; comm) echo "Checking communication dependencies:" - check_nccl check_nccl_tests ;; all) echo "Checking all dependencies:" check_cuda - check_nccl check_infinicore check_vllm check_infinilm @@ -163,7 +164,6 @@ main() { # Display environment echo "Environment: INFINI_ROOT=$INFINI_ROOT" - echo "NCCL_ROOT=$NCCL_ROOT" echo "CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES:-not set}" echo ""