Created by: guru4elephant
add test_dist_collective_base for collective training in pe only, and result will be written into pkl files test=develop