o
    ºiÏ  ã                   @   sZ   d Z ddlZddlZddlmZ ddlmZ defdd„Zdd	d
„Z	e
d
kr+e	ƒ  dS dS )z6Config utilities for finding paths to lib and headers.é    N)ÚPathé   )ÚlibinfoÚreturnc                  C   s   t t ¡ ƒ} t| jƒS )z'Find the library directory for tvm-ffi.)r   r   Úfind_libtvm_ffiÚstrÚparent)Ú
libtvm_ffi© r
   úK/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/tvm_ffi/config.pyÚ_find_libdir   s   
r   c                  C   s$  t jdd} | jdddd | jdddd | jd	dd
d | jdddd | jdddd | jddd
d | jdddd | jdddd | jdddd | jdddd | jdddd |  ¡ }ttjƒdkro|  ¡  dS |jrxt	t
 ¡ ƒ |jrt	t
 ¡ ƒ |jrŠt	t
 ¡ ƒ |jr’t	tƒ ƒ |jr¨tj d¡r¢t	t
 ¡ ƒ nt	t
 ¡ ƒ |jr±t	t
 ¡ ƒ |jrºt	t
 ¡ ƒ |jrÐt
 ¡ }t
 ¡ }t	d|› d|› dƒ |jråt
 ¡ }t
 ¡ }t	d|› d|› ƒ |jrùtj d¡rõt	t
 ¡ ƒ nt	d ƒ |jrtj d¡st	d!tƒ › ƒ dS dS dS )"z?Parse CLI args and print build and include configuration paths.zDGet various configuration information needed to compile with tvm-ffi)Údescriptionz--includedirÚ
store_truezPrint include directory)ÚactionÚhelpz--dlpack-includedirzPrint dlpack include directoryz
--cmakedirzPrint library directoryz--sourcedirzPrint source directoryz
--libfilesz!Fully qualified library filenamesz--libdirz--libszLibraries to be linkedz--cython-lib-pathzPrint cython pathz
--cxxflagszPrint cxx flagsz--cflagszPrint c flagsz	--ldflagszPrint ld flagsr   NÚwin32z-Iz -Iz -std=c++17z	-ltvm_ffiz-L)ÚargparseÚArgumentParserÚadd_argumentÚ
parse_argsÚlenÚsysÚargvÚ
print_helpÚ
includedirÚprintr   Úfind_include_pathÚdlpack_includedirÚfind_dlpack_include_pathÚcmakedirÚfind_cmake_pathÚlibdirr   ÚlibfilesÚplatformÚ
startswithÚfind_windows_implibr   Ú	sourcedirÚfind_source_pathÚcython_lib_pathÚfind_cython_libÚcxxflagsÚcflagsÚlibsÚldflags)ÚparserÚargsÚinclude_dirÚdlpack_include_dirr
   r
   r   Ú__main__    sn   ÿý
þr2   )r   N)Ú__doc__r   r   Úpathlibr   Ú r   r   r   r2   Ú__name__r
   r
   r
   r   Ú<module>   s   
@
ÿ