Add tests for PDT (#58211)
Summary:
This is a duplicate of the PR https://github.com/pytorch/pytorch/issues/56029
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58211
Reviewed By: gmagogsfm
Differential Revision: D28403903
Pulled By: nikithamalgifb
fbshipit-source-id: 290c46709c77c1a6fd647a2348419d12bf0a5ed6
