jwfromm opened a new pull request, #12505: URL: https://github.com/apache/tvm/pull/12505
Currently in tir, using `cache_read` will rewrite all consumer blocks of the specified buffer in the current scope. However, there may be cases where we want to apply that transformation only to a subset of consumer blocks. In the te version of cache_read, this is explicitly specified through a list of `consumer_blocks`. This PR extends the tir version of `cache_read` with an optional `consumer_blocks` argument. When provided, only blocks listed in `consumer_blocks` will be rewritten. If not provided, the current behavior of `cache_read` remains the same. -- 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: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
