chunit-quic commented on PR #14168: URL: https://github.com/apache/tvm/pull/14168#issuecomment-1451289044
Hi community, This is second part of TVM Explorer infrastructures. After this patch, user can backtrack what (schedule) primitives have been applied to a schedule. Pair with the tool, TEDD, these changes can be visualized beautifully. To give you more context, we briefly describe what have been done below in this PR. Here is the list of main modifications: 1. Make a Stage know which Schedule it belongs to. 2. Add two attributes, primitive_record and schedule_record to Schedule. 3. Create a new class, ScheduleContext, to record primitive in the RAII way. 4. Register a pass config variable, keep_schedule_record to enable/disable the recording. Everytime a Stage being modified by a primitive, we will use ScheduleContext to record the optimized schedule and corresponding primitive. So that we can obtain the modification sequence later. You can reference to [preRFC](https://discuss.tvm.apache.org/t/pre-rfc-tvm-explorer-infrastructure/13457#schedulestage-visualization-enhancement-7) for more details. Thanks! # For your reference, here is the related links. [PreRFC in forum](https://discuss.tvm.apache.org/t/pre-rfc-tvm-explorer-infrastructure/13457/24) https://github.com/apache/tvm-rfcs/pull/92 https://github.com/apache/tvm/issues/13116 [@haowhsu-quic](https://github.com/haowhsu-quic), [@zack-ch](https://github.com/zack-ch) -- 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]
