Lunderberg opened a new pull request, #16958: URL: https://github.com/apache/tvm/pull/16958
Prior to this commit, the `dlight` scheduling rules were applied solely based on the global `tvm.target.Target.current()`. However, a TIR PrimFunc may be annotated with the target, rather than using the global `Target.current()`. In this case, the `dlight` scheduling may produce a scheduled PrimFunc that is not compatible with its target. For example, using a thread binding to `"threadIdx.x"` on a CPU target. This commit updates `dlight` to check for a TIR PrimFunc's annotations when scheduling, matching the behavior of `tvm.build`. -- 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]
