Commits on Apr 30, 2024

1. [DLight] Check for target in function attributes …

   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`.

   

   Lunderberg committed Apr 30, 2024
   Configuration menu

   • View commit details
   • Copy full SHA for 3a7b15a
   • Browse repository at this point

   Copy the full SHA

   3a7b15a View commit details

   Browse the repository at this point in the history