The performance of model parallelism (MP) is not good about colossalai HOT 7 CLOSED

Hi @feifeibear . Thank you so much for your effort. We would appreciate it if you could also share the configurations used to test the same models with Deepspeed and PatrickStar? We would like to evaluate and improve the performance on a similar node scale as well as larger scale.
BTW, did you try 3d-8mp? 3d requires a cube number of mp.

from colossalai.

The DeepSpeed benchmark script
https://github.com/feifeibear/DeepSpeedZeRO3Benchmark
The PatrickStar
https://github.com/Tencent/PatrickStar/blob/master/examples/run_transformers.sh
The benchmarking is very easy.

export SUFFIX="colossal_compare"
env GPU_NUM=8 MODEL_TYPE="GPT" MODEL_NAME=GPT3_10B BS=2 CPU_EBD=0 AMM=1 MSC=1 CACHE=1 SP=0 CS=288 HYB=1 TILING=0 ACT_OFFLOAD=0 SUFFIX=${SUFFIX} bash run_transformers.sh

from colossalai.

I have uploaded the logs of DeepSpeed and PatirckStar to Baidu WangPan...
Note that for DeepSpeed, the SamplesPerSec is not equal to 'Throughput'. You have to calculate it by batch/elapse.

link: https://pan.baidu.com/s/1vEHl0hPuxDb7HjOlpuW-YA?pwd=1mfd
code: 1mfd

from colossalai.

@feifeibear Thank you!

from colossalai.

This issue is stale because it has been open for 14 days with no activity.

from colossalai.

Thanks for your report, detailed tests with stable code will come soon.

from colossalai.

We have updated a lot. This issue was closed due to inactivity. Thanks.

from colossalai.