[Sparse] add sparse tensor computation support (#1289) (d87c94d4) · Commits · cld / ml / tvm · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

T

tvm
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Releases
- Model registry
- Environments
- Incidents
- Service Desk

Snippets Groups Projects

- cld
- ml
- tvm
- Commits
d87c94d4

Commit d87c94d4 authored 6 years ago by Liangfu Chen Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

[Sparse] add sparse tensor computation support (#1289)

parent 75654835

No related branches found

No related tags found

No related merge requests found

Changes 9

Hide whitespace changes

Inline Side-by-side

Showing

python/tvm/autotvm/task/dispatcher.py 1 addition, 1 deletion

python/tvm/autotvm/task/dispatcher.py
python/tvm/contrib/sparse.py 163 additions, 0 deletions

python/tvm/contrib/sparse.py
tests/python/contrib/test_sparse.py 100 additions, 0 deletions

tests/python/contrib/test_sparse.py
topi/python/topi/__init__.py 1 addition, 0 deletions

topi/python/topi/__init__.py
topi/python/topi/sparse/__init__.py 7 additions, 0 deletions

topi/python/topi/sparse/__init__.py
topi/python/topi/sparse/csrmm.py 94 additions, 0 deletions

topi/python/topi/sparse/csrmm.py
topi/python/topi/sparse/csrmv.py 90 additions, 0 deletions

topi/python/topi/sparse/csrmv.py
topi/python/topi/sparse/dense.py 173 additions, 0 deletions

topi/python/topi/sparse/dense.py
topi/tests/python/test_topi_sparse.py 205 additions, 0 deletions

topi/tests/python/test_topi_sparse.py

with 834 additions and 1 deletion

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment