jcf94 commented on a change in pull request #6073:
URL: https://github.com/apache/incubator-tvm/pull/6073#discussion_r456769902
##########
File path: src/auto_scheduler/compute_dag.cc
##########
@@ -276,10 +276,18 @@ std::pair<te::Schedule, Array<te::Tensor>>
ComputeDAG::ApplySteps(
// return value, so the ApplyToSchedule is not able to be merged to single
interface
if (auto ps = step.as<ReorderStepNode>()) {
ps->ApplyToSchedule(stages, stage_to_axes);
+ } else if (auto ps = step.as<ComputeAtStepNode>()) {
Review comment:
Ok, in the .h & .py I was tring to place the functions with similar
return format together.
void reorder
void compute_at
void compute_root
void compute_inline
Array\<Iterator\> split
Iterator fuse
Iterator vectorize
Iterator parallel
Iterator unroll
Iterator bind
It's fine to modified to the order you suggested.
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
[email protected]
