You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by ma...@apache.org on 2019/12/03 21:18:26 UTC

[incubator-tvm] branch master updated (119c5c9 -> 77bdd5f)

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.


    from 119c5c9  [MEMORY] Fix gcc 4.8 compact (#4461)
     add 77bdd5f  [RUNTIME] Add cudnn conv3d (#4418)

No new revisions were added by this update.

Summary of changes:
 python/tvm/contrib/cudnn.py                     | 328 +++++++++---------
 src/runtime/contrib/cudnn/conv_forward.cc       | 440 ++++++++++++++++--------
 src/runtime/contrib/cudnn/cudnn_utils.h         |   9 +
 tests/python/contrib/test_cudnn.py              | 128 +++++--
 topi/python/topi/cuda/conv2d.py                 |  21 +-
 topi/python/topi/testing/__init__.py            |   1 +
 topi/python/topi/testing/conv3d_ncdhw_python.py | 106 ++++++
 7 files changed, 674 insertions(+), 359 deletions(-)
 create mode 100644 topi/python/topi/testing/conv3d_ncdhw_python.py