o Y۷i@sZdZddlZddlZddlmZddlmZdefddZd d d Z e d kr+e dSdS) z6Config utilities for finding paths to lib and headers.N)Path)libinforeturncCstt}t|jS)z'Find the library directory for tvm-ffi.)rrfind_libtvm_ffistrparent) libtvm_ffir D/home/ubuntu/vllm_env/lib/python3.10/site-packages/tvm_ffi/config.py _find_libdirs  r cCs$tjdd}|jdddd|jdddd|jd dd d|jd dd d|jd ddd|jddd d|jdddd|jdddd|jdddd|jdddd|jdddd|}ttjdkro|dS|jrxt t |j rt t |jrt t |jrt t|jrtjdrt t nt t |jrt t |jrt t |jrt }t }t d|d|d|jrt }t }t d|d||jrtjdrt t nt d |jrtjdst d!tdSdSdS)"z?Parse CLI args and print build and include configuration paths.zDGet various configuration information needed to compile with tvm-ffi) descriptionz --includedir store_truezPrint include directory)actionhelpz--dlpack-includedirzPrint dlpack include directoryz --cmakedirzPrint library directoryz --sourcedirzPrint source directoryz --libfilesz!Fully qualified library filenamesz--libdirz--libszLibraries to be linkedz--cython-lib-pathzPrint cython pathz --cxxflagszPrint cxx flagsz--cflagsz Print c flagsz --ldflagszPrint ld flagsrNwin32z-Iz -Iz -std=c++17z -ltvm_ffiz-L)argparseArgumentParser add_argument parse_argslensysargv print_help includedirprintrfind_include_pathdlpack_includedirfind_dlpack_include_pathcmakedirfind_cmake_pathlibdirr libfilesplatform startswithfind_windows_implibr sourcedirfind_source_pathcython_lib_pathfind_cython_libcxxflagscflagslibsldflags)parserargs include_dirdlpack_include_dirr r r __main__ sn         r2)rN) __doc__rrpathlibrrrr r2__name__r r r r s   @