J/PhZdZddlZddlZddlZddlZddlZddlmZddlm Z dZ iZ dZ dZ dZejZGd d Zd Zd Zejeed ZdZdZdZdZGddejZdZeeje<GddejZGddZdS)z/ Serialization support for compiled functions. N) cloudpickle)irc|j|S)zM Global hook to rebuild a given class from its __reduce__ arguments. )_rebuild)clsargss T/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/core/serialize.py_rebuild_reductionr s 3< c||f} t|}n.#t$r!tj|xt|<}YnwxYw|S)zUsed by `numba_unpickle` from _helperlib.c Parameters ---------- address : int bytedata : bytes hashed : bytes Returns ------- obj : object unpickled object )_unpickled_memoKeyErrorrloads)addressbytedatahashedkeyobjs r _numba_unpicklersb F CAc" AAA%0%6x%@%@@sssA Js (??ct}tj5}||d}|||}dddn #1swxYwY|S)zISimilar to `pickle.dumps()`. Returns the serialized object in bytes. )protocolN) NumbaPicklerioBytesIOdumpgetvalue)rpicklerbufppickleds r dumpsr"4sG ! GC! $ $ $ s ,,..!!!!!!!!!!!!!!! Ns7AA"%A"c&tj|\}}}g}t|}|D]T}t|tjr#|t|?||U|t||fSN) rriter isinstancerValueappendnexttuple) static_excexc_argsexc static_argslocinfo real_args exc_args_iterargs r runtime_build_excinfo_structr3@s + 1* = =CgINNM"" c28 $ $ "   T-00 1 1 1 1   S ! ! ! ! y!!7 ++r c8eZdZdZdZdZdZedZdS)_CustomPickledaA wrapper for objects that must be pickled with `NumbaPickler`. Standard `pickle` will pick up the implementation registered via `copyreg`. This will spawn a `NumbaPickler` instance to serialize the data. `NumbaPickler` overrides the handling of this type so as not to spawn a new pickler for the object when it is already being pickled by a `NumbaPickler`. ctorstatesc"||_||_dSr$r6)selfr7r8s r __init__z_CustomPickled.__init__]s  r c6tj|j|jffSr$)r5rr7r8r:s r _reducez_CustomPickled._reduceas&DK(@@@r c|||Sr$)rr7r8s r rz_CustomPickled._rebuilddss4   r N) __name__ __module__ __qualname____doc__ __slots__r;r> classmethodrr@r r r5r5Psa!IAAA!![!!!r r5cFt|\}}t||S)zPstandard unpickling for `_CustomPickled`. Uses `NumbaPickler` to load. )rr5) serializedr7r8s r _unpickle__CustomPickledrIis% $$LD& $ ' ''r cLt|j|jf}t|ffS)zNstandard pickling for `_CustomPickled`. Uses `NumbaPickler` to dump. )r"r7r8rI)cprHs r _pickle__CustomPickledrLrs' +,,J #j] 22r c2tt||ffS)aFor customizing object serialization in `__reduce__`. Object states provided here are used as keyword arguments to the `._rebuild()` class method. Parameters ---------- states : dict Dictionary of object states to be serialized. Returns ------- result : tuple This tuple conforms to the return type requirement for `__reduce__`. )custom_rebuildr5)rr8s r custom_reducerOs N3779 99r c8|j|j}}|jdi|S)ziCustomized object deserialization. This function is referenced internally by `custom_reduce()`. r@)r7r8r)custom_pickledrr8s r rNrNs+ !%~'<C 3< ! !& ! !!r ctj5}t|} || ddddS#tj$rYddddSwxYw#1swxYwYdS)zCheck if *obj* can be serialized. Parameters ---------- obj : object Returns -------- can_serialize : bool NTF)rrrrpickle PicklingError)rfoutrs r is_serialiablerVs t$$  LL   #      s3A)AA)A&A)%A&&A))A-0A-cLtjdt|d)Nz Pickling of z is unsupported)rSrTtype)rs r _no_picklerYs%  Hd3iiHHH I IIr cDtj||S)z1This is called on a type to disable pickling )rdisabled_typesadd)typs r disable_picklingr^s ##C((( Jr c4eZdZeZ fdZxZS)rct||jvrt|t|Sr$)rXr[rYsuperreducer_override)r:r __class__s r rbzNumbaPickler.reducer_overrides: 99+ + + sOOOww'',,,r )rArBrCsetr[rb __classcell__rcs@r rrsHSUUN---------r rc*|Sr$)r>)rKs r _custom_reduce__custompickledrhs ::<<r c^eZdZdZejdZejdZdZ dZ dS) ReduceMixinznA mixin class for objects that should be reduced by the NumbaPickler instead of the standard pickler. ctr$NotImplementedErrorr=s r _reduce_stateszReduceMixin._reduce_states!!r c tr$rl)rkwargss r rzReduceMixin._rebuildror c|jSr$rfr=s r _reduce_classzReduceMixin._reduce_classs ~r cjt||Sr$)rOrsrnr=s r __reduce__zReduceMixin.__reduce__s*T//1143F3F3H3HIIIr N) rArBrCrDabcabstractmethodrnabstractclassmethodrrsrur@r r rjrjs  """ """  JJJJJr rjc:eZdZdZdZdZdZedZdS)PickleCallableByPathaWrap a callable object to be pickled by path to workaround limitation in pickling due to non-pickleable objects in function non-locals. Note: - Do not use this as a decorator. - Wrapped object must be a global that exist in its parent module and it can be imported by `from the_module import the_object`. Usage: >>> def my_fn(x): >>> ... >>> wrapped_fn = PickleCallableByPath(my_fn) >>> # refer to `wrapped_fn` instead of `my_fn` c||_dSr$_fn)r:fns r r;zPickleCallableByPath.__init__s r c|j|i|Sr$r|)r:rrqs r __call__zPickleCallableByPath.__call__stx((((r cZt|j|jj|jjffSr$)rXrr}rBrAr=s r ruzPickleCallableByPath.__reduce__s%Dzz"TX%8$(:K$MMMr cT|ttj||Sr$)getattrsysmodules)rmodnamefn_paths r rzPickleCallableByPath._rebuilds#s73;w/99:::r N) rArBrCrDr;rrurFrr@r r rzrzsi)))NNN;;[;;;r rz)rDrrvrcopyregrSnumbarllvmliterr r rr"r3rr5rIrLrOrNrVrYr^ CloudPicklerrrhdispatch_tableABCrjrzr@r r rs  ,    , , , !!!!!!!!2(((333~5666:::&"""*JJJ - - - - -;+ - - -/L N+JJJJJ#'JJJ2;;;;;;;;;;r