projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: fbb2a35) | patch
[Perf] Enhance cudnn and cublas backend and enable TensorCore (#4353)
authorSiyuan Feng <Hzfengsy@vip.qq.com>
Mon, 25 Nov 2019 06:01:55 +0000 (22:01 -0800)
committerLeyuan Wang <laurawly@gmail.com>
Mon, 25 Nov 2019 06:01:55 +0000 (22:01 -0800)
commitdabde40f15423fc9632824dcd95696f36cf17a01
tree6b74eea48e21e24003093b7c40e2008ad4c8c0betree | snapshot
parentfbb2a356df990e8fb8b5d76d4d0124fd3b0378c8commit | diff
[Perf] Enhance cudnn and cublas backend and enable TensorCore (#4353)

* add half and mix precision support to cublas backend

* add TensorCore support in CuDNN

* enhance CuDNN support

* address comments and fix lint

* fix

* add fp16 test
include/tvm/runtime/util.h diff | blob | history
python/tvm/contrib/cublas.py diff | blob | history
python/tvm/contrib/cudnn.py diff | blob | history
src/runtime/contrib/cblas/gemm_common.h diff | blob | history
src/runtime/contrib/cublas/cublas.cc diff | blob | history
src/runtime/contrib/cublas/cublas_utils.h diff | blob | history
src/runtime/contrib/cudnn/conv_forward.cc diff | blob | history
tests/python/contrib/test_cublas.py diff | blob | history
tests/python/contrib/test_cudnn.py diff | blob | history
topi/python/topi/cuda/conv2d.py diff | blob | history
topi/python/topi/cuda/dense.py diff | blob | history
Domain: Machine Learning / ML Framework;