[GitHub] [incubator-tvm] merrymercy commented on a change in pull request #6297: [Ansor][AutoTVM v2.0] Phase 2: Layout Rewrite in AutoScheduler

[GitBox <gi...@apache.org>]

merrymercy commented on a change in pull request #6297:
URL: https://github.com/apache/incubator-tvm/pull/6297#discussion_r481595812



##########
File path: include/tvm/auto_scheduler/compute_dag.h
##########
@@ -205,19 +205,28 @@ class ComputeDAG : public ObjectRef {
    */
   TVM_DLL explicit ComputeDAG(Array<te::Tensor> tensors);
 
+  /*!
+   * \brief Rewrite the layout of placeholder specified by attr `layout_free_placeholders`
+   * according to the loop nest derived with `transform_steps`.
+   * \param transform_steps Transform steps of a state.
+   */
+  void RewriteLayout(const Array<Step>& transform_steps);
+
   /*!
    * \brief Apply the history transform steps to get a TVM schedule.
    * \param transform_steps Transform steps of a state.
    * \param stages The list of stages after applying the steps.
    * Pass a valid pointer if this information needs to be used outside this function.
    * \param stage_to_axes The map that stores all axes for one stage.
    * Pass a valid pointer if this information needs to be used outside this function.
+   * \param layout_rewrite Rewrite the layout of placeholder.

Review comment:
       ```suggestion
      * \param layout_rewrite Rewrite the layout of placeholders specified by
      * attr `layout_free_placeholders`
   ```




----------------------------------------------------------------
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:
users@infra.apache.org