From commits-return-9709-archive-asf-public=cust-asf.ponee.io@tvm.apache.org Fri Mar 27 23:20:52 2020 Return-Path: X-Original-To: archive-asf-public@cust-asf.ponee.io Delivered-To: archive-asf-public@cust-asf.ponee.io Received: from mail.apache.org (hermes.apache.org [207.244.88.153]) by mx-eu-01.ponee.io (Postfix) with SMTP id 42055180637 for ; Sat, 28 Mar 2020 00:20:52 +0100 (CET) Received: (qmail 30672 invoked by uid 500); 27 Mar 2020 23:20:51 -0000 Mailing-List: contact commits-help@tvm.apache.org; run by ezmlm Precedence: bulk List-Help: List-Unsubscribe: List-Post: List-Id: Reply-To: dev@tvm.apache.org Delivered-To: mailing list commits@tvm.apache.org Received: (qmail 30661 invoked by uid 99); 27 Mar 2020 23:20:51 -0000 Received: from ec2-52-202-80-70.compute-1.amazonaws.com (HELO gitbox.apache.org) (52.202.80.70) by apache.org (qpsmtpd/0.29) with ESMTP; Fri, 27 Mar 2020 23:20:51 +0000 Received: by gitbox.apache.org (ASF Mail Server at gitbox.apache.org, from userid 33) id 34526819F7; Fri, 27 Mar 2020 23:20:51 +0000 (UTC) Date: Fri, 27 Mar 2020 23:20:50 +0000 To: "commits@tvm.apache.org" Subject: [incubator-tvm] branch master updated (949dca4 -> f4286cc) MIME-Version: 1.0 Content-Type: text/plain; charset=utf-8 Content-Transfer-Encoding: 8bit Message-ID: <158535125063.6891.3311509667401487631@gitbox.apache.org> From: tqchen@apache.org X-Git-Host: gitbox.apache.org X-Git-Repo: incubator-tvm X-Git-Refname: refs/heads/master X-Git-Reftype: branch X-Git-Oldrev: 949dca4d078dbeb7990294b1f7a83de5b6ed4a25 X-Git-Newrev: f4286cc71776e1e731be711fcbe7b161d595e39a X-Git-NotificationType: ref_changed X-Git-Multimail-Version: 1.5.dev Auto-Submitted: auto-generated 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%)