wehub-resource-sync
59 lines
1.7 KiB
Bash
59 lines
1.7 KiB
Bash
#!/bin/bash
|
|
MODELSCOPE_CACHE_DIR=/modelscope_cache
|
|
CODE_DIR=$PWD
|
|
MODELSCOPE_SDK_DEBUG=True
|
|
echo "$USER"
|
|
npus='0,1 2,3'
|
|
is_get_file_lock=false
|
|
CI_COMMAND=${CI_COMMAND:-bash .dev_scripts/ci_container_test.sh python tests/run.py --parallel 2 --run_config tests/run_config.yaml}
|
|
echo "ci command: $CI_COMMAND"
|
|
PR_CHANGED_FILES="${PR_CHANGED_FILES:-}"
|
|
echo "PR modified files: $PR_CHANGED_FILES"
|
|
PR_CHANGED_FILES=${PR_CHANGED_FILES//[ ]/#}
|
|
echo "PR_CHANGED_FILES: $PR_CHANGED_FILES"
|
|
idx=0
|
|
for npu in $npus
|
|
do
|
|
exec {lock_fd}>"/tmp/npu$npu" || exit 1
|
|
flock -n "$lock_fd" || { echo "WARN: npu $npu is in use!" >&2; idx=$((idx+1)); continue; }
|
|
echo "get npu lock $npu"
|
|
|
|
let is_get_file_lock=true
|
|
|
|
# 设置环境变量
|
|
export CI_TEST=True
|
|
export SWIFT_CI_USE_NPU=True
|
|
export TEST_LEVEL=$TEST_LEVEL
|
|
export MODELSCOPE_CACHE=${MODELSCOPE_CACHE:-$MODELSCOPE_CACHE_DIR}
|
|
export MODELSCOPE_DOMAIN=$MODELSCOPE_DOMAIN
|
|
export HUB_DATASET_ENDPOINT=$HUB_DATASET_ENDPOINT
|
|
export TEST_ACCESS_TOKEN_CITEST=$TEST_ACCESS_TOKEN_CITEST
|
|
export TEST_ACCESS_TOKEN_SDKDEV=$TEST_ACCESS_TOKEN_SDKDEV
|
|
export MODELSCOPE_ENVIRONMENT='ci'
|
|
export TEST_UPLOAD_MS_TOKEN=$TEST_UPLOAD_MS_TOKEN
|
|
export MODEL_TAG_URL=$MODEL_TAG_URL
|
|
export MODELSCOPE_API_TOKEN=$MODELSCOPE_API_TOKEN
|
|
export PR_CHANGED_FILES=$PR_CHANGED_FILES
|
|
export ASCEND_RT_VISIBLE_DEVICES=$npu
|
|
|
|
if [ "$MODELSCOPE_SDK_DEBUG" == "True" ]; then
|
|
export MODELSCOPE_SDK_DEBUG=True
|
|
echo 'debugging'
|
|
fi
|
|
|
|
# 切换到代码目录并执行命令
|
|
cd $CODE_DIR
|
|
eval $CI_COMMAND
|
|
|
|
if [ $? -ne 0 ]; then
|
|
echo "Running test case failed, please check the log!"
|
|
exit -1
|
|
fi
|
|
break
|
|
done
|
|
|
|
if [ "$is_get_file_lock" = false ] ; then
|
|
echo 'No free NPU!'
|
|
exit 1
|
|
fi
|