yincs-intellif commented on PR #12450: URL: https://github.com/apache/tvm/pull/12450#issuecomment-1216398026
Hi @junrushao1994 , what do you think of my follow suggestion? 1. we give a integer parameter instead of to_early_stage, which means block number, masked as N. 2. Default we will get a compute at position, which is after the last producer and before the first consumer, masked as [P, C]. 3. when N in [P,C], we use N for the position of compute at, otherwise use C. The above rules apply to "reverse_compute_at" at same time Thanks in advance! -- 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]
