Let bench_one_batch support enable_dp_attention
#4058
Merged
bench_one_batch support enable_dp_attention
#4058