support other formats

python/tvm/auto_scheduler/measure_record.py

@@ -18,6 +18,7 @@
 
 """ Serialization and other I/O support for measurement records (tuning logs). """
 import argparse
+import json
 import logging
 import os
 import itertools
@@ -79,6 +80,8 @@ def check_workload_key(self, inputs):
         """
         for inp in inputs:
             _, args = decode_workload_key(inp.task.workload_key)
+            if args is None:
+                continue
             if not args:
                 msg = (
                     "MeasureInput with old format workload key %s should be updated "

python/tvm/auto_scheduler/utils.py

@@ -44,7 +44,9 @@
 
 
 def decode_workload_key(workload_key):
-    """Decode the workload key from a string to a list.
+    """Decode the workload key from a string to the name and arguments. The wokrload key
+    is expected to be a list of "[func_name/hash, args ...]" in a JSON string. If not,
+    then simply return the workload key as the name without arguments.
 
     Parameters
     ----------
@@ -55,12 +57,16 @@ def decode_workload_key(workload_key):
     -------
     name: str
         The workload function name or the DAG hash.
-    args: List[Any]
-        The arguments of the workload.
+    args: Optional[List[Any]]
+        The arguments of the workload, or None if the workload key format is not decodeable.
     """
-    key_list = json.loads(workload_key)
-    assert len(key_list) >= 1
-    return key_list[0], key_list[1:]
+    try:
+        key_list = json.loads(workload_key)
+        if isinstance(key_list, list) and len(key_list) >= 1:
+            return key_list[0], key_list[1:]
+    except json.decoder.JSONDecodeError:
+        pass
+    return workload_key, None
 
 
 def get_func_name(func):