chrischoy / torch_spmm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CUDA cusparseSpmm example

Installation

export CXX=g++-7
python setup.py install

Unittest

python -m unittest tests.test.TestSparseMM.spmm_int32
python -m unittest tests.test.TestSparseMM.spmm_int64

About


Languages

Language:Cuda 79.3%Language:Python 20.7%