#!/usr/bin/env bash
set -o pipefail
source /usr/local/Ascend/ascend-toolkit/set_env.sh 2>/dev/null || source /usr/local/Ascend/cann/set_env.sh 2>/dev/null || true
source /data/vllm-ascend-smoke/venv/bin/activate
export VLLM_TARGET_DEVICE=ascend
export ASCEND_RT_VISIBLE_DEVICES=${ASCEND_RT_VISIBLE_DEVICES:-0}
export PYTORCH_NPU_ALLOC_CONF=${PYTORCH_NPU_ALLOC_CONF:-expandable_segments:True}
export VLLM_LOGGING_LEVEL=${VLLM_LOGGING_LEVEL:-INFO}
python /data/vllm-ascend-smoke/run_smoke.py 2>&1 | tee /data/vllm-ascend-smoke/logs/smoke-rerun.log
