diff --git a/hpvm/tools/py-approxhpvm/main.py.in b/hpvm/tools/py-approxhpvm/main.py.in index 4b604d172a1515f6a11dccd82402a53f16410300..7f5459d9e68f1e084c114dc1c39c0312726059d1 100644 --- a/hpvm/tools/py-approxhpvm/main.py.in +++ b/hpvm/tools/py-approxhpvm/main.py.in @@ -111,10 +111,16 @@ def link_hpvm_rt(src_file: PathLike, target_file: PathLike) -> List[str]: def link_binary(src_file: PathLike, target_file: PathLike) -> List[str]: + def drop_suffix(libname: str): + import re + + match = re.match(r"lib(.*)\.so", libname) + return libname if match is None else match.group(1) + linker_dir_flags = [] for path in TRT_LINK_DIRS: linker_dir_flags.extend([f"-L{path}", f"-Wl,-rpath={path}"]) - linker_lib_flags = [f"-l{lib}" for lib in TRT_LINK_LIBS] + linker_lib_flags = [f"-l{drop_suffix(lib)}" for lib in TRT_LINK_LIBS] return [ str(LLVM_BUILD_BIN / "clang++"), str(src_file), str(TRT_PATH), "-o", str(target_file), *linker_dir_flags, *linker_lib_flags