add some missing includes (#100049)
add some missing includes
Summary:
These were failing in my build environment. Clang16, Fedora38, no
extra build config.
Test Plan: Rely on CI.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/100049
Approved by: https://github.com/Skylion007