load empty config (#6100)
authorCody Yu <comaniac0422@gmail.com>
Tue, 21 Jul 2020 14:52:17 +0000 (07:52 -0700)
committerGitHub <noreply@github.com>
Tue, 21 Jul 2020 14:52:17 +0000 (07:52 -0700)
python/tvm/autotvm/record.py

index 260edbb..5a61c34 100644 (file)
@@ -209,13 +209,7 @@ def load_from_file(filename):
             ret = decode(row)
             if ret is None:
                 continue
-            inp, res = ret
-            # Avoid loading the record with an empty config. The TOPI schedule with no entities
-            # will result in an empty entity map (e.g., depthwise_conv2d_nchw on x86).
-            # Using an empty config will cause problems when applying alter op like NCHW to NCHWc.
-            if not inp.config._entity_map:
-                continue
-            yield (inp, res)
+            yield ret
 
 
 def split_workload(in_file, clean=True):