mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
90c821814e
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> |
||
|---|---|---|
| .. | ||
| ComputeLibrary.cmake | ||
| cuda.cmake | ||
| gflags.cmake | ||
| glog.cmake | ||
| LoadHIP.cmake | ||
| mkl.cmake | ||
| mkldnn.cmake | ||
| protobuf.cmake | ||
| utils.cmake | ||
| xpu.cmake | ||