-
- Downloads
[CONTRIB] cuBLAS integration (#744)
* add cublas support * integrate cublas to topi dense * add cublas error check * minor fix * fix lint * remove topi import from contrib unittest
Showing
- Makefile 1 addition, 0 deletionsMakefile
- make/config.mk 3 additions, 0 deletionsmake/config.mk
- make/contrib/cublas.mk 8 additions, 0 deletionsmake/contrib/cublas.mk
- python/tvm/contrib/cblas.py 1 addition, 1 deletionpython/tvm/contrib/cblas.py
- python/tvm/contrib/cublas.py 32 additions, 0 deletionspython/tvm/contrib/cublas.py
- python/tvm/target.py 2 additions, 1 deletionpython/tvm/target.py
- src/contrib/cblas/cblas.cc 1 addition, 1 deletionsrc/contrib/cblas/cblas.cc
- src/contrib/cublas/cublas.cc 81 additions, 0 deletionssrc/contrib/cublas/cublas.cc
- tests/python/contrib/test_cublas.py 33 additions, 0 deletionstests/python/contrib/test_cublas.py
- tests/python/contrib/test_cudnn.py 1 addition, 2 deletionstests/python/contrib/test_cudnn.py
- topi/python/topi/cuda/__init__.py 1 addition, 1 deletiontopi/python/topi/cuda/__init__.py
- topi/python/topi/cuda/dense.py 43 additions, 0 deletionstopi/python/topi/cuda/dense.py
- topi/python/topi/nn/dense.py 25 additions, 3 deletionstopi/python/topi/nn/dense.py
Loading
Please register or sign in to comment