x-jM-ddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z ddlZddlmZddlmZmZmZmZmZddlmZddlmZmZmZdd lmZmZm Z m!Z!e rdd l"m#Z#gZ$eZ%d Z&e&a'ddZ(ej)ddZ*ej)ddZ+dZ,GddZ-dZ.dS)) annotationsN) TYPE_CHECKINGAnyCallable)Layer)convert_enumerate convert_len convert_print convert_range convert_zip)TranslatorLogger)StaticFunctionconvert_to_staticunwrap_decorators)TransformOptions is_builtinis_paddle_funcpatch_method_guard) ModuleTypectttttt t tg} ddl}| |n#t$rYnwxYw|S)z! Return builtin modules. rN) copy collectionsinspectloggingnumpyospdbresixappend ImportError)modulesr s f/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/jit/dy2static/convert_call_func.pybuiltin_modulesr%<sm      G  s       NsA AAr#list[ModuleType]cV|D]%}|tvrt|&dS)z0 Adds modules that ignore transcription N)BUILTIN_LIKELY_MODULESr!)r#modules r$add_ignore_moduler*Ws> 22 / / / " ) )& 1 1 122r)rreturnlist[Callable[..., Any]]cPtddfd|S)Nr,list[tuple[str, Any]]cP tj|S#t$rgcYSwxYwN)r getmembers Exception)r)s r$_try_get_membersz.get_module_functions.._try_get_memberses< %f-- -   III s  %%c|vrgS|g}|D][\}}t|r||tj|r||\|Sr1)addcallabler!rismoduleextend)r)results _member_namemember_get_module_functionsr4visiteds r$r=z3get_module_functions.._get_module_functionsks W  I F$4$4V$<$< > > L& 'v&&&'' >44V<<===r+)r,r/)set)r)r=r4r>s @@@r$get_module_functionsr@asXeeG        !  ( ((r+ str | Nonec\dd}t|dr|jdS||jS)N file_pathstrr,c.tjdd|S)Nz __init__.py$)rsub)rCs r$_remove_module_init_suffixz._remove_module_init_suffix|svor9555r+__file__)rCrDr,rD)hasattrrI)r)rHs r$get_module_defining_pathrKzsI6666 6: & &&/*At % %fo 6 66r+cfdtD}t|drQ|jjrE|jjt fd|Drt dd|dSdtD}|D]%}||urt dd|dS&t|rt dd|dSd S) zJ Checks whether the func is supported by dygraph to static graph. c6g|]}t|xSr1)rK).0r) module_paths r$ z"is_unsupported..s6 3F;; ;K H  H H Hr+__code__c3BK|]}|VdSr1) startswith)rNrO func_paths r$ z!is_unsupported..sE     - -      r+zMWhitelist: %s is part of built-in module and does not have to be transformed.Tc6g|]}t|D]}|S)r@)rNr)funcs r$rPz"is_unsupported..sG (00  r+zKWhitelist: %s is part of Paddle module and does not have to be transformed.F)r(rJrQ co_filenameanytranslator_loggerlogr)rYbuiltin_module_pathsbuiltin_functions builtin_fnrTrOs @@r$is_unsupportedrasb ,tZ  T]%> M-     3       ! !_    4,( :    ! !_    44 d Y    t 5r+ceZdZdZdZdS)StaticLayerWrapperc||_dSr1)layer)selfres r$__init__zStaticLayerWrapper.__init__s  r+ct|jdt|jj5|j|i|cdddS#1swxYwYdS)Nforward)rre convert_callri)rfargskwargss r$__call__zStaticLayerWrapper.__call__s  J < 0B#C#C   / /4:t.v.. / / / / / / / / / / / / / / / / / /s AA A N)__name__ __module__ __qualname__rgrmrXr+r$rcrcs2/////r+rcctdd|d}d}t|\}}tj|tjjstdd||St|drtSt|drtSt|drtSt|d rtSt|d rtSt|st|r|Stj|r>> # doctest: +SKIP('`paddle.jit.to_static` can not run in xdoctest') >>> import paddle >>> from paddle.jit.dy2static import Call >>> paddle.enable_static() >>> def dyfunc(x): ... if paddle.mean(x) < 0: ... x_v = x - 1 ... else: ... x_v = x + 1 ... return x_v ... >>> new_func = Call(dyfunc) >>> x = paddle.tensor.manipulation.fill_constant(shape=[3, 3], value=0, dtype='float64') >>> x_v = new_func(x) >>> exe = paddle.static.Executor(paddle.CPUPlace()) >>> out = exe.run(fetch_list=[x_v]) >>> print(out[0]) [[1. 1. 1.] [1. 1. 1.] [1. 1. 1.]] rz$Convert callable object: convert %s.NrVzG%s is not converted when it is decorated by 'paddle.jit.not_to_static'.lenziprange enumerateprintz *z Your function:`zk` doesn't support to transform to static function because it is a generator function, it will be run as-is. z__self__zr%s doesn't have to be transformed to static function because it has been transformed before, it will be run as-is. __class__riz Callable z# can not be transformed at present.zO%s doesn't have to be transformed to static function, and it will be run as-is.)-r\r]rrcheck_fn_need_transform ToStaticModeASTrr r r r r rarisgeneratorfunctionwarnrn isfunctionunwrapr? __globals__valuesr6 isinstancerisclass__dict__get staticmethodrgetattrAttributeErrorOSErrorismethodrJr7r{rrcrm TypeErrorNotImplementedError functoolspartial) rY func_selfconverted_call_number_of_stars _origfuncglobal_functionsfn call_funcs r$rjrjsJ!CTJJJIN %%GAt  3 +/    U     $$$  $ $$!  $  $>$// "4((  O# $]$-]]] ] O#  $       $O =J & &K2 " t,,I"uu +2244  %b)) $((,,,,N33 -b11EAr$((,,,,_R((!  t<9EE s=;EK K KK5 L L%$L%NN98N9)r#r&)r)rr,r-)r)rr,rA)/ __future__rrrrrrrrrtypingrrrr paddle.nnrconvert_operatorsr r r r r logging_utilsrprogram_translatorrrrutilsrrrrtypesr__all__r\r%r(r* lru_cacher@rKrarcrjrXr+r$rsk#"""""  ////////// ,+++++ !      %$&&0)**2222 ))))0 7777111h////////rrrrrr+