From 12f426c63f7eee04a8bfdf2b282008a76a04462f Mon Sep 17 00:00:00 2001 From: Yi Zhang Date: Wed, 21 Aug 2024 09:30:05 +0800 Subject: [PATCH] update size limit check of training GPU wheel (#21762) ### Description ### Motivation and Context The training wheel size limit should be 400M --- .../templates/py-packaging-training-cuda-stage-steps.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/ci_build/github/azure-pipelines/templates/py-packaging-training-cuda-stage-steps.yml b/tools/ci_build/github/azure-pipelines/templates/py-packaging-training-cuda-stage-steps.yml index 9b65ddbfdf..fc163d17e4 100644 --- a/tools/ci_build/github/azure-pipelines/templates/py-packaging-training-cuda-stage-steps.yml +++ b/tools/ci_build/github/azure-pipelines/templates/py-packaging-training-cuda-stage-steps.yml @@ -165,7 +165,7 @@ stages: set -e -x whlfilename=$(ls $(Build.ArtifactStagingDirectory)/Release/dist/*.whl | head -n 1) ; \ echo $whlfilename ; du -sh $whlfilename ; \ - (( $(wc -c < "$whlfilename") - 300*1024*1024 < 0 )) || ( echo 'Wheel size bigger than 300M'; exit 1) + (( $(wc -c < "$whlfilename") - 400*1024*1024 < 0 )) || ( echo 'Wheel size bigger than 400M'; exit 1) displayName: 'Check wheel size' continueOnError: true