Revert "Revert D32498570: make codegen'd device guards not cuda-specific. Allow them to be used in external codegen" (#69951)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/69951
This reverts commit 0ef523633fddf2d63e97d5028b00af10ff344561.
Test Plan: Imported from OSS
Reviewed By: albanD
Differential Revision: D33113543
Pulled By: bdhirsh
fbshipit-source-id: b28073ee0870b413ea9f617f27671ae5c6f3c696