Pytorch add agent api tests (#56985)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56985
Pytorch add agent api tests
Test Plan: ci/cd
Reviewed By: cbalioglu
Differential Revision: D28020485
fbshipit-source-id: e6acf095f26ce4b99cddfbf7641fb4fa885b0c86
