PyTorch 2.1: Quansight’s Improvements to BSR Sparse Matrix Multiplication
This post presents Quansight work implementing new block sparse row (BSR) kernels for sparse matrix multiplication in PyTorch 2.1.
This post presents Quansight work implementing new block sparse row (BSR) kernels for sparse matrix multiplication in PyTorch 2.1.