OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
b2eb0e8c6a
pytorch/caffe2
History
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) 2024-07-24 10:07:36 +00:00
.clang-format
CMakeLists.txt SparseCsrCUDA: cuDSS backend for linalg.solve (#129856) 2024-08-22 07:57:30 +00:00
unexported_symbols.lds Hide all symbols in llvm namespace (#63272) 2021-08-15 11:29:43 -07:00
version_script.lds Hide all symbols in llvm namespace (#63272) 2021-08-15 11:29:43 -07:00