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/09 01:04:06 UTC

[tvm] branch main updated (0095b21 -> 465cd14)

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 0095b21  [BYOC][TRT] Support batch norm for all ranks <=5, and all axes (#7026)
     add 465cd14  [TOPI] GPU scatter 1D via sorting based approach (#7056)

No new revisions were added by this update.

Summary of changes:
 cmake/modules/CUDA.cmake             |   1 +
 python/tvm/topi/cuda/scatter.py      | 106 ++++++++++++++++++++++++++++++++++-
 python/tvm/topi/cuda/sort.py         |  59 ++++++++++++++++++-
 src/runtime/contrib/thrust/thrust.cu |  73 ++++++++++++++++++++++++
 tests/python/contrib/test_sort.py    |  34 +++++++++++
 5 files changed, 271 insertions(+), 2 deletions(-)