Fix cpp extension compile failure on some envs (#37221)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/37221
Test Plan: Imported from OSS
Differential Revision: D21226873
Pulled By: glaringlee
fbshipit-source-id: 0a390bbeaf153ee5ec355943f92c2dbcc5e04b59