Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[AutoTVM] Load configs even it has no entity #6100

Merged
merged 1 commit into from Jul 21, 2020
Merged

[AutoTVM] Load configs even it has no entity #6100

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
8 changes: 1 addition & 7 deletions python/tvm/autotvm/record.py
View file
Open in desktop
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