Skip to content

Commit

Permalink
load empty config (#6100)
Browse files Browse the repository at this point in the history
  • Loading branch information
comaniac authored Jul 21, 2020
1 parent 7cc0c93 commit 79be003
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions python/tvm/autotvm/record.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 79be003

Please sign in to comment.