comaniac commented on a change in pull request #7344:
URL: https://github.com/apache/tvm/pull/7344#discussion_r565517145
##########
File path: python/tvm/auto_scheduler/dispatcher.py
##########
@@ -126,18 +127,53 @@ class ApplyHistoryBest(DispatchContext):
If is str, then it should be the filename of a records log file.
Each row of this file is an encoded record pair. Otherwise, it is an
iterator.
n_lines: Optional[int]
- if it is not None, only load the first `n_lines` lines of log
+ if it is not None, only load the first `n_lines` lines of log.
+ include_compatible: bool
+ When set to True, compatible records will also be considered.
"""
- def __init__(self, records, n_lines=None):
+ def __init__(self, records, n_lines=None, include_compatible=False):
super(ApplyHistoryBest, self).__init__()
+ self.include_compatible = include_compatible
+ # Dict[str (target key),
+ # Dict[str (workload hash),
+ # Dict[tuple (workload args), tuple (State, cost)]]]
self.best_by_targetkey = {}
self.best_by_model = {}
self._best_user_defined = {}
self.load(records, n_lines)
+ @staticmethod
+ def get_workload_entry(best_records, target_key, workload_key):
+ """Get the entry of the target key and workload key hash in the given
best record map.
+
+ Parameters
+ ----------
+ best_records: Dict[str, Dict[str, Dict[str, Any]]]
+ The best record map.
+ target_key: str
+ The first key to the best_records.
+ workload_key: str
+ The workload key that can be decoded to workload hash and args.
+
+ Returns
+ -------
+ entry: Dict[str, Any]
+ The entry in best_records with target key and workload hash.
+ workload_hash: str
+ The workload hash.
Review comment:
```suggestion
The workload hash decoded from workload_key
```
##########
File path: python/tvm/auto_scheduler/dispatcher.py
##########
@@ -126,18 +127,53 @@ class ApplyHistoryBest(DispatchContext):
If is str, then it should be the filename of a records log file.
Each row of this file is an encoded record pair. Otherwise, it is an
iterator.
n_lines: Optional[int]
- if it is not None, only load the first `n_lines` lines of log
+ if it is not None, only load the first `n_lines` lines of log.
+ include_compatible: bool
+ When set to True, compatible records will also be considered.
"""
- def __init__(self, records, n_lines=None):
+ def __init__(self, records, n_lines=None, include_compatible=False):
super(ApplyHistoryBest, self).__init__()
+ self.include_compatible = include_compatible
+ # Dict[str (target key),
+ # Dict[str (workload hash),
+ # Dict[tuple (workload args), tuple (State, cost)]]]
self.best_by_targetkey = {}
self.best_by_model = {}
self._best_user_defined = {}
self.load(records, n_lines)
+ @staticmethod
+ def get_workload_entry(best_records, target_key, workload_key):
+ """Get the entry of the target key and workload key hash in the given
best record map.
+
+ Parameters
+ ----------
+ best_records: Dict[str, Dict[str, Dict[str, Any]]]
+ The best record map.
+ target_key: str
+ The first key to the best_records.
+ workload_key: str
+ The workload key that can be decoded to workload hash and args.
+
+ Returns
+ -------
+ entry: Dict[str, Any]
+ The entry in best_records with target key and workload hash.
+ workload_hash: str
+ The workload hash.
Review comment:
```suggestion
The workload hash decoded from workload_key.
```
----------------------------------------------------------------
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]
