CjpUddlZddlmZddlmZddlmZeZiae e eegdffe d<ia e e ej efe d<gaeee d<defd Zd edefd ZdZd e d eddfdZd e defdZdeddfdZddZdedee egdfde fdZd edede fdZddlmZeedS)N)Callable)Any)Sourceindex_to_bytecode_constructor index_to_external_object_weakref keep_alivereturnc*ttSN)boolrQC:\Projects\DSCons\.venv\Lib\site-packages\torch\_dynamo\graph_bytecode_inputs.pyhas_user_objectsrs - . ..robjc:t||Sr )rappend)rs rstash_graph_created_objectrsc Jrindexvaluecrt|tj|t|<dS)z;Update an entry in the external object registry at runtime.N)rrweakrefrefr)rrs rset_external_object_by_indexr$s1e.5k%.@.@$U+++rc|tvrtdt|}|tdt|S)Nz4Index not registered in index_to_user_object_weakrefzUser object is no longer alive)rAssertionError)rrs rget_external_object_by_indexr*sR 444STTT *5 1 3 3C {=>>> +E 2 4 44rargscttdt|DdS)Nc>i|]\}}|tj|Sr )rr).0iargs r z.store_user_object_weakrefs..7s(;;;CGK  ;;;r)rclearupdate enumerate)rs rstore_user_object_weakrefsr(3sM$**,,,$++;;9T??;;;rctttdSr )rr%rrr rrreset_user_object_trackingr*;s>!'')))$**,,,r example_value construct_fnct|ttfdt< t j|t <n0#t$r#}ddlm }|dd|dg|Yd}~nd}~wwxYwS)Nc|Sr r )cgr,rs rz/register_graph_created_object..Hsll5"6M6Mr 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 TypeErrorexcr3)r+r,er3rs ` @rregister_graph_created_objectr@As m$$$ - . .E+M+M+M+M+M!%(  29+m2L2L(//     &&&&&& M3M33J             LsA B 'BB sourcectt}fdt|< tj|t|<n0#t $r#}ddlm}|dd|dg|Yd}~nd}~wwxYw|S)Nc|Sr r )r/rAs rr0z®ister_user_object..\sbbjjrr1r2z%Failed to make weakref to User Objectr4r5r6)r<rrrrr=r>r3)rrArr?r3s ` rregister_user_objectrDYs - . .E+@+@+@+@!%(  29+e2D2D(//     &&&&&& ;+E++J             LsA A. A))A.)"set_leaf_function_module_retriever)r N)rcollections.abcrtypingrtorch._dynamo.sourcer PyCodegenrdictint__annotations__r ReferenceTyperlistr rrCURRENT_STREAM_INDEXrrr(r*r@rD,torch._higher_order_ops.invoke_leaf_functionrEr rrrQs$$$$$$''''''  IKtC9+t2C)D$DEJJJJL $sG,A#,F'F"GLLL DI/$////CC AACADAAAA 555555cd &.Y/?/E&F0V, #"#?@@@@@r