a d@sddlZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZejjjZejjjZejjjZejjZddd Zd d ZeedZeed Zed dZejd dddZGddde Z!Gddde Z"ddZ#ddZ$dS)N) build_ext) Distribution)mkdtemp)contextmanager)Path)z.dllz.pyd).sor)windefaultcCsttjddtd|S)Nr )_configsgetsysplatformargr\/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/numba/pycc/platform.py get_configssrc cszVt}ttj|d|d}|VWz|t|Wn Yq0t|n0z|t|Wn Yn 0t|0dS)Nztemp%swt)ropenospathjoincloseremovermdir)suffixtmpdirntfrrr _gentmpfile#s  r )maxsizec Cst}t|dD]}zdt|H}d}|||||j|jgt|jj dWdn1sn0YWqt yYdS0qdS)zv Returns True if the "external compiler" bound in numpy.distutil is present and working, False otherwise. )z.cz.cxxzint main(void) { return 0; }) output_dirNFT) new_compilercustomize_compilerr writeflushrcompilenameranchor Exception)compilerrrZsimple_crrrexternal_compiler_works5s  <  r,c@seZdZgZdS)_DummyExtensionN)__name__ __module__ __qualname__ librariesrrrrr-Lsr-c@sjeZdZddZeddZejddZddZdd d Zdd d Z ddZ ddZ ddZ ddZ d S) ToolchaincCsts|d|_t|_t|jtt|_|j |jj |_ |jj |_ dtgi}tjdkrrg|d<n dg|d<||_dS)NF include_dirswin32r1m)r,_raise_external_compiler_error_verboser# _compilerr$rr _build_extfinalize_options library_dirs _py_lib_dirsr3_py_include_dirsnpZ get_includer r _math_info)selfZnp_compile_argsrrr__init__Rs        zToolchain.__init__cCs|jSN)r7r@rrrverbosedszToolchain.verbosecCs ||_t|rtjntjdSrB)r7log set_thresholdINFOWARN)r@valuerrrrDhscCsd}d}tj}|drLtjdkr,ddg}nddg}d ||d |f}nN|d rvd d g}d ||d |f}n$|drd}d ||f}nd|}t|dS)NzYAttempted to compile AOT function without the compiler used by `numpy.distutils` present.z(If using conda try: #> conda install %slinuxlz gcc_linux-32z gxx_linux-32z gcc_linux-64z gxx_linux-64z%s %s darwinz clang_osx-64zclangxx_osx-64r4zCannot find suitable msvc.zUnknown platform %s)r r startswithr!r RuntimeError)r@ZbasemsgZ conda_msgZpltZ compilersmsgZwinmsgrrrr6os      z(Toolchain._raise_external_compiler_errorrNcCs |jj|||||pg|d}|S)a Compile the given source files into a separate object file each, all beneath the *output_dir*. A list of paths to object files is returned. *macros* has the same format as in distutils: a list of 1- or 2-tuples. If a 1-tuple (name,), the given name is considered undefined by the C preprocessor. If a 2-tuple (name, value), the given name is expanded into the given value by the C preprocessor. )r"r3dependsmacros extra_preargs)r8r')r@sourcesr"r3rPrQZ extra_cflagsobjectsrrrcompile_objectsszToolchain.compile_objectsc Cs2tj|\}}|jjtj|||||||ddS)zu Create a shared library *output* linking the given *objects* and *libraries* (all strings). )export_symbolsrRN)rrsplitr8link CCompilerZ SHARED_OBJECT) r@outputrTr1r;rVZ extra_ldflagsr"Zoutput_filenamerrr link_shareds zToolchain.link_sharedcCs0|jt}tjdkr"|dg}||jdS)zJ Get the library arguments necessary to link with Python. r4msvcrtr1)r9Z get_librariesr-r rr?)r@Zlibsrrrget_python_librariess  zToolchain.get_python_librariescCs t|jS)zL Get the library directories necessary to link with Python. )listr<rCrrrget_python_library_dirssz!Toolchain.get_python_library_dirscCst|j|jdS)zo Get the include directories necessary to compile against the Python and Numpy C APIs. r3)r^r=r?rCrrrget_python_include_dirssz!Toolchain.get_python_include_dirscCs |j|S)zR Given a C extension's module name, return its intended filename. )r9get_ext_filename)r@Zext_namerrrraszToolchain.get_ext_filename)rrrN)rrrN)r.r/r0rApropertyrDsetterr6rUr[r]r_r`rarrrrr2Ps"     r2cCsd|vrd|vrd|S|S)zB Quote the argument for safe use in a shell command line. "rKz"%s"rrrrr _quote_argsrecCs:t|ttfrdSzt|WdSty4YdS0dS)NFT) isinstancestrbyteslenr*rrrr _is_sequences rj)% setuptoolsZsetuptools.command.build_extrZsetuptools.distrnumpyr> functoolsr subprocessr tempfiler contextlibrpathlibr distutilsZ ccompilerrYr# sysconfigr$rEr rpartialZfind_shared_endingZfind_pyext_endingr lru_cacher,objectr-r2rerjrrrrs8             x