Clean up common functionality

Author: driazati

python/tvm/_ffi/libinfo.py

@@ -17,6 +17,10 @@
 """Library information."""
 import sys
 import os
+from pathlib import Path
+
+
+REPO_ROOT = Path(__file__).resolve().parent.parent.parent.parent
 
 
 def split_env_var(env_var, split):
@@ -68,6 +72,10 @@ def get_dll_directories():
     dll_path.append(os.path.join(ffi_dir, ".."))
     # Default cmake build directory
     dll_path.append(os.path.join(source_dir, "build"))
+
+    # Other possible build directories
+    dll_path += list(REPO_ROOT.glob("build-*"))
+
     dll_path.append(os.path.join(source_dir, "build", "Release"))
     # Default make build directory
     dll_path.append(os.path.join(source_dir, "lib"))
@@ -98,6 +106,10 @@ def find_lib_path(name=None, search_path=None, optional=False):
     use_runtime = os.environ.get("TVM_USE_RUNTIME_LIB", False)
     dll_path = get_dll_directories()
 
+    manual_dll_path = os.getenv("TVM_DLL_PATH")
+    if manual_dll_path is not None:
+        dll_path = [manual_dll_path] + dll_path
+
     if search_path is not None:
         if isinstance(search_path, list):
             dll_path = dll_path + search_path