jcf94 commented on a change in pull request #7156:
URL: https://github.com/apache/tvm/pull/7156#discussion_r548513446



##########
File path: python/tvm/auto_scheduler/search_task.py
##########
@@ -245,15 +254,18 @@ def tune(self, tuning_options, search_policy=None):
 
         _ffi_api.AutoSchedule(search_policy, tuning_options)
 
-    def apply_best(self, log_file, layout_rewrite_option=None):
+    def apply_best(self, log_file, 
layout_rewrite_option=LayoutRewriteOption.NO_REWRITE):

Review comment:
       Emm ... Then we should also add this to maybe MeasureInput? I'm thinking 
this may not be so necessary, since we'll always try to apply layout rewrite on 
network, and this API may only be used in the replying of standalone ops. The 
user of this API should know what this means.




----------------------------------------------------------------
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]


Reply via email to