test_multi_gpu_data_parallel_forward
vit
deit
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>