#!/bin/bash
# start_occupy.sh

LOG_FILE="/home/jenkins/.npu-scripts/occupy.log"

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
NC='\033[0m' # No Color

echo -e "${GREEN}[$(date)]: Starting occupy_npu_8cards.py${NC}" | tee -a "$LOG_FILE"

# Load CANN environment (consistent with manual execution)
source /usr/local/Ascend/ascend-toolkit/set_env.sh
export LD_LIBRARY_PATH=/usr/local/Ascend/driver/lib64/driver:$LD_LIBRARY_PATH

cd /home/jenkins/.npu-scripts
nohup python3 -u occupy_npu_8cards.py >> "$LOG_FILE" 2>&1 &

# Record PID (stop script can use pkill, but recording PID is more precise)
echo $! > /home/jenkins/.npu-scripts/occupy.pid
echo -e "${GREEN}[$(date)]: Started with PID $(cat /home/jenkins/.npu-scripts/occupy.pid)${NC}" | tee -a "$LOG_FILE"