[TP][2D][EZ] Fix Error in FSDP 2D test (#107975)
As title, TP dimension should be the second dim, so we need to pass tp_degree to the second rather the first dim of the mesh tensor.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/107975
Approved by: https://github.com/wz337, https://github.com/awgu
