merrymercy commented on a change in pull request #7156:
URL: https://github.com/apache/tvm/pull/7156#discussion_r548960303
##########
File path: src/auto_scheduler/measure_record.cc
##########
@@ -165,12 +165,16 @@ struct Handler<::tvm::auto_scheduler::SearchTaskNode> {
writer->WriteArrayItem(*data.hardware_params.get());
if (data.target_host.defined()) {
writer->WriteArrayItem(data.target_host->str());
+ } else {
+ writer->WriteArrayItem(std::string(""));
}
+ writer->WriteArrayItem(static_cast<int>(data.layout_rewrite_option));
Review comment:
bump the version number for this change.
##########
File path: include/tvm/auto_scheduler/search_task.h
##########
@@ -144,9 +147,10 @@ class SearchTask : public ObjectRef {
* \param target The target device of this search task.
* \param target_host The target host device of this search task.
* \param hardware_params Hardware parameters used in this search task.
+ * \param layout_rewrite_option The default layout rewrite option used
during program measuring.
Review comment:
```suggestion
* \param layout_rewrite_option The layout rewrite option used for
measuring programs.
```
----------------------------------------------------------------
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]
