{-jddlZddlZddlmZddlmZeeejdZGddeZ Gdd Z Gd d Z d Z dS) N)Enum) get_loggerz&%(asctime)s-%(levelname)s: %(message)s)fmtceZdZdZdZdS) FunctionTyperN)__name__ __module__ __qualname__ FP16_ONLYCOMMONc/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/static/amp/function_overload.pyrrsI FFFrrceZdZdZdZdZdS)Functiona; Function is a wrap over standard python function An instance of this Function class is also callable just like the python function that it wrapped. When the instance is "called" like a function it fetches the function to be invoked from the virtual namespace and then invokes the same. c||_dSN)fn)selfrs r__init__zFunction.__init__,s rcXtj|i|}||i|S)z] Overriding the __call__ function which makes the instance callable. ) Namespace get_instanceget)rargskwargsrs r__call__zFunction.__call__/s: # # % % )4 :6 : :r4"6"""rN)r r r __doc__rrrrrrr"s< # # # # #rrc>eZdZdZdZdZedZdZdZ dS)rza Namespace is the singleton class that is responsible for holding all the functions. NcX|ji|_|t_dStd)Nz#cannot instantiate Namespace again.)_Namespace__instance function_mapr Exception)rs rrzNamespace.__init__Cs/ ? " "D #'I ABB BrcNtjttjSr)rr"rrrrzNamespace.get_instanceJs   ' KKK##rct|tsJdt|dt|}||j|<|S)a Register the function in the virtual namespace and return an instance of callable Function that wraps the function fn. Args: fn (function): the native python function handle. key (FunctionType): the specified type. z>The type of key is expected to be FunctionType, but received .) isinstancertyperr#)rrkeyfuncs rregisterzNamespace.registerPs\#|,,  YTRUYY Y Y Y  ,||!## rc td|d|t|j}t |t |z}|jD]}||vr|j|}t j|}t |t |t |zkr7td|d|d||t |dkr[| D]F\}} ||j vr8td|d|d||nGt |dkrtt|} nGt |dkr(t|d tr tj} n tj} |j| S) z Get the matching function from the virtual namespace according to the actual arguments. Return None if it did not find any matching function. zget function: args=z , kwargs=zfn=z (key=z) is not satisfied and removed.rr)_loggerdebugsetr#keysleninspectgetfullargspecremoveitemsrnextiterr(floatrr r r) rrrsatisfied_function_keysnum_actual_argsfunc_keyrspecsarg_namevaluer*s rrz Namespace.get`s  CDCC6CCDDD"%d&7&<&<&>&>"?"?d))c&kk1)..00  H666"8,B*2..E5zzCIIF 333 M"MMHMMM(..x8886{{Q'-||~~OHeuz11 U"UUHUUU066x@@@ 2 & ' '1 , ,t34455CC YY!^^ 47E : :^(CC%C $$S)))r) r r r rr"r staticmethodrr,rrrrrr;sp JCCC$$\$  #*#*#*#*#*rrcfd}|S)zjoverload is the decorator that wraps the function and returns a callable object of type Function. c^t|Sr)rrr,)rr*s r decoratorzoverload..decorators%%%''00S999rr)r*rEs` roverloadrFs$ ::::: r) r5loggingenumrpaddle.base.log_helperrr INFOr0rrrrFrrrrKs$------ * gl H    4 ########2H*H*H*H*H*H*H*H*Vr