[PyTorch] Fix `lkj_cholesky` device error (#73980)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/73980
Fix missing device arg to `torch.arange` in `lkj_cholesky`.
Test Plan: TBD
Reviewed By: Balandat
Differential Revision: D34763123
fbshipit-source-id: 67914e1dd0335b20cb60048e13b6df70a9fdd9f5
(cherry picked from commit a8b979a0815876100d7e44fdb65cf40872b39882)