pytorch

35ce68bb - [PyTorch] Fix `lkj_cholesky` device error (#73980)

[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)