Commit 4daf1a0
Fix GPU type preservation in mixed sparse-dense matrix multiplication
Three functions were always returning CPU matrices even with GPU inputs:
- SparseMatrixMPI * MatrixMPI
- MatrixMPI * SparseMatrixMPI
- transpose(MatrixMPI) * SparseMatrixMPI
Now correctly preserve the dense matrix's backend type (CPU or GPU).1 parent dbe94fd commit 4daf1a0
1 file changed
Lines changed: 171 additions & 69 deletions
0 commit comments