[TIR][Schedule] Support for specific consumer block targeting in cache_write

[wrongtest-intellif]

Hi, this change try add optional consumer_blocks param to cache_write symmetrically, following change on cache_read of #12505.

This is helpful when there are multiple consumers to the targeting block who also write to global output, and when we do cache_write, we may want to control some of the consumers read data reside on cache.

As a comparation, for s.cache_read(B, 0, "cache_scope", consumer_blocks=[B]):

For s.cache_write(P, 0, "cache_scope", consumer_blocks=[B])

cc @Hzfengsy @jwfromm

[tvm-bot]

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

• cc @Hzfengsy, @junrushao _{See #10317 for details}
• Built docs for commit 97c4721 can be found here.

_{Generated by tvm-bot}