Add ROCm-specific half_support_literal for JIT. (#38899)
Summary:
CC ezyang xw285cornell sunway513 lcskrishna
Pull Request resolved: https://github.com/pytorch/pytorch/pull/38899
Differential Revision: D21721855
Pulled By: ezyang
fbshipit-source-id: 3739c462f04cee40ff979f44387ef66b971f5303