FrozenGene commented on a change in pull request #6750:
URL: https://github.com/apache/incubator-tvm/pull/6750#discussion_r511737309
##########
File path: include/tvm/auto_scheduler/compute_dag.h
##########
@@ -194,6 +194,22 @@ class ComputeDAGNode : public Object {
TVM_DECLARE_FINAL_OBJECT_INFO(ComputeDAGNode, Object);
};
+/*!
+ * \brief Several options for applying layout rewrite.
+ * This is a optimization to rewrite the shape of input tensor according to
the schedule we get.
+ */
+enum class LayoutRewriteOption : int {
Review comment:
`enum class LayoutRewriteOption : uint8` should be enough.
----------------------------------------------------------------
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]
