mbs-octoml opened a new pull request #9326: URL: https://github.com/apache/tvm/pull/9326
NOT READY FOR REVIEW Continuing on from #9313, as part of https://github.com/apache/tvm-rfcs/pull/38, we switch plan_devices.cc to yield SEScopes instead of DLDeviceTypes. - "on_device" and "device_copy" attributes use SEScopes. Python bindings still use Devices so the only observable change will be that device_ids in annotations now flow all the way through to codegen. We can extend these bindings to allow an exact target to be given in the future should we outgrow the existing 'device type implies unique target' convention. - Move responsibility for resolving device types to targets into device planner and out of LowerTEPass. All that does is make sure every SEScope entering the system via an annotation ends up with a target if it doesn't already have one. - Update VM, AOT, Interpreter and Graph executors to use SEScopes on memory operators instead of just device type or Device. - Centralize all device defaulting logic to use the CompilerConfig helper class. -- 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]
