[FSDP] import ``_symbolic_trace`` only when ``torch.fx`` is enabled (#81339)
This is used to fix internal tests where ``torch.fx`` is not available in the module.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81339
Approved by: https://github.com/rohan-varma, https://github.com/zhaojuanmao
