[sparse] access the sparse matrix element like A[i,j]

taichi-dev / taichi

Productive, portable, and performant GPU programming in Python.

https://taichi-lang.org

Apache License 2.0

25.48k stars 2.28k forks source link

[sparse] access the sparse matrix element like A[i,j] #8317

Open chunleili opened 1 year ago

chunleili commented 1 year ago

Concisely describe the proposed feature I want to access sparse matrix like A[i,j] in a kernel function. The advantage of this usage is that I can define my own matrix solver like jacobian iteration method.

Possible solution Expose the Eigen API which could transfer the matrix to COO format could be easy to implement.

jim19930609 commented 1 year ago

Hi @houkensjtu, can you help take a look? Thanks!