ó
    Eñi  ã                   óD  • % S SK r S SKJr  S SKJr  S SKJr  \r0 q\	\
\\/S4   4   \S'   0 q\	\
\ R                  \   4   \S'   / q\\   \S'   S\4S	 jrS
\S\4S jrS\
S\4S jrS\SS4S jrSS jrS\S\\
\/S4   S\
4S jrS\S\S\
4S jrS SKJr  \" \5        g)é    N)ÚCallable)ÚAny)ÚSourceÚindex_to_bytecode_constructorÚ index_to_external_object_weakrefÚ
keep_aliveÚreturnc                  ó    • [        [        5      $ ©N)Úboolr   © ó    Ú`/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_dynamo/graph_bytecode_inputs.pyÚhas_user_objectsr      s   € ÜÔ-Ó.Ð.r   Úobjc                 ó0   • [         R                  U 5        U $ r   )r   Úappend)r   s    r   Ústash_graph_created_objectr      s   € Ü×Ñ�cÔØ€Jr   Úindexc                 óp   • U [         ;   d   S5       e[         U    " 5       nUc   S5       e[         U    " 5       $ )Nz4Index not registered in index_to_user_object_weakrefzUser object is no longer alive)r   )r   r   s     r   Úget_external_object_by_indexr   !   sH   € ØÔ4Ó4ð Ø>óÐ4ô +¨5Ò
1Ó
3€CØ‰?Ð<Ð<Ó<ˆ?Ü+¨EÒ2Ó4Ð4r   Úargsc            
      óÈ   • [         R                  5         [         R                  [        U 5       VVs0 s H  u  pU[        R
                  " U5      _M     snn5        g s  snnf r   )r   ÚclearÚupdateÚ	enumerateÚweakrefÚref)r   ÚiÚargs      r   Ústore_user_object_weakrefsr!   *   sE   € ä$×*Ñ*Ô,Ü$×+Ñ+Ü+4°T¬?Ô;ª?¡ ˆŒG�KŠK˜ÓÒ	©?Ò;õùÛ;s   ²#A
c                  ó|   • [         R                  5         [        R                  5         [        R                  5         g r   )r   r   r   r   r   r   r   Úreset_user_object_trackingr#   2   s&   € Ü!×'Ñ'Ô)Ü$×*Ñ*Ô,Ü×ÑÕr   Úexample_valueÚconstruct_fnc           	      ó   ^^• [         R                  U 5        [        [        5      mUU4S j[        T'    [        R
                  " U 5      [        T'   T$ ! [         a  nSSKJ	n  U" SSU  3S/ US9   S nAT$ S nAff = f)Nc                 ó   >• T" TU 5      $ r   r   )Úcgr%   r   s    €€r   Ú<lambda>Ú/register_graph_created_object.<locals>.<lambda>?   s   ø€ ±lÀ5È"Ô6Mr   é   ©Úunimplementedz7Failed to make weakref to graph-created external objectúuser_object: ú0Object does not allow us to make a weakref to it©Úgb_typeÚcontextÚexplanationÚhintsÚfrom_exc)
r   r   Úlenr   r   r   r   Ú	TypeErrorÚexcr-   )r$   r%   Úer-   r   s    `  @r   Úregister_graph_created_objectr:   8   sƒ   ù€ ô
 ×Ñ�mÔ$ÜÔ-Ó.€EÝ+MÔ! %Ñ(ð
Ü29·+²+¸mÓ2LÔ(¨Ñ/ð €Løô ó 	
Ý&áØMØ# M ?Ð3ØJØØö	
ð €Lûð	
ús   ¶A Á
A=ÁA8Á8A=ÚvalueÚsourcec           	      óÒ   ^• [        [        5      nU4S j[        U'    [        R                  " U 5      [        U'   U$ ! [
         a  nSSKJn  U" SSU  3S/ US9   S nAU$ S nAff = f)Nc                 ó   >• U " T5      $ r   r   )r(   r<   s    €r   r)   Ú&register_user_object.<locals>.<lambda>S   s	   ø€ ±b¸´jr   r+   r,   z%Failed to make weakref to User Objectr.   r/   r0   )r6   r   r   r   r   r7   r8   r-   )r;   r<   r   r9   r-   s    `   r   Úregister_user_objectr@   P   ss   ø€ äÔ-Ó.€EÜ+@Ô! %Ñ(ð
Ü29·+²+¸eÓ2DÔ(¨Ñ/ð €Løô ó 	
Ý&áØ;Ø# E 7Ð+ØJØØö	
ð €Lûð	
ús   Ÿ> ¾
A&ÁA!Á!A&)Ú"set_leaf_function_module_retriever)r	   N)r   Úcollections.abcr   Útypingr   Útorch._dynamo.sourcer   Ú	PyCodegenr   ÚdictÚintÚ__annotations__r   ÚReferenceTyper   Úlistr   r   r   r   r!   r#   r:   r@   Ú,torch._higher_order_ops.invoke_leaf_functionrA   r   r   r   Ú<module>rL      s  ðÜ Ý $Ý å 'ð €	ð IKÐ ˜t C¨°9°+¸tÐ2CÑ)DÐ$DÑEÓ JàJLÐ   $ s¨G×,AÑ,AÀ#Ñ,FÐ'FÑ"GÓ Là€
ˆD�‰IÓ ð/˜$ô /ð Cð ¨Cô ð
5¨ð 5°ô 5ð cð ¨dô ôðØðØ&.°°YÐ/?ÀÐ/EÑ&Fðàôð0 ð ¨Vð ¸ô õ,ñ
 #Ð#?Õ @r   