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 2021/02/17 04:31:40 UTC

[tvm] branch main updated (564aae0 -> e57e644)

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 564aae0  [Relay][Topi] Add max mode to ROI align (#7440)
     add e57e644  [ROCM] Add Thrust support (#7458)

No new revisions were added by this update.

Summary of changes:
 cmake/modules/ROCM.cmake             |  17 +++++
 python/tvm/relay/op/strategy/rocm.py |  92 +++++++++++++++++++++++++
 python/tvm/topi/cuda/nms.py          |   3 +-
 python/tvm/topi/cuda/scan.py         |   5 +-
 src/runtime/contrib/thrust/thrust.cu |   1 +
 tests/python/contrib/test_thrust.py  | 127 ++++++++++++++++++++---------------
 6 files changed, 187 insertions(+), 58 deletions(-)