IEEE

Sparse Matrix Selection for CSR-Based SpMV Using Deep Learning

Abstract: CSR (Compressed Sparse Row) is the most popular and widely used sparse matrix representation format for Sparse Matrix-Vector Multiplication (SpMV), which is a key operation in many ...
GitHub

sparse_csr bfloat16 matrix multiplication backward is 10x slower than float16

The minimal reproducible code is described below. Consider a standard autocast training framework, where a weight matrix is a learnable parameter stored in float type; and input is a sparse_csr ...
IEEE

Optimization of Sparse Matrix-Vector Multiplication with Variant CSR on GPUs

Abstract: Sparse Matrix-Vector multiplication (SpMV) is one of the most significant yet challenging issues in computational science area. It is a memory-bound application whose performance mostly ...
GitHub

Sparse_std fails for CSR matrices.

Sparse_std should work both csc and csr matrices. X, y = make_regression(n_samples=1000, n_features=1000) X[X < 2.5] = 0.0 X = sparse.csr_matrix(X) sparse_std(X.shape[0], X.shape[1], X.data, X.indices ...