Fix test build (#19444)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19444
ghimport-source-id: c85db00e8037e7f6f0424eb8bd17f957d20b7247
Reviewed By: eellison
Differential Revision: D15008679
Pulled By: driazati
fbshipit-source-id: 0987035116d9d0069794d96395c8ad458ba7c121