From 8ee2fa9120727319c57f4c1acb90f4ab2b6284bc Mon Sep 17 00:00:00 2001 From: Eric Lunderberg Date: Wed, 16 Feb 2022 13:21:55 -0600 Subject: [PATCH] [VTA] Search for libvta_fsim.so in $TVM_LIBRARY_PATH This adds `$TVM_LIBRARY_PATH` to the search directory of `vta.libinfo.find_libvta`, matching the behavior of `tvm._ffi.libinfo.find_lib_path`. --- vta/python/vta/libinfo.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/vta/python/vta/libinfo.py b/vta/python/vta/libinfo.py index 8b300c3acb97..65ea7083a255 100644 --- a/vta/python/vta/libinfo.py +++ b/vta/python/vta/libinfo.py @@ -59,16 +59,17 @@ def find_libvta(lib_vta, optional=False): Enable error check """ curr_path = os.path.dirname(os.path.abspath(os.path.expanduser(__file__))) - lib_search = [ - os.path.join( + tvm_library_path = os.environ.get("TVM_LIBRARY_PATH", None) + if tvm_library_path is None: + tvm_library_path = os.path.join( curr_path, - "..", - "..", - "..", + os.pardir, + os.pardir, + os.pardir, "build", ) - ] - lib_search += [os.path.join(get_vta_hw_path(), "build")] + + lib_search = [tvm_library_path, os.path.join(get_vta_hw_path(), "build")] lib_name = _get_lib_name(lib_vta) lib_path = [os.path.join(x, lib_name) for x in lib_search] lib_found = [x for x in lib_path if os.path.exists(x)]