alec-anyvision opened a new issue #6629: URL: https://github.com/apache/incubator-tvm/issues/6629
I believe the following two lines should have `valid_count=valid_count`, as in [the default implementation](https://github.com/apache/incubator-tvm/blob/21002cd094c34716b1e07a63ed76f53dadd60e23/python/tvm/topi/vision/nms.py#L558). I believe the `data` array is zeroed at initialization for cuda, so the issue only manifests if nms is run more than once, with the first run containing more valid boxes than the second run. https://github.com/apache/incubator-tvm/blob/21002cd094c34716b1e07a63ed76f53dadd60e23/python/tvm/topi/cuda/nms.py#L488 https://github.com/apache/incubator-tvm/blob/21002cd094c34716b1e07a63ed76f53dadd60e23/python/tvm/topi/cuda/nms.py#L492 ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
