^j5NUddlZddlmZddlmZddlmZeZiae e eegdffe d<ia e e ejefe d<gaeee d<defd Zd edefd ZdZd e d eddfdZd e defdZdeddfdZddZdedee egdfde fdZd edede fdZddlmZeey)N)Callable)Any)Sourceindex_to_bytecode_constructor index_to_external_object_weakref keep_alivereturnc ttSN)boolrn/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/_dynamo/graph_bytecode_inputs.pyhas_user_objectsrs - ..robjc0tj||Sr )rappend)rs rstash_graph_created_objectrsc Jrindexvaluecftj|tj|t|<y)z;Update an entry in the external object registry at runtime.N)rrweakrefrefr)rrs rset_external_object_by_indexr$s$e.5kk%.@$U+rcv|tvr tdt|}| tdt|S)Nz4Index not registered in index_to_user_object_weakrefzUser object is no longer alive)rAssertionError)rrs rget_external_object_by_indexr*sC 44STT *5 1 3C {=>> +E 2 44rargsc tjtjt|Dcic]\}}|t j |c}}ycc}}wr )rclearupdate enumeraterr)riargs rstore_user_object_weakrefsr%3sC$**,$+++4T?;CGKK  ;;s A c|tjtjtjyr )rr rrr rrreset_user_object_trackingr';s&!'')$**,r example_value construct_fnctj|ttfdt< t j |t <S#t$r!}ddlm }|dd|dg|Yd}~Sd}~wwxYw)Nc|Sr r )cgr)rs rz/register_graph_created_object..Hsl5"6Mr unimplementedz7Failed to make weakref to graph-created external object user_object: 0Object does not allow us to make a weakref to itgb_typecontext explanationhintsfrom_exc) rrlenrrrr TypeErrorexcr0)r(r)er0rs ` @rregister_graph_created_objectr=As m$ - .E+M!%(  29++m2L(/ L   &M#M?3J    L  sA A=A88A=sourcectt}fdt|< tj|t|<|S#t $r!}ddlm}|dd|dg|Yd}~|Sd}~wwxYw)Nc|Sr r )r,r>s rr-z®ister_user_object..\s bjrr.r/z%Failed to make weakref to User Objectr1r2r3)r9rrrrr:r;r0)rr>rr<r0s ` rregister_user_objectrAYsx - .E+@!%(  29++e2D(/ L   &;#E7+J    L  s< A&A!!A&)"set_leaf_function_module_retriever)r N)rcollections.abcrtypingrtorch._dynamo.sourcer PyCodegenrdictint__annotations__r ReferenceTyperlistr rrCURRENT_STREAM_INDEXrrr%r'r=rA,torch._higher_order_ops.invoke_leaf_functionrBr rrrNs1$'  IKtC9+t2C)D$DEJJL $sG,A,A#,F'F"GL DI/$/CC AACADA 555cd &.Y/?/E&F0V, ##?@r