Compile THCTensorTopK per dtype. (#36074)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36074
ROCm builds fail inconsistently on this file by timing out.
Test Plan: Imported from OSS
Differential Revision: D20872395
Pulled By: gchanan
fbshipit-source-id: 20d0890433b7290c36ed99bc7bfb73a93971ead1