9jNUddlZddlmZddlmZddlmZeZiae 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)boolrc/media/conek/DATA/Code/OCR/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+rcf|tvsJdt|}|Jdt|S)Nz4Index not registered in index_to_user_object_weakrefzUser object is no longer alive)r)rrs rget_external_object_by_indexr*sH 4 4> 4 +5 1 3C ?<<z/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 TypeErrorexcr/)r'r(er/rs ` @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 Objectr0r1r2)r8rrrrr9r:r/)rr=rr;r/s ` rregister_user_objectr@Ysx - .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<r@,torch._higher_order_ops.invoke_leaf_functionrAr rrrMs1$'  IKtC9+t2C)D$DEJJL $sG,A,A#,F'F"GL DI/$/CC AACADA 555cd &.Y/?/E&F0V, ##?@r