Skip to content

T
tic
Unverified Commit 406b5f76 by Andrew Committed by GitHub
Options

Fix tvm.target.generic_func runtime detection (#4910)

parent d2ae8c95
Showing with 3 additions and 3 deletions
python/tvm/target/generic_func.py
...@@ -20,10 +20,10 @@ import tvm._ffi ...@@ -20,10 +20,10 @@ import tvm._ffi
try: try:
from decorator import decorate from decorator import decorate
except ImportError as err_msg: except ImportError:
# Allow decorator to be missing in runtime # Allow decorator to be missing in runtime
if _LIB_NAME != "libtvm_runtime.so": if not tvm._ffi.base._RUNTIME_ONLY:
raise err_msg raise
from tvm.runtime import Object from tvm.runtime import Object
from . target import Target from . target import Target
......
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