[benchmark] migrate to fsdp/ddp after jit, from fsdp/ddp before jit (…

[benchmark] migrate to fsdp/ddp after jit, from fsdp/ddp before jit (… #3651