Skip to content

Conversation

@mercush
Copy link

@mercush mercush commented Nov 19, 2025

Proposed changes

Feature: added function metal kernel sparse_matmul_csr for sparse matrix-dense matrix multiplication. The function takes as input a sparse matrix (represented as three arrays representing the row pointers, the column indices, and the values at the nonzero entries) and a dense matrix represented as a dense array with two dimensions. I have implemented the backends for metal and for the cpu but not cuda.

Please let me know if there is any documentation I should update.

Checklist

Put an x in the boxes that apply.

  • I have read the CONTRIBUTING document
  • I have run pre-commit run --all-files to format my code / installed pre-commit prior to committing changes
  • I have added tests that prove my fix is effective or that my feature works
  • I have updated the necessary documentation (if needed)

@mercush mercush marked this pull request as draft November 19, 2025 16:56
@mercush mercush changed the title Kernel for sparse matrix-dense matrix multiplication [Experiment] Kernel for sparse matrix-dense matrix multiplication Nov 19, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant