File tree 1 file changed +10
-4
lines changed 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -221,15 +221,17 @@ function llama_dygraph_auto_bs4_bf16_SD2() {
221
221
222
222
export CUDA_DEVICE_MAX_CONNECTIONS=1
223
223
224
- flags=(" " " FLAGS_fuse_allreduce_in_opt" " FLAGS_fuse_reducescatter_in_opt" )
224
+ flags=(" " " FLAGS_fuse_allreduce_in_opt" " FLAGS_fuse_reducescatter_in_opt" " FLAGS_enable_tensor_fusion FLAGS_enable_sharding_overlap " )
225
225
for i in " ${! flags[@]} " ; do
226
226
flag=" ${flags[$i]} "
227
227
228
228
if [ -n " $flag " ]; then
229
- export " $flag =true"
229
+ for f in $flag ; do
230
+ export " $f =true"
231
+ done
230
232
fi
231
233
232
- task_name=" llama_dygraph_auto_bs4_bf16_SD2_$flag "
234
+ task_name=" llama_dygraph_auto_bs4_bf16_SD2_$f "
233
235
case_out_dir=" output/$task_name "
234
236
case_log_dir=" output/$task_name " " _log"
235
237
rm -rf $case_out_dir
@@ -299,6 +301,8 @@ function llama_dygraph_auto_bs4_bf16_SD2() {
299
301
loss_base=9.23502579
300
302
elif [ " $flag " = " FLAGS_fuse_reducescatter_in_opt" ]; then
301
303
loss_base=9.23504105
304
+ elif [ " $flag " = " FLAGS_enable_tensor_fusion FLAGS_enable_sharding_overlap" ]; then
305
+ loss_base=9.23504868
302
306
else
303
307
loss_base=-1
304
308
fi
@@ -308,7 +312,9 @@ function llama_dygraph_auto_bs4_bf16_SD2() {
308
312
check_result $FUNCNAME ${loss_base} ${loss} ${ips_base} ${ips} ${mem_base} ${mem}
309
313
310
314
if [ -n " $flag " ]; then
311
- export " $flag =false"
315
+ for f in $flag ; do
316
+ export " $f =false"
317
+ done
312
318
fi
313
319
done
314
320
echo " =========== $FUNCNAME run end ==========="
You can’t perform that action at this time.
0 commit comments