projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
7cc0c93
)
load empty config (#6100)
author
Cody Yu
<comaniac0422@gmail.com>
Tue, 21 Jul 2020 14:52:17 +0000
(07:52 -0700)
committer
GitHub
<noreply@github.com>
Tue, 21 Jul 2020 14:52:17 +0000
(07:52 -0700)
python/tvm/autotvm/record.py
patch
|
blob
|
history
diff --git
a/python/tvm/autotvm/record.py
b/python/tvm/autotvm/record.py
index
260edbb
..
5a61c34
100644
(file)
--- a/
python/tvm/autotvm/record.py
+++ b/
python/tvm/autotvm/record.py
@@
-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):