Re: [PR] [Relax,Topi] Allow passing workspace to thrust to avoid allocations [tvm]

2024-04-06 Thread via GitHub
tqchen merged PR #16851: URL: https://github.com/apache/tvm/pull/16851 -- 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. To unsubscribe, e-mail: commits-unsubscr...@tvm.apache.or

[PR] [Relax,Topi] Allow passing workspace to thrust to avoid allocations [tvm]

2024-04-05 Thread via GitHub
vinx13 opened a new pull request, #16851: URL: https://github.com/apache/tvm/pull/16851 This adds an optional parameter `workspace` to thrust APIs so that a workspace can be passed to replace the internal allocations of thrust. `DispatchSortScan` pass is updated to allocate a workspace by o