CSRSPMM is a high-performance library for multiplying Compressed Sparse Row (CSR) matrices with dense matrices on NVIDIA GPUs. It includes a generic CUDA backend, and a PyTorch extension for easy ...
Currently, I need to perform Gaussian filtering on images, but I want to avoid using MATLAB's built-in functions. Given an image of size N x N, my idea is to construct a sparse matrix where both ...