heat
heat copied to clipboard
Chunk function for CSR sparse arrays
Figured out how to obtain the data specific to each process. Should be efficient enough.
Question:
What should happen to the Row indices vector?
The row indices keep track of the number of non-zero elements before that row. Since we are splitting it into chunks, the original row index vector (or a part of it) no longer makes sense in the local array. Building a new row index vector also seems pointless because I don't see how that would be useful in matrix operations.
GPU cluster tests are currently disabled on this Pull Request.
👇 Click on the image for a new way to code review
-
Make big changes easier — review code in small groups of related files
-
Know where to start — see the whole change at a glance
-
Take a code tour — explore the change with an interactive tour
-
Make comments and review — all fully sync’ed with github