You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by lm...@apache.org on 2020/11/12 13:03:52 UTC
[incubator-tvm] branch main updated (a7b22ab -> 1fbe174)
This is an automated email from the ASF dual-hosted git repository.
lmzheng pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from a7b22ab Dynamic gpu tests, add dynamic strided slice to topi (#6870)
add 1fbe174 [AutoScheduler] Add winograd support in tuning networks (#6877)
No new revisions were added by this update.
Summary of changes:
python/tvm/relay/op/strategy/cuda.py | 163 ++++++++++------
python/tvm/topi/cuda/conv2d_alter_op.py | 64 +++++--
python/tvm/topi/cuda/conv2d_winograd.py | 14 ++
python/tvm/topi/nn/conv2d.py | 211 +++++++++++++++++++++
src/auto_scheduler/search_policy/sketch_policy.cc | 2 +-
.../search_policy/sketch_policy_rules.cc | 37 ++--
.../relay/test_auto_scheduler_task_extraction.py | 23 +++
tests/python/relay/test_auto_scheduler_tuning.py | 24 ++-
.../topi/python/test_topi_conv2d_nhwc_winograd.py | 2 +-
.../topi/python/test_topi_conv2d_winograd.py | 78 +++++++-
.../test_auto_scheduler_sketch_generation.py | 11 +-
11 files changed, 520 insertions(+), 109 deletions(-)