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 2020/12/17 22:03:50 UTC

[tvm] branch main updated (bad149e -> 9e766d9)

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

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


    from bad149e  [TOPI] Fix GPU Dynamic Op Schedule (#7117)
     add 9e766d9  [Relay][Topi]Add Sort Op to Relay (#6978)

No new revisions were added by this update.

Summary of changes:
 python/tvm/relay/op/_algorithm.py              |   9 ++-
 python/tvm/relay/op/algorithm.py               |  22 ++++++
 python/tvm/relay/op/strategy/cuda.py           |  20 +++++
 python/tvm/relay/op/strategy/generic.py        |  24 ++++++
 python/tvm/topi/cuda/sort.py                   | 103 +++++++++++++++++++++++++
 python/tvm/topi/generic/sort.py                |  17 ++++
 python/tvm/topi/sort.py                        |  42 ++++++++++
 src/relay/op/algorithm/{argsort.cc => sort.cc} |  26 +++----
 src/runtime/contrib/sort/sort.cc               |  63 ++++++++++++++-
 tests/python/relay/test_op_level6.py           |  60 +++++++++++---
 tests/python/topi/python/test_topi_sort.py     |  53 +++++++++++++
 11 files changed, 411 insertions(+), 28 deletions(-)
 copy src/relay/op/algorithm/{argsort.cc => sort.cc} (69%)