Laurawly commented on a change in pull request #5116: Add thrust support for nms
URL: https://github.com/apache/incubator-tvm/pull/5116#discussion_r396131529
##
File path: topi/python/topi/cuda/sort.py
##
@@ -237,6 +237,53 @@ def sort_nms_ir(data, valid_count, output, axis,
Laurawly commented on a change in pull request #5116: Add thrust support for nms
URL: https://github.com/apache/incubator-tvm/pull/5116#discussion_r396130839
##
File path: topi/python/topi/cuda/sort.py
##
@@ -237,6 +237,53 @@ def sort_nms_ir(data, valid_count, output, axis,
Laurawly commented on a change in pull request #5116: Add thrust support for nms
URL: https://github.com/apache/incubator-tvm/pull/5116#discussion_r396131139
##
File path: topi/python/topi/cuda/sort.py
##
@@ -237,6 +237,53 @@ def sort_nms_ir(data, valid_count, output, axis,
Laurawly commented on a change in pull request #5116: Add thrust support for nms
URL: https://github.com/apache/incubator-tvm/pull/5116#discussion_r396131139
##
File path: topi/python/topi/cuda/sort.py
##
@@ -237,6 +237,53 @@ def sort_nms_ir(data, valid_count, output, axis,
Laurawly commented on a change in pull request #5116: Add thrust support for nms
URL: https://github.com/apache/incubator-tvm/pull/5116#discussion_r396130839
##
File path: topi/python/topi/cuda/sort.py
##
@@ -237,6 +237,53 @@ def sort_nms_ir(data, valid_count, output, axis,
Laurawly commented on a change in pull request #5116: Add thrust support for nms
URL: https://github.com/apache/incubator-tvm/pull/5116#discussion_r396052230
##
File path: src/runtime/contrib/thrust/thrust.cu
##
@@ -65,6 +65,105 @@ void thrust_sort(DLTensor* input,
}
}
Laurawly commented on a change in pull request #5116: Add thrust support for nms
URL: https://github.com/apache/incubator-tvm/pull/5116#discussion_r396052223
##
File path: src/runtime/contrib/thrust/thrust.cu
##
@@ -65,6 +65,105 @@ void thrust_sort(DLTensor* input,
}
}