mirror of https://github.com/hpcaitech/ColossalAI
aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
861 B
22 lines
861 B
for MODEL_TYPE in "gpt2_medium"; do |
|
for DISTPLAN in "CAI_Gemini"; do |
|
for BATCH_SIZE in 16; do |
|
for GPUNUM in 1 2 4 8; do |
|
for TPDEGREE in 1 2 4 8; do |
|
if [ ${TPDEGREE} -gt ${GPUNUM} ]; then |
|
continue |
|
fi |
|
for PLACEMENT in "cpu" "auto"; do |
|
echo "****************** Begin ***************************" |
|
echo "+ benchmrking MODEL ${MODEL_TYPE} DISTPLAN ${DISTPLAN} GPU ${GPUNUM} BS ${BATCH_SIZE} TP ${TPDEGREE} POLICY ${PLACEMENT}" |
|
MODEL_TYPE=${MODEL_TYPE} DISTPLAN=${DISTPLAN} BATCH_SIZE=${BATCH_SIZE} GPUNUM=${GPUNUM} TPDEGREE=${TPDEGREE} PLACEMENT=${PLACEMENT} \ |
|
bash ./run_gemini.sh |
|
echo "****************** Finished ***************************" |
|
echo "" |
|
echo "" |
|
done |
|
done |
|
done |
|
done |
|
done |
|
done
|
|
|