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

alexwong · Feb 26, 2020 · ed8584b · ed8584b
1 parent 6e15f3c
commit ed8584b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/tvm/target/generic_func.py b/python/tvm/target/generic_func.py
@@ -20,10 +20,10 @@
 
 try:
     from decorator import decorate
-except ImportError as err_msg:
+except ImportError:
     # Allow decorator to be missing in runtime
-    if _LIB_NAME != "libtvm_runtime.so":
-        raise err_msg
+    if not tvm._ffi.base._RUNTIME_ONLY:
+        raise
 
 from tvm.runtime import Object
 from . target import Target