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

Posted to commits@tvm.apache.org by tq...@apache.org on 2020/03/27 23:20:50 UTC

[incubator-tvm] branch master updated (949dca4 -> f4286cc)

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

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


    from 949dca4  [External Codegen] Fix annotate pass static variable (#5023)
     add f4286cc  [TOPI][Tensor Core] Conv2d and Dense ops support on Tensor Core (#5099)

No new revisions were added by this update.

Summary of changes:
 python/tvm/relay/op/strategy/cuda.py               |  39 ++-
 python/tvm/relay/testing/resnet.py                 |  59 +++-
 src/driver/driver_api.cc                           |   7 +
 topi/python/topi/cuda/__init__.py                  |   2 +
 topi/python/topi/cuda/conv2d.py                    |  35 ++-
 topi/python/topi/cuda/conv2d_nhwc.py               | 131 +++++++++
 topi/python/topi/cuda/conv2d_nhwc_tensorcore.py    | 318 +++++++++++++++++++++
 topi/python/topi/cuda/dense_tensorcore.py          | 252 ++++++++++++++++
 topi/python/topi/cuda/tensor_intrin.py             | 147 +++++++++-
 topi/tests/python/test_topi_conv2d_nhwc.py         |  12 +-
 .../python/test_topi_conv2d_nhwc_tensorcore.py     | 126 ++++++++
 .../tests/python/test_topi_dense_tensorcore.py     |  63 ++--
 12 files changed, 1120 insertions(+), 71 deletions(-)
 create mode 100644 topi/python/topi/cuda/conv2d_nhwc.py
 create mode 100644 topi/python/topi/cuda/conv2d_nhwc_tensorcore.py
 create mode 100644 topi/python/topi/cuda/dense_tensorcore.py
 create mode 100644 topi/tests/python/test_topi_conv2d_nhwc_tensorcore.py
 copy tests/webgl/test_local_topi_dense.py => topi/tests/python/test_topi_dense_tensorcore.py (56%)