ó
    EñiF  ã                   óÈ   • S SK Jr  S SKJr  S SKJrJr  S SKJrJ	r	  S SK
Jr  SSKJr  \" \SS	9S
\S\\/\4   S\4S j5       r\" \SS	9S
\S\\/\4   S\4S j5       rSS/rg)é    )ÚCallable)ÚAny)Ú_fx_map_aggregateÚ_fx_map_arg)Úimmutable_dictÚimmutable_list)ÚNodeé   )Úsubstitute_in_graphT)Úcan_constant_fold_throughÚaÚfnÚreturnc                 ó$   ^• [        U U4S j5      $ )Nc                 óB   >• [        U [        5      (       a  T" U 5      $ U $ ©N)Ú
isinstancer	   )Úxr   s    €ÚW/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_dynamo/polyfills/fx.pyÚ<lambda>Úmap_arg.<locals>.<lambda>   s   ø€ ¬z¸!¼T×/BÑ/B¡b¨£eÐ&IÈÐ&Ió    ©Úmap_aggregate)r   r   s    `r   Úmap_argr      s   ø€ ä˜ÔIÓJÐJr   c                 ó¨  ^• [        U [        5      (       a:  U4S jU  5       n[        U S5      (       a  [        U 5      " U6 nU$ [        U5      nU$ [        U [        5      (       a(  [        U  Vs/ s H  n[        UT5      PM     sn5      nU$ [        U [        5      (       a<  [        U R                  5        VVs/ s H  u  pVU[        UT5      4PM     snn5      nU$ [        U [        5      (       aK  [        [        U R                  T5      [        U R                  T5      [        U R                  T5      5      nU$ T" U 5      nU$ s  snf s  snnf )Nc              3   ó<   >#   • U  H  n[        UT5      v •  M     g 7fr   r   )Ú.0Úelemr   s     €r   Ú	<genexpr>Ú map_aggregate.<locals>.<genexpr>   s   øé € Ð4²!¨$Œm˜D "×%Ð%²!ùs   ƒÚ_fields)r   ÚtupleÚhasattrÚtypeÚlistr   r   Údictr   ÚitemsÚsliceÚstartÚstopÚstep)r   r   ÚitÚresultr   ÚkÚvs    `     r   r   r      s+  ø€ ô �!”U×ÑÜ4±!Ó4ˆä!(¨¨I×!6Ñ!6”�a”˜"�ˆð €Mô =BÀ"»Iˆð €Mô 
�A”t×	Ñ	ÜÁQÓ GÂQ¸T¤¨t°RÖ!8ÁQÑ GÓHˆð €Mô 
�A”t×	Ñ	ÜÀqÇwÁwÄyÔ QÂy¹t¸q !¤]°1°bÓ%9Ó!:ÁyÒ QÓRˆð €Mô 
�A”u×	Ñ	ÜÜ˜!Ÿ'™' 2Ó&Ü˜!Ÿ&™& "Ó%Ü˜!Ÿ&™& "Ó%ó
ˆð €Mñ �A“ˆØ€Mùò !Hùã Qs   Á/E	Â;E
r   r   N)Úcollections.abcr   Útypingr   Útorch._Cr   r   Útorch.fx.immutable_collectionsr   r   Útorch.fx.noder	   Ú
decoratorsr   r   r   Ú__all__© r   r   Ú<module>r9      s¤   ðÝ $Ý ç 3ß IÝ å ,ñ �[¸DÑAðKˆsð K˜ $ ¨ Ñ-ð K°#ó Kó BðKñ Ð&À$ÑGð�Sð ˜h¨ u¨c zÑ2ð °só ó Hðð, Øð�r   