You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by la...@apache.org on 2020/03/20 17:00:33 UTC
[incubator-tvm] branch master updated (dbd805c -> 2e8f3a9)
This is an automated email from the ASF dual-hosted git repository.
laurawly pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from dbd805c [AutoTVM] Temporary fix to the stack overflow issue in autotvm task extraction (#5019)
add 2e8f3a9 [TOPI][OP] Use Thrust sort for argsort and topk (#5097)
No new revisions were added by this update.
Summary of changes:
CMakeLists.txt | 4 ++
cmake/config.cmake | 3 +
cmake/modules/CUDA.cmake | 8 +++
python/tvm/relay/op/strategy/cuda.py | 11 +++
src/runtime/contrib/thrust/thrust.cu | 133 +++++++++++++++++++++++++++++++++++
tests/lint/check_file_type.py | 1 +
topi/python/topi/cuda/sort.py | 112 ++++++++++++++++++++++++++++-
7 files changed, 271 insertions(+), 1 deletion(-)
create mode 100644 src/runtime/contrib/thrust/thrust.cu