pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Zitong Zhan 90c821814e SparseCsrCUDA: cuDSS backend for linalg.solve (#129856 )

This PR switches to cuDSS library and has the same purpose of #127692, which is to add Sparse CSR tensor support to linalg.solve.
Fixes #69538

Minimum example of usage:
```
import torch

if __name__ == '__main__':
    spd = torch.rand(4, 3)
    A = spd.T @ spd
    b = torch.rand(3).to(torch.float64).cuda()
    A = A.to_sparse_csr().to(torch.float64).cuda()

    x = torch.linalg.solve(A, b)
    print((A @ x - b).norm())

```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/129856
Approved by: https://github.com/amjames, https://github.com/lezcano, https://github.com/huydhn

Co-authored-by: Zihang Fang <zhfang1108@gmail.com>
Co-authored-by: Huy Do <huydhn@gmail.com>

2024-08-22 07:57:30 +00:00

core

Revert "Remove unused Caffe2 macros (#132979 )"

2024-08-12 18:34:56 +00:00

perfkernels

Don't install remaining caffe2 python files (#129067 )

2024-06-27 17:25:59 +00:00

serialize

Revert "Make c10::string_view an alias of std::string_view (#130417 )"

2024-07-12 00:37:04 +00:00

utils

[BE] Remove suppression of inconsistent missing overrides (#131524 )