Fix tensor device in test_kthvalue_overlap (#56869)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56869
ghstack-source-id: 127914015
Test Plan: auto test
Reviewed By: ezyang
Differential Revision: D27986559
fbshipit-source-id: f4a638d737b06dd5f384b54e20490d76543d4e78
