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
64870ffb

Commit 64870ffb authored 7 years ago by ziheng Committed by Tianqi Chen 7 years ago

Downloads
- Patches
- Plain Diff

[Contrib] CuDNN v7 Support (#311)

* [Contrib] CuDNN v7 Support

* Add test

parent 0ccc281d

No related branches found

No related tags found

Loading

Changes 10

Hide whitespace changes

Inline Side-by-side

Showing

Makefile 1 addition, 0 deletions

Makefile
make/config.mk 7 additions, 3 deletions

make/config.mk
make/contrib/cudnn.mk 8 additions, 0 deletions

make/contrib/cudnn.mk
python/tvm/contrib/cudnn.py 293 additions, 0 deletions

python/tvm/contrib/cudnn.py
src/contrib/cudnn/conv_forward.cc 159 additions, 0 deletions

src/contrib/cudnn/conv_forward.cc
src/contrib/cudnn/cudnn_utils.cc 126 additions, 0 deletions

src/contrib/cudnn/cudnn_utils.cc
src/contrib/cudnn/cudnn_utils.h 74 additions, 0 deletions

src/contrib/cudnn/cudnn_utils.h
src/op/tensorize.cc 2 additions, 2 deletions

src/op/tensorize.cc
src/pass/vectorize_loop.cc 1 addition, 1 deletion

src/pass/vectorize_loop.cc
tests/python/contrib/test_cudnn.py 61 additions, 0 deletions

tests/python/contrib/test_cudnn.py

with 732 additions and 6 deletions

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