pytorch

d136852b - [CUDA][Linalg] Add a `driver=` kwarg to `torch.linalg.svd` and `svdvals`; add cusolver gesvdaStridedBatched driver to svd (#74521)

[CUDA][Linalg] Add a `driver=` kwarg to `torch.linalg.svd` and `svdvals`; add cusolver gesvdaStridedBatched driver to svd (#74521) [CUDA][Linalg] Add a driver= kwarg to torch.linalg.svd and svdvals; add cusolver gesvdaStridedBatched driver to svd cusolver doc: https://docs.nvidia.com/cuda/cusolver/index.html#cuSolverDN-lt-t-gt-gesvda Todo: - [X] add cusolver `gesvdaStridedBatched` driver - [X] add `driver=` kwarg to `torch.linalg.svd` and `torch.linalg.svdvals` - [X] doc - [X] error out (?) on other non-cusolver use cases: CPU, MAGMA - [X] change svd api in `torch/csrc/api/include/torch/linalg.h` ? Close https://github.com/pytorch/pytorch/issues/41306 Related https://github.com/pytorch/pytorch/issues/75494 Pull Request resolved: https://github.com/pytorch/pytorch/pull/74521 Approved by: https://github.com/Lezcano, https://github.com/IvanYashchuk, https://github.com/mruberry