Skip to content

Commit

Permalink
[Bugfix][AutoScheduler] Strictly select impl using plevel (apache#6956)
Browse files Browse the repository at this point in the history
* [Bugfix][AutoScheduler] Strictly select impl using plevel

* lint
  • Loading branch information
comaniac authored and Trevor Morris committed Dec 4, 2020
1 parent 9fcb6ff commit 57ba863
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion python/tvm/relay/backend/compile_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
import numpy as np
import tvm
from tvm import te, autotvm
from tvm.ir.transform import PassContext
from tvm.runtime import Object
from tvm.support import libinfo
from tvm.target import Target
Expand Down Expand Up @@ -287,7 +288,10 @@ def lower_call(call, inputs, target):
env.tracing = False
reenable_tracing = True

if not is_dyn:
# check if auto_scheduler is enabled, and use pevel to select the implementation if so
use_auto_scheduler = PassContext.current().config.get("relay.backend.use_auto_scheduler", False)

if not is_dyn and not use_auto_scheduler:
best_impl, outputs = select_implementation(op, call.attrs, inputs, ret_type, target)
else:
# TODO(@icemelon9): Allow tvm to generate multiple kernels for dynamic shapes.
Expand Down

0 comments on commit 57ba863

Please sign in to comment.