[FSDP] records for composable (#98428)
Add some function recording since composable API does record a FSDP.forward
Differential Revision: [D44715137](https://our.internmc.facebook.com/intern/diff/D44715137/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/98428
Approved by: https://github.com/awgu