qh b^dZddlmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZddlZddlmZddlZddlmZddlmZddlmZdd lmZdd lmZm Z dd l!m"Z"d Z#Gd deZ$Gdde$Z%GddeZ&Gdde'Z(Gdde(e&Z)Gdde(e&Z*Gdde(e&Z+Gdde&Z,Gdde(e&Z-Gd d!eZ.Gd"d#e.Z/Gd$d%e.Z0Gd&d'e'Z1Gd(d)e$Z2Gd*d+e2Z3e4d,gZ5d-Z6y).z+ Caching mechanism for compiled functions. )ABCMetaabstractmethodabstractpropertyN)AppDirs)Path) NumbaWarning) BaseContext) CodeLibrary) CompileResult)configcompilerdumpscFtjr||z}t|yyN)r DEBUG_CACHEprint)msgargss L/var/www/html/engine/venv/lib/python3.12/site-packages/numba/core/caching.py _cache_logr s  Dj c cleZdZedZedZedZedZedZ edZ y)_Cachecy)zW The base filesystem path of this cache (for example its root folder). Nselfs r cache_pathz_Cache.cache_path(rcy)z Load an overload for the given signature using the target context. The saved object must be returned if successful, None if not found in the cache. Nrrsigtarget_contexts r load_overloadz_Cache.load_overload.r rcy)z< Save the overload for the given signature. Nrrr#datas r save_overloadz_Cache.save_overload6r rcy)z# Enable the cache. Nrrs renablez _Cache.enable<r rcy)z$ Disable the cache. Nrrs rdisablez_Cache.disableBr rcy)z" Flush the cache. Nrrs rflushz _Cache.flushHr rN) __name__ __module__ __qualname__rrrr%r)r+r-r/rrrrr&sx            rr) metaclassc:eZdZedZdZdZdZdZdZ y) NullCachecyrrrs rrzNullCache.cache_pathPsrcyrrr"s rr%zNullCache.load_overloadT rcyrr)rr#cress rr)zNullCache.save_overloadWr8rcyrrrs rr+zNullCache.enableZr8rcyrrrs rr-zNullCache.disable]r8rcyrrrs rr/zNullCache.flush`r8rN) r0r1r2propertyrr%r)r+r-r/rrrr5r5Os-      rr5cfeZdZdZdZedZedZedZe dZ e dZ y) _CacheLocatorz< A filesystem locator for caching a given function. c|j}tj|dtj|j y)NT)exist_ok)dir)get_cache_pathosmakedirstempfile TemporaryFileclose)rpaths rensure_cache_pathz_CacheLocator.ensure_cache_pathis5""$ D4(4(..0rcy)zA Return the directory the function is cached in. Nrrs rrDz_CacheLocator.get_cache_pathor rcy)z{ Get a timestamp representing the source code's freshness. Can return any picklable Python object. Nrrs rget_source_stampz_CacheLocator.get_source_stampur rcy)z Get a string disambiguator for this locator's function. It should allow disambiguating different but similarly-named functions. Nrrs rget_disambiguatorz_CacheLocator.get_disambiguator|r rct)ze Create a locator instance for the given function located in the given file. )NotImplementedErrorclspy_funcpy_files r from_functionz_CacheLocator.from_functions "!rcJtjj|}tjj|}tjj |d}t j |jj}dj||gS)zGiven the Python file path, compute a suitable path inside the cache directory. This will reduce a file path that is too long, which can be a problem on some operating system (i.e. Windows 7). _) rErJabspathdirnamesplithashlibsha1encode hexdigestjoin)rTrVrJsubpath parentdirhasheds rget_suitable_cache_subpathz(_CacheLocator.get_suitable_cache_subpathsswww'''//$'GGMM'*2. gnn./99;xxF+,,rN) r0r1r2__doc__rKrrDrNrP classmethodrWrfrrrr@r@dsp1       ""--rr@c,eZdZdZdZdZedZy)_SourceFileBackedLocatorMixinzf A cache locator mixin for functions which are backed by a well-known Python source file. cttddr$tjtj}ntj|j }|j |jfSNfrozenF)getattrsysrEstat executable_py_filest_mtimest_sizersts rrNz._SourceFileBackedLocatorMixin.get_source_stampsG 3% ((B'B{{BJJ&&rc,t|jSr)str_linenors rrPz/_SourceFileBackedLocatorMixin.get_disambiguators4<<  rctjj|sy|||} |j|S#t$rYywxYwr)rErJexistsrKOSErrorrTrUrVrs rrWz+_SourceFileBackedLocatorMixin.from_functionsMww~~g& 7G$   " " $    s= A A N)r0r1r2rgrNrPrhrWrrrrjrjs% '!  rrjc8eZdZdZdZdZefdZxZS)_UserProvidedCacheLocatorzd A locator that always point to the user provided directory in `numba.config.CACHE_DIR` c||_|jj|_|j |}t j jtj||_ yr) rr__code__co_firstlinenoryrfrErJrbr CACHE_DIR _cache_path)rrUrV cache_subpaths r__init__z"_UserProvidedCacheLocator.__init__sI ''66 77@ 77<<(8(8-Hrc|jSrrrs rrDz(_UserProvidedCacheLocator.get_cache_pathrcjtjsytt|}|j ||Sr)r rsuperrrW)rTrUrVparent __class__s rrWz'_UserProvidedCacheLocator.from_functions/ 0#6##GW55r) r0r1r2rgrrDrhrW __classcell__rs@rrrs'I  66rrceZdZdZdZdZy)_InTreeCacheLocatorzn A locator for functions backed by a regular Python module with a writable __pycache__ directory. c||_|jj|_tj j tj j|jd|_y)N __pycache__) rrrrryrErJrbr\r)rrUrVs rrz_InTreeCacheLocator.__init__sD ''66 77<< (F Vrc|jSrrrs rrDz"_InTreeCacheLocator.get_cache_pathrrN)r0r1r2rgrrDrrrrrs W  rrc,eZdZdZdZdZedZy)_UserWideCacheLocatorz A locator for functions backed by a regular Python module or a frozen executable, cached into a user-wide cache directory. c||_|jj|_t dd}|j }|j |}tjj|||_ yNnumbaF)appname appauthor) rrrrryruser_cache_dirrfrErJrbrrrUrVappdirs cache_dirrs rrz_UserWideCacheLocator.__init__sZ ''66 'U;** 77@ 77<< =Arc|jSrrrs rrDz$_UserWideCacheLocator.get_cache_pathrrctjj|sttddsy|||} |j |S#t $rYywxYwrl)rErJr{rnrorKr|r}s rrWz#_UserWideCacheLocator.from_functionsZw'73%+H 7G$   " " $    sA AAN)r0r1r2rgrrDrhrWrrrrrs& B   rrc8eZdZdZdZdZdZdZedZ y)_IPythonCacheLocatorzT A locator for functions entered at the IPython prompt (notebook or other). c||_tj|}t|tr||_y|j d|_y)Nzutf-8)rrinspect getsource isinstancebytes _bytes_sourcer`)rrUrVsources rrz_IPythonCacheLocator.__init__s> ""7+ fe $!'D !'w!7D rc ddlm}tj j |dS#t$r ddlm}Y6wxYw)Nr)get_ipython_cache_dir numba_cache) IPython.pathsr ImportErrorIPython.utils.pathrErJrb)rrs rrDz#_IPythonCacheLocator.get_cache_paths<  A ;ww||13]CC A @ As -??c\tj|jjSr)r^sha256rrars rrNz%_IPythonCacheLocator.get_source_stamps ~~d001;;==rcdj|jjddd}tj|j ddS)NrT )rbr splitlinesr^rra)r firstliness rrPz&_IPythonCacheLocator.get_disambiguator sI XXd00;;DA"1EF ~~j)335cr::rc|jdsLtjjtjj |jdsy|||} |j |S#t $rYywxYw)Nz ;  rrcBeZdZdZdZedZdZdZe dZ y)_ZipCacheLocatorzP A locator for functions backed by Python modules within a zip archive. c"||_|jj|_|j |\|_|_tdd}|j}|j|}tjj|||_ yr)rrrrry_split_zip_path _zip_path_internal_pathrrrfrErJrbrrs rrz_ZipCacheLocator.__init__>sv ''66 .2.B.B7.K++'U;** 77@ 77<< =Arc t|}t|jD]^\}}|jdst t|jd|dz}t t|j|dzd}||fcSt d)N.zipzNo zip file found in path)r enumeratepartsendswithrx ValueError)rVrJipartzip_path internal_paths rrz _ZipCacheLocator._split_zip_pathJsG} , /GAt}}V$tTZZ!a%%89: #D$**QUW*=$> ? ..  / 455rc|jSrrrs rrDz_ZipCacheLocator.get_cache_pathTrrcptj|j}|j|jfSr)rErprrsrtrus rrNz!_ZipCacheLocator.get_source_stampWs' WWT^^ ${{BJJ&&rcd|vry|||S)NrrrSs rrWz_ZipCacheLocator.from_function[s  7G$$rN) r0r1r2rgr staticmethodrrDrNrhrWrrrrr9s? B66 '%%rrczeZdZdZeeeeegZ dZ dZ e dZ e dZedZedZedZy ) CacheImplz Provides the core machinery for caching. - implement how to serialize and deserialize the data in the cache. - control the filename of the cache. - provide the cache locator c8|jj|_ |j}t j|}|jD]}|j||}|ntd|d|||_ t j|}tjjtjj|d}|d|}t!t"dd} |j%|| |_y#t$r|j }YwxYw)Nzcannot cache function z : no locator available for file r.abiflags)rrryr2AttributeErrorr0rgetfile_locator_classesrW RuntimeError_locatorrErJsplitextrrnroget_filename_base_filename_base) rrUqualname source_pathrTlocatorfilenamemodnamefullnamers rrzCacheImpl.__init__qs ''66  (++Hoog. (( HC''=G" H 08+ GH H ??7+''""277#3#3H#=>qA%x03 B/"44XxH% (''H (s DDDc|jddjdd}d}|||jjtjdtjd|fzS)Nz%s-%s.py%d%d%srr)replacerrPro version_info)rrrfixed_fullnamefmts rrzCacheImpl.get_filename_basesk"))#r2::3Cndll&D&D&F&&q)3+;+;A+>JJ Jrc|jSr)rrs r filename_basezCacheImpl.filename_bases"""rc|jSr)rrs rrzCacheImpl.locators }}rcy)z$Returns the serialized form the dataNrrr(s rreducezCacheImpl.reduce rcy)z4Returns the de-serialized form of the *reduced_data*Nr)rr$ reduced_datas rrebuildzCacheImpl.rebuildrrcy)zEReturns True if the given data is cachable; otherwise, returns False.Nrrs rcheck_cachablezCacheImpl.check_cachablerrN)r0r1r2rgrrrrrrrrr>rrrrrrrrrrras " I2J##      rrc"eZdZdZdZdZdZy)CompileResultCacheImplz> Implements the logic to cache CompileResult objects. c"|jS)z4 Returns a serialized CompileResult )_reduce)rr:s rrzCompileResultCacheImpl.reduces||~rcBtjj|g|S)z8 Returns the unserialized CompileResult )r r _rebuildrr$payloads rrzCompileResultCacheImpl.rebuilds %%..~HHHrcLd}td|jDrd}n|jjrd}|rid|jj j ddd|}tj|t|jj|jy y ) z@ Check cachability of the given compile result. Nc36K|]}|j ywr) can_cache).0xs r z8CompileResultCacheImpl.check_cachable..s411;;4szas it uses lifted codezLas it uses dynamic globals (such as ctypes pointers and large global arrays)z Cannot cache compiled function "rrYz" FT) anyliftedlibraryhas_dynamic_globalsfndescrr]warnings warn_explicitrrrrry)rr: cannot_cachers rrz%CompileResultCacheImpl.check_cachables 4 4 43L \\ - -PL kk**005b9<IC  " "3 #'==#9#94<< IrN)r0r1r2rgrrrrrrrrs I rrc8eZdZdZdZdZdZdZfdZxZ S)CodeLibraryCacheImplz< Implements the logic to cache CodeLibrary objects. Nc"|jS)z2 Returns a serialized CodeLibrary )serialize_using_object_codercodelibs rrzCodeLibraryCacheImpl.reduces2244rc@|jj|S)z6 Returns the unserialized CodeLibrary )codegenunserialize_libraryrs rrzCodeLibraryCacheImpl.rebuilds%%';;GDDrc|j S)z= Check cachability of the given CodeLibrary. )r rs rrz#CodeLibraryCacheImpl.check_cachables....rctt|}|j||}dj|j|gS)N-)rrrrb_filename_prefix)rrrrresrs rrz&CodeLibraryCacheImpl.get_filename_bases<+T2&&x:xx..455r) r0r1r2rgrrrrrrrs@rrrs*5 E / 66rrcveZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zej d Zy)IndexDataCacheFilezP Implements the logic for the index file and data file used by a cache. c||_|d|_tjj |j|j|_|d|_||_tj|_ y)Nz.nbiz.{number:d}.nbc) r _index_namerErJrb _index_path_data_name_pattern _source_stampr __version___version)rrr source_stamps rrzIndexDataCacheFile.__init__sX%'4677<<(8(8$:J:JK9F"H))) rc&|jiyr) _save_indexrs rr/zIndexDataCacheFile.flushs rc2|j} ||}|j||y#t$rct|j}t j dD]}|j |}||vsn||<|j|Y~wxYw)z? Save a new cache entry with *key* and *data*. rN) _load_indexKeyErrorsetvalues itertoolscount _data_namer' _save_data)rkeyr( overloads data_nameexistingrs rsavezIndexDataCacheFile.saves$$&  (!#I  4( (9++-.H__Q'  OOA. H, 'IcN   Y ' (s*AB:BBc|j}|j|}|y |j|S#t$rYywxYw)z0 Load a cache entry with *key*. N)r)get _load_datar|)rr1r2r3s rloadzIndexDataCacheFile.loadsO$$& MM#&    ??9- -   s7 AAcj t|jd5}tj|}|j }ddd|j k7riStj\}}td|j||jk7riS|S#1swY[xYw#t $ricYSwxYw)z| Load the cache index and return it as a dictionary (possibly empty if cache is empty or obsolete). rbNz[cache] index loaded from %r) openr pickler9readFileNotFoundErrorr$loadsrr")rfversionr(stampr2s rr)zIndexDataCacheFile._load_indexs  d&&-  ++a.vvx dmm #I!<<-y143C3CD D&& &I #  ! I s'B$&BB$B!B$$ B21B2c4|j|f}|j|}|j|j5}t j |j |d|j|dddtd|jy#1swY xYw)NrY)protocolz[cache] index saved to %r) r"_dump_open_for_writer r=dumpr$writer)rr2r(rAs rr'zIndexDataCacheFile._save_index6s}!!9,zz$  ! !$"2"2 3 q KK q2 6 GGDM  .0@0@A  s 4BBc|j|}t|d5}|j}dddtj}t d||S#1swY,xYw)Nr;zache] data loaded from %r) _data_pathr<r>r=r@r)rnamerJrAr(tups rr8zIndexDataCacheFile._load_data>sZt$ $  668D ll4 0$7   s AA#c|j|}|j|}|j|5}|j|dddt d|y#1swYxYw)Nz[cache] data saved to %r)rFrKrGrIr)rrLr(rJrAs rr0zIndexDataCacheFile._save_dataFsYzz$t$  ! !$ ' 1 GGDM -t4  s AA$c:|jj|S)N)number)r!format)rrPs rr/zIndexDataCacheFile._data_nameMs&&--V-<.pyXY.nbi") which contains a mapping of signatures and architectures to data files. It is prefixed by a versioning key and a timestamp of the Python source file containing the function. There is one data file ("function_name-.pyXY..nbc") per function, function signature, target architecture and Python version. Separate index and data files per Python version avoid pickle compatibility problems. Note: This contains the driver logic only. The core logic is provided by a subclass of ``CacheImpl`` specified as *_impl_class* in the subclass. Nc~t||_||_|j||_|jj j |_|jj j}|jj}t|j|||_ |jy)N)rrr%) repr_name_py_func _impl_class_implrrDrrNrr _cache_filer+)rrUr%rs rrzCache.__init__s']  %%g. ::--<<>zz))::<  00 -9I9I#9#9#;<$$S)  ::%%nd;D rcr|j5|j||dddy#1swYyxYw)zE Save the data for the given signature in the cache. N)ru_save_overloadr's rr)zCache.save_overloads5 3 3 5 +   T * + + +s-6cD|jsy|jj|sy|jjj |j ||j }|jj|}|jj||yr) rprirrrKrxrrrjr5)rr#r(r1s rrzzCache._save_overloadsu}} zz((.  ,,.ooc4<<0zz  & c4(rc#Ktjdk(r dydy#t$r(}|jtjk7rYd}~yd}~wwxYww)Nnt)rErLr|errnoEACCES)res rruz'Cache._guard_against_spurious_io_errorssG 77d?    77ell*+ s*A!A AA A AAcJ|jjj}|jjCt |jjDcgc]}|j c}}t |}nd}d}||j||||ffScc}w)a  Compute index key for the given signature and codegen. It includes a description of the OS, target architecture and hashes of the bytecode for the function and, if the function has a __closure__, a hash of the cell_contents. rcHtj|jSr)r^rra)rs rz"Cache._index_key..s7>>!,668r)rgrco_code __closure__tuple cell_contentsr magic_tuple)rr#r codebytesrcvars cvarbyteshashers rrxzCache._index_keysMM**22 == $ $ 0DMM4M4MNq1??NOEe II8W((*VI->-3I->-AB BOsB )r0r1r2rgrhrrlr>rr+r-r/r%rvr)rzr`rarurxrrrrcrcksp,K I  !<+)  BrrcceZdZdZeZy) FunctionCachezF Implements Cache that saves and loads CompileResult objects. N)r0r1r2rgrrhrrrrrs)KrrrctvsJtjGfddtGfddt}|S)z Create a Cache class for additional compilation features to cache their result for reuse. The cache is saved in filename pattern like in ``FunctionCache`` but with additional *prefix* as specified. ceZdZWZy)6make_library_cache..CustomCodeLibraryCacheImplN)r0r1r2r)prefixsrCustomCodeLibraryCacheImplrs !rrceZdZdZWZy)(make_library_cache..LibraryCachez Implements Cache that saves and loads CodeLibrary objects for additional feature for the specified python function. N)r0r1r2rgrh)rsr LibraryCachers 1 rr)_lib_cache_prefixesaddrrc)rrrs` @rmake_library_cachersC , ,, ,F#"%9"1u1 r)7rgabcrrrr`r~r^rr-rEr=rorGrXr numba.misc.appdirsrzipfilepathlibrrnumba.core.errorsrnumba.core.baser numba.core.codegenr numba.core.compilerr numba.corer r numba.core.serializerrrr5r@objectrjrrrrrrrrrrcrr+rrrrrrs; :9   & *'*-'& & w& R  *6-g6-rF@6 =}6,  7  9=@4=4n&%4m&%PF 'F R!Y!H696>{{|{BF{B|)E)2$ir