kevinthesun commented on a change in pull request #7117:
URL: https://github.com/apache/tvm/pull/7117#discussion_r544558622
##
File path: python/tvm/topi/cuda/conv2d_transpose_nchw.py
##
@@ -194,6 +197,8 @@ def _callback(op):
if cfg.is_fallback:
kevinthesun commented on a change in pull request #7117:
URL: https://github.com/apache/tvm/pull/7117#discussion_r544558368
##
File path: python/tvm/topi/cuda/nms.py
##
@@ -754,7 +782,22 @@ def non_max_suppression(
)
score_axis = score_index
score_shape = (batch_
kevinthesun commented on a change in pull request #7117:
URL: https://github.com/apache/tvm/pull/7117#discussion_r544551395
##
File path: tests/python/relay/test_any.py
##
@@ -1430,6 +1439,21 @@ def test_non_max_suppression():
disable_targets=["nvptx"],
)
+n