masahi commented on pull request #7303: URL: https://github.com/apache/tvm/pull/7303#issuecomment-763264832
> Once it is merged, I can try on my end with TF models as well. Perf improvement is not expected, since it only improves `get_valid_count` slightly if you use thrust scan instead of TIR scan. The purpose of this PR is to enable parallelization for other ops, that are difficult without it. `argwhere` is a perfect example that I'll demonstrate soon after this one. @anijain2305 The term you want to search for is "gpu stream compaction". ---------------------------------------------------------------- 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: us...@infra.apache.org
