fix fx tests (#88886)
Summary:
Some source files are missing and TPX couldn't handle the default test
names.
Test Plan: Rely on CI.
Differential Revision: D41218564
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88886
Approved by: https://github.com/zou3519
