`ic'dZddlmZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z ddl mZdZe d e Zd7d8dZ d9d:dZe d;dZe dd!Zd?d#Zd@d%ZdAd'Zd7dBd*ZdCd/ZdDd3ZdEd5Zgd6ZdS)Fz.FFI registry to register function and objects.) annotationsN)AnyCallableLiteralSequenceTypeVaroverload)core)TypeInfoF_T)boundtype_key str | NonereturnCallable[[_T], _T]cd dttr d fd }|Sd fd }|Sttr |Std ) aRegister object type. Parameters ---------- type_key The type key of the node. It requires ``type_key`` to be registered already on the C++ side. If not specified, the class name will be used. Examples -------- The following code registers MyObject using type key "test.MyObject", if the type key is already registered on the C++ side. .. code-block:: python @tvm_ffi.register_object("test.MyObject") class MyObject(Object): pass clsr object_namestrrctj|}|tr|Std|tj||}t ||t |d||S)z+Register the object type with the FFI core.Nz"Cannot find object type index for )type_cls type_info__tvm_ffi_type_info__)r _object_type_key_to_index_SKIP_UNKNOWN_OBJECTS ValueError_register_object_by_index_add_class_attrssetattr)rr type_indexinfos d/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/tvm_ffi/registry.py _registerz"register_object.._register9sy3K@@  $  O+OOPP P-j#>>#6666,d333 c|SN)rr$rs r#_decorator_with_namez-register_object.._decorator_with_nameGs9S(++ +r%c&||jSr')__name__)rr$s r#_decorator_defaultz+register_object.._decorator_defaultLsycl+++r%Nz(type_key must be a string, type, or None)rr rrrr )rr rr ) isinstancertype TypeError)rr)r,r$s` @r#register_objectr0#s,    (C  $ , , , , , , ,$#,,,,,,!!(D!!,!!(+++ > ? ??r% func_namestr | Callable[..., Any]fCallable[..., Any] | Noneoverrideboolrctr }|jttst ddfd }| ||S|S) aRegister global function. Parameters ---------- func_name The function name f The function to be registered. override Whether override existing entry. Returns ------- fregister Register function if f is not specified. Examples -------- .. code-block:: python import tvm_ffi # we can use decorator to register a function @tvm_ffi.register_global_func("mytest.echo") def echo(x): return x # After registering, we can get the function by its name f = tvm_ffi.get_global_func("mytest.echo") assert f(1) == 1 # we can also directly register a function tvm_ffi.register_global_func("mytest.add_one", lambda x: x + 1) f = tvm_ffi.get_global_func("mytest.add_one") assert f(1) == 2 See Also -------- :py:func:`tvm_ffi.get_global_func` :py:func:`tvm_ffi.remove_global_func` zexpect string function namemyfCallable[..., Any]rrc0tj|S)z/Register the global function with the FFI core.)r _register_global_func)r8r1r5s r#registerz®ister_global_func..registers))S(CCCr%N)r8r9rr)callabler+r-rr)r1r3r5r<s` ` r#register_global_funcr>Vsf  J i % %86777DDDDDDD }x{{ Or%namer allow_missing Literal[True]core.Function | NonecdSr'r(r?r@s r#get_global_funcrEsVYVYr%Literal[False] core.FunctioncdSr'r(rDs r#rErEsX[X[r%c,tj||S)amGet a global function by name. Parameters ---------- name The name of the global function allow_missing Whether allow missing function or raise an error. Returns ------- func The function to be returned, ``None`` if function is missing. Examples -------- .. code-block:: python import tvm_ffi @tvm_ffi.register_global_func("demo.echo") def echo(x): return x f = tvm_ffi.get_global_func("demo.echo") assert f(123) == 123 See Also -------- :py:func:`tvm_ffi.register_global_func` )r _get_global_funcrDs r#rErEsH  } 5 55r% list[str]c~tdd}fdt|DS)zxGet list of global functions registered. Returns ------- names List of global functions names. z"ffi.FunctionListGlobalNamesFunctorc&g|] }|Sr(r().0i name_functors r# z*list_global_func_names..s! 6 6 6LLOO 6 6 6r%)rErange) num_namesrQs @r#list_global_func_namesrUsMI?#GHHJJL R  I 6 6 6 6U9%5%5 6 6 66r%Nonec6td|dS)aaRemove a global function by name. Parameters ---------- name The name of the global function. Examples -------- .. code-block:: python import tvm_ffi @tvm_ffi.register_global_func("my.temp") def temp(): return 42 assert tvm_ffi.get_global_func("my.temp", allow_missing=True) is not None tvm_ffi.remove_global_func("my.temp") assert tvm_ffi.get_global_func("my.temp", allow_missing=True) is None See Also -------- :py:func:`tvm_ffi.register_global_func` :py:func:`tvm_ffi.get_global_func` zffi.FunctionRemoveGlobalNrEr?s r#remove_global_funcrZs#<0O.//55555r%dict[str, Any]cZtjtd|pdS)aGet metadata (including type schema) for a global function. Parameters ---------- name The name of the global function. Returns ------- metadata A dictionary containing function metadata. The ``type_schema`` field encodes the callable signature. Examples -------- .. code-block:: python import tvm_ffi meta = tvm_ffi.get_global_func_metadata("testing.add_one") print(meta) See Also -------- :py:func:`tvm_ffi.get_global_func` Retrieve a callable for an existing global function. :py:func:`tvm_ffi.register_global_func` Register a Python callable as a global FFI function. zffi.GetGlobalFuncMetadataz{})jsonloadsrErYs r#get_global_func_metadatar_s-> :Bo&ABB4HHPD Q QQr% namespacetarget_module_namec|r|n|}|dr |dd}n|}tj|}tD]}}||s|t |dzd}|ddkrLt |}t|d|t|||~dS)aInitialize register ffi api functions into a given module. Parameters ---------- namespace The namespace of the source registry target_module_name The target module name if different from namespace Examples -------- A typical usage pattern is to create a _ffi_api.py file to register the functions under a given module. The following code populates all registered global functions prefixed with ``mypackage.`` into the current module, then we can call the function through ``_ffi_api.func_name(*args)`` which will call into the registered global function "mypackage.func_name". .. code-block:: python # _ffi_api.py import tvm_ffi tvm_ffi.init_ffi_api("mypackage", __name__) ztvm.Nr .rMr+) startswithsysmodulesrUlenfindrEr )r`raprefix target_moduler?fnamer3s r# init_ffi_apirms80BP++yF##122K 23M&(( ) )v&&  S[[1_&&' ::c??b  D ! !:u%%% ua(((( ) )r%rr.rr c|jD]=}|j}t||s$t||||>d}|jD]G}|j}|dkrd}d}t||s$t||||Hd|jvrR|r t|dt|dn0t|tj st|dt|S)NF __ffi_init____c_ffi_init__T__init__) fieldsr?hasattrr as_propertymethods as_callable__dict__getattr issubclassr PyNativeObject__init__invalid)rrfieldr? has_c_initmethods r#rrKs!AAzx&& A HdE$5$5h$?$? @ @ @J#BB{ > ! !#DJx&& B HdF$6$6x$@$@ A A A***  ; Hj'(N*K*K L L L LHd&9:: ; Hj/ : : : Or%selfargskwargsc td)Nz5The __init__ method of this class is not implemented.) RuntimeError)rrrs r#r{r{`s N O OOr% Sequence[str]c0tdS)zGet the list of valid type keys registered to TVM-FFI. Returns ------- type_keys List of valid type keys. zffi.GetRegisteredTypeKeysrXr(r%r#get_registered_type_keysrds 8?6 7 7 9 99r%)rEr_rrmrUr>r0rZr')rrrr)NF)r1r2r3r4r5r6rr)r?rr@rArrB)F)r?rr@rFrrG)r?rr@r6rrB)rrK)r?rrrV)r?rrr[)r`rrarrrV)rr.rr rr.)rrrrrrrrV)rr)__doc__ __future__rr]rftypingrrrrrr r r rr.r r0r>rErUrZr_rmrr{r__all__r(r%r#rs"54"""""" FFFFFFFFFFFFFFFF WT0@0@0@0@0@j$(@@@@@F YYY Y [[[[ [$6$6$6$6$6N 7 7 7 76666BRRRRD/)/)/)/)/)d*PPPP : : : :   r%