Follow-up fix for compilation error on CUDA92 (#60287)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60287
Follow up of #60017
Test Plan: Imported from OSS
Reviewed By: gchanan
Differential Revision: D29236208
Pulled By: ejguan
fbshipit-source-id: f1acf9630b45fea8cbdf7d64e47661643d0a52b8