diff --git a/examples/generate_config.sh b/examples/generate_config.sh index 6513c2c..4476be7 100755 --- a/examples/generate_config.sh +++ b/examples/generate_config.sh @@ -117,7 +117,8 @@ zero="\ elif [ $ZERO_STAGE == 1 ]; then zero="\ \"zero_optimization\": { - \"stage\": $ZERO_STAGE + \"stage\": $ZERO_STAGE, + \"overlap_comm\": true }," if [ $ZERO_STAGE == 1 ]; then if [ $PP > 1 ]; then