Commit 84eea572 by Tianqi Chen Committed by GitHub

Update VTA schedule (#1464)

parent 5e16ba2b
pip3 install sphinx sphinx-gallery sphinx_rtd_theme sphinx_autodoc_annotator matplotlib Image commonmark>=0.7.3 docutils>=0.11
pip3 install sphinx sphinx-gallery sphinx_rtd_theme sphinx_autodoc_annotation matplotlib Image commonmark>=0.7.3 docutils>=0.11
......@@ -276,7 +276,7 @@ def schedule_conv2d(attrs, outs, target):
target = tvm.target.create(target)
if target.device_name == "vta":
return schedule_packed_conv2d(outs)
elif target.startswith("llvm"):
elif str(target).startswith("llvm"):
return tvm.create_schedule([x.op for x in outs])
else:
raise RuntimeError("not support target %s" % target)
......@@ -353,8 +353,11 @@ def schedule_packed_conv2d(outs):
else:
pad_data = None
wrkld = _get_workload(data, pad_data, kernel, output)
if wrkld in _WL2PLAN:
plan = _WL2PLAN[wrkld]
else:
plan = find_schedules(wrkld, vt_only=True, best_only=True)[0]
logging.info("Trying to find plan for %s", wrkld)
env = get_env()
load_inp = load_wgt = load_out = store_out = env.dma_copy
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment