diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh index b9b2e87aeba9284269ecdcc0b21dceda8cfbac78..d1220e453758258b65fe2cc5e5ea0ff54318db08 100755 --- a/paddle/scripts/paddle_build.sh +++ b/paddle/scripts/paddle_build.sh @@ -2778,7 +2778,8 @@ function exec_samplecode_test() { if [ "$1" = "cpu" ] ; then python sampcd_processor.py cpu; example_error=$? elif [ "$1" = "gpu" ] ; then - python sampcd_processor.py --threads=16 gpu; example_error=$? + SAMPLE_CODE_EXEC_THREADS=${SAMPLE_CODE_EXEC_THREADS:-2} + python sampcd_processor.py --threads=${SAMPLE_CODE_EXEC_THREADS} gpu; example_error=$? fi if [ "$example_error" != "0" ];then echo "Code instance execution failed" >&2