diff --git a/tests/release_task/cpp_run.bat b/tests/release_task/cpp_run.bat index 84bc21f516..57bf446707 100644 --- a/tests/release_task/cpp_run.bat +++ b/tests/release_task/cpp_run.bat @@ -5,6 +5,7 @@ set NO_PROXY=bj.bcebos.com,xly.bce.baidu.com,aliyun.com set PLATFORM=%1 set DEVICE=%2 set VERSION=%3 +set CUSTOM_CPP_FASTDEPLOY_PACKAGE=%4 set CURRENT_DIR=%cd% echo "CURRENT_DIR" %CURRENT_DIR% @@ -27,7 +28,13 @@ if "%DEVICE%" == "gpu" ( set RUN_CASES=ort paddle openvino ) -echo "CPP_FASTDEPLOY_PACKAGE: " %CPP_FASTDEPLOY_PACKAGE% +@rem Use custom package name if not empty +if "%CUSTOM_CPP_FASTDEPLOY_PACKAGE%" neq "" ( + set CPP_FASTDEPLOY_PACKAGE=%CUSTOM_CPP_FASTDEPLOY_PACKAGE% + echo "using custom package: %CUSTOM_CPP_FASTDEPLOY_PACKAGE%" +) + +echo "current package: %CPP_FASTDEPLOY_PACKAGE%" echo "RUN_CASES" %RUN_CASES% python -c "import wget; wget.download('https://fastdeploy.bj.bcebos.com/resource/images/000000014439.jpg')" diff --git a/tests/release_task/cpp_run.sh b/tests/release_task/cpp_run.sh index ddc1a27442..139bb464f5 100644 --- a/tests/release_task/cpp_run.sh +++ b/tests/release_task/cpp_run.sh @@ -4,6 +4,7 @@ CURRENT_DIR=$(cd $(dirname $0); pwd) PLATFORM=$1 DEVICE=$2 VERSION=$3 +CUSTOM_CPP_FASTDEPLOY_PACKAGE=$4 if [ "$VERSION" = "0.0.0" ];then DOWNLOAD_DIR=dev else @@ -14,10 +15,16 @@ if [ "$DEVICE" = "gpu" ];then else CPP_FASTDEPLOY_PACKAGE=fastdeploy-$PLATFORM-$VERSION fi -echo $CPP_FASTDEPLOY_PACKAGE +# Use custom package name if not empty +if [ ! "$CUSTOM_CPP_FASTDEPLOY_PACKAGE" = "" ];then + CPP_FASTDEPLOY_PACKAGE=$CUSTOM_CPP_FASTDEPLOY_PACKAGE + echo "using custom package: $CUSTOM_CPP_FASTDEPLOY_PACKAGE" +fi +echo "current package: $CPP_FASTDEPLOY_PACKAGE" + LINUX_X64_GPU_CASE=('ort' 'paddle' 'trt') LINUX_X64_CPU_CASE=('ort' 'paddle' 'openvino') -#LINUX_AARCH_CPU_CASE=('ort' 'openvino') +# LINUX_AARCH_CPU_CASE=('ort' 'openvino') LINUX_AARCH_CPU_CASE=('ort') MACOS_INTEL_CPU_CASE=('ort' 'openvino') MACOS_ARM64_CPU_CASE=('default')