+
    –°³jÒ$  ã                   ó.  € ^ RI t ^ RIHt ^ RIHt ^ RIHt ^ RIt^ RIHt ^ RI	H
t
HtHt ]],          t]]]R3,          ,          tR R	 ltR
 R ltR R ltR R ltRR R lltR R ltR R ltR R lt]! R]R7      RR R ll4       tR R R lltR# )!é    N©ÚCallable)ÚAny)Ú
deprecated)ÚTensor)Ú_broadcast_to_and_flattenÚtree_flattenÚtree_unflatten.c                ó`   € V ^8„  d   QhR\         \        R,          ,          R\         R\        /# )é   Úflat_in_dimsNÚ	flat_argsÚreturn)ÚlistÚint)Úformats   "ÚV/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_vmap_internals.pyÚ__annotate__r      s/   € ÷ ñ Ü”s˜T•zÕ"ðäðô 	ñó    c                 ó8  a€ \        W4       UUu. uF  w  r#Vf   K  VP                  V4      NK  	  uppoS'       dP   \        ;QJ d    V3R lS 4       F  '       g   K   RM	  RM! V3R lS 4       4      '       d   \        RS R24      hS^ ,          # u uppi )Nc              3   ó:   <"  € T F  qS^ ,          8g  x € K  	  R# 5i)r   N© )Ú.0ÚsizeÚbatch_sizess   & €r   Ú	<genexpr>Ú/_validate_and_get_batch_size.<locals>.<genexpr>   s   øé € ÐJ¹k°d ;¨q¥>Ö1»kùs   ƒTFzTvmap: Expected all tensors to have the same size in the mapped dimension, got sizes z for the mapped dimension)Úzipr   ÚanyÚ
ValueError)r   r   Úin_dimÚargr   s   &&  @r   Ú_validate_and_get_batch_sizer#      s�   ø€ ô ˜|Ô7ôá7‰KˆFØô 	ˆ�‰�ÖÙ7ò€K÷
 —s“sÔJ¹kÓJ—s—s’sÔJ¹kÓJ×JÒJÜð$Ø$/ =Ð0IðKó
ð 	
ð �q�>Ðùós
   �B�Bc                ó`   € V ^8„  d   QhR\         \        \         R3,          ,          R\        /# )r   Úbatched_outputs.r   )r   Útupler   )r   s   "r   r   r   "   s(   € ÷ ñ ¤&¬5´¸°Õ+=Õ"=ð Ä#ñ r   c                 óH   € \        V \        4      '       d   \        V 4      # ^# )é   )Ú
isinstancer&   Úlen)r%   s   &r   Ú_num_outputsr+   "   s   € Ü�/¤5×)Ò)Ü�?Ó#Ð#Ùr   c                ób   € V ^8„  d   QhR\         R\        R\        . \        3,          R\        /# )r   ÚvalueÚnum_elementsÚerror_message_lambdar   )r   r   r   Ústrr&   )r   s   "r   r   r   *   s9   € ÷ 	ñ 	Üð	äð	ô # 2¤s 7Õ+ð	ô ñ		r   c                 ó†   € \        V \        4      '       g   V 3V,          # \        V 4      V8w  d   \        V! 4       4      hV # ©N)r)   r&   r*   r    )r-   r.   r/   s   &&&r   Ú	_as_tupler3   *   s>   € ô
 �eœU×#Ò#Øˆx˜,Õ&Ð&Ü
ˆ5ƒz�\Ô!ÜÑ-Ó/Ó0Ð0Ø€Lr   c                óv   € V ^8„  d   QhR\         R\        R\        R\        R\        \        \        3,          /# )r   Úin_dimsÚargsÚ
vmap_levelÚfuncr   )Ú	in_dims_tr&   r   r   )r   s   "r   r   r   8   sJ   € ÷ 9Añ 9AÜð9Aä
ð9Aô ð9Aô ð	9Aô
 Œ5”#ˆ:Õñ9Ar   c                 óF  € \        V \        4      '       g>   \        V \        4      '       g(   \        R \	        V4       RV  R\        V 4       R24      h\        V4      ^ 8X  d   \        R \	        V4       R24      h\        V4      w  rE\        W4      pVf2   \        R \	        V4       RV  R\        V 4      ^,           RV R2	4      h\        WF4       Fô  w  rx\        V\        4      '       g#   Ve   \        R \	        V4       RV  RV R24      h\        V\        4      '       dA   \        V\        4      '       g+   \        R \	        V4       RV  RV R	\        V4       R
2	4      hVf   K™  V^ 8  g   W‡P                  4       8¼  g   K¶  \        R \	        V4       RV  RV RVP                  4        RVP                  4        R24      h	  \        Wd4      p	\        Wd4       UUu. uF"  w  r‡Vf   TM\        P                  ! WxV4      NK$  	  p
pp\        W¥4      V	3# u uppi )úvmap(z
, in_dims=zv, ...)(<inputs>): expected `in_dims` to be int or a (potentially nested) tuple matching the structure of inputs, got: Ú.z�)(<inputs>): got no inputs. Maybe you forgot to add inputs, or you are trying to vmap over a function with no inputs. The latter is unsupported.zb, ...)(<inputs>): in_dims is not compatible with the structure of `inputs`. in_dims has structure z but inputs has structure z, ...)(<inputs>): Got in_dim=zE for an input but in_dim must be either an integer dimension or None.z' for an input but the input is of type zT. We cannot vmap over non-Tensor arguments, please use None as the respective in_dimz> for some input, but that input is a Tensor of dimensionality z- so expected in_dim to satisfy 0 <= in_dim < )r)   r   r&   r    Ú	_get_nameÚtyper*   r	   r   r   r   Údimr#   ÚtorchÚ_add_batch_dimr
   )r5   r6   r7   r8   r   Ú	args_specr   r"   r!   Ú
batch_sizeÚbatched_inputss   &&&&       r   Ú_create_batched_inputsrE   8   sc  € ô �gœs×#Ò#¬J°wÄ×,FÒ,FÜØ”I˜d“OÐ$ J¨w¨ið 86ä6:¸7³m°_ÀAðGó
ð 	
ô
 ˆ4ƒy�A„~ÜØ”I˜d“OÐ$ð %)ð *ó
ð 	
ô (¨Ó-Ñ€IÜ,¨WÓ@€LØÒÜØ”I˜d“OÐ$ J¨w¨ið 8%ä%1°'Ó%:¸1Õ%=Ð$>ð ?Ø&˜K qð*ó
ð 	
ô ˜9Ö3‰ˆÜ˜&¤#×&Ò&¨6Ò+=ÜØœ	 $›Ð(¨
°7°)ð <Ø$˜Xð &0ð1óð ô
 �fœc×"Ò"¬:°c¼6×+BÒ+BÜØœ	 $›Ð(¨
°7°)ð <Ø$˜XÐ%LÜ˜“9�+ð ;ð<óð ð Ô 6¨A¤:°¿7¹7»9Ö1DÜØœ	 $›Ð(¨
°7°)ð <Ø$˜Xð &%Ø%(§W¡W£Y Kð 0!Ø!$§¡£ ¨1ð.óð ñ 4ô, .¨lÓF€Jô ˜|Ô7ôá7‰KˆFð Š~‰¤5×#7Ò#7¸ÀZÓ#PÒPÙ7ð ñ ô ˜.Ó4°jÐ@Ð@ùó	s   Ç%(Hc                óœ   € V ^8„  d   QhR\         \        \         R3,          ,          R\        R\        R\        R\        R\
        R\        /# )	r   r%   .Úout_dimsr7   rC   r8   Úallow_none_pass_throughr   )r   r&   Ú
out_dims_tr   r   Úbool)r   s   "r   r   r   u   s\   € ÷ #
ñ #
Üœe¤F¨C KÕ0Õ0ð#
äð#
ô ð#
ô ð	#
ô
 ð#
ô "ð#
ô ñ#
r   c                 óà  aaaaa€ \        V 4      o\        SSVVV3R  l4      p\        V \        4      '       d#   V^ ,          p\        P
                  ! V SSV4      # V'       dG   \        ;QJ d!    . VV3R l\        W4       4       F  NK  	  5# ! VV3R l\        W4       4       4      # \        ;QJ d!    . VV3R l\        W4       4       F  NK  	  5# ! VV3R l\        W4       4       4      # )c            
      óF   <€ R \        S 4       RS RS R\        S 4       R2	# )r;   ú, ..., out_dims=z0): `out_dims` must have one dim per output (got z outputs) of r<   )r=   )r8   Únum_outputsrG   s   €€€r   Ú<lambda>Ú!_unwrap_batched.<locals>.<lambda>�   s4   ø€ �%œ	 $›Ð(Ð(8¸¸
ð C(Ø(3 }°MÄ)ÈDÃ/ÐARÐRSñUr   c              3   óf   <"  € T F&  w  rVe   \         P                  ! VSSV4      MR x € K(  	  R # 5ir2   ©r@   Ú_remove_batch_dim©r   ÚoutÚout_dimrC   r7   s   &  €€r   r   Ú"_unwrap_batched.<locals>.<genexpr>Œ   s@   øé € ð 
ñ !H‘�ð ’?ô ×'Ò'¨¨Z¸ÀWÔMàôó !Hùs   ƒ.1c              3   óZ   <"  € T F   w  r\         P                  ! VSSV4      x € K"  	  R # 5ir2   rR   rT   s   &  €€r   r   rW   •   s.   øé € ð 
á G‘�ô ×#Ò# C¨°ZÀ×IÐIÛ Gùs   ƒ(+)r+   r3   r)   r   r@   rS   r&   r   )	r%   rG   r7   rC   r8   rH   Úout_dims_as_tuplerV   rN   s	   &ffff&  @r   Ú_unwrap_batchedrZ   u   sØ   ü€ ô ˜Ó/€KÜ!ØØö	UóÐô �/¤6×*Ò*Ø# AÕ&ˆÜ×&Ò& ¸
ÀJÐPWÓXÐXßßŒuõ 
ô !$ OÔ Gó
�uð 	
ˆuõ 
ô !$ OÔ Gó
ó 
ð 	
÷ Œuõ 
ä # OÔ Gó
�uð 	
ˆuõ 
ä # OÔ Gó
ó 
ð 	
r   c                ó4   € V ^8„  d   QhR\         R\        RR/# )r   Úoutputsr8   r   N)r   r   )r   s   "r   r   r   Ÿ   s!   € ÷ 
ñ 
œsð 
¬(ð 
°tñ 
r   c                 ó|  € \        V \        4      '       d   R # \        V \        4      '       g1   \        R\	        V4       R\	        V4       R\        V 4       R24      h\        V 4       FN  w  r#\        V\        4      '       d   K  \        R\	        V4       R\	        V4       R\        V4       RV R2	4      h	  R # )Nr;   z	, ...): `z%` must only return Tensors, got type z as the return.z for return r<   )r)   r   r&   r    r=   r>   Ú	enumerate)r\   r8   ÚidxÚoutputs   &&  r   Ú_validate_outputsra   Ÿ   s¼   € Ü�'œ6×"Ò"ÙÜ�gœu×%Ò%ÜØ”I˜d“OÐ$ I¬i¸«oÐ->ð ?!Ü!% g£ ¨ð@ó
ð 	
ô ! Ö)‰ˆÜ�fœf×%Ò%ÙÜØ”I˜d“OÐ$ I¬i¸«oÐ->ð ?!Ü!% f£ ¨l¸3¸%¸qðBó
ð 	
ó *r   c                ó4   € V ^8„  d   QhR\         R\        RR/# )r   rG   r8   r   N)rI   r   )r   s   "r   r   r   °   s"   € ÷ 

ñ 

´*ð 

ÄHð 

ÐQUñ 

r   c                 ó  € \        V \        4      '       d   R # \        V \        4      '       d;   \        ;QJ d    R V  4       F  '       d   K   RM	  RM! R V  4       4      '       g   \	        R\        V4       RV  R24      hR # )Nc              3   óB   "  € T F  p\        V\        4      x € K  	  R # 5ir2   )r)   r   )r   rV   s   & r   r   Ú6_check_out_dims_is_int_or_int_tuple.<locals>.<genexpr>³   s   é € ð 2Ù08 WŒ
�7œC× Ð ³ùs   ‚FTr;   rM   zu): `out_dims` must be an int or a tuple of int representing where in the outputs the vmapped dimension should appear.)r)   r   r&   Úallr    r=   )rG   r8   s   &&r   Ú#_check_out_dims_is_int_or_int_tuplerg   °   s~   € Ü�(œC× Ò ÙÜ�h¤×&Ò&¯c«cñ 2Ù08ó2¯c¯cªcñ 2Ù08ó2÷ /ò /ô Ø”I˜d“OÐ$Ð$4°X°Jð ?/ð 0ó
ð 	
ñ/r   c                ó$   € V ^8„  d   QhR\         /# )r   r8   r   )r   s   "r   r   r   ½   s   € ÷ ñ ”Hñ r   c                 óT   € \        V R 4      '       d   V P                  # \        V 4      # )Ú__name__)Úhasattrrj   Úrepr)r8   s   &r   r=   r=   ½   s%   € Üˆt�Z× Ò Ø�}‰}Ðô
 �‹:Ðr   z@Please use `torch.vmap` instead of `torch._vmap_internals.vmap`.)Úcategoryc                óH   € V ^8„  d   QhR\         R\        R\        R\         /# )r   r8   r5   rG   r   )r   r9   rI   )r   s   "r   r   r   Î   s(   € ÷ *ñ *Œxð *¤)ð *¼:ð *Ìhñ *r   c                ó   € \        WV4      # )z,
Please use torch.vmap instead of this API.
)Ú_vmap)r8   r5   rG   s   &&&r   Úvmaprq   Ê   s   € ô � Ó)Ð)r   c          
      óT   € V ^8„  d   QhR\         R\        R\        R\        R\         /# )r   r8   r5   rG   rH   r   )r   r9   rI   rJ   )r   s   "r   r   r   Ö   s:   € ÷  ñ  Ü
ð äð ô ð ô "ð	 ô
 ñ r   c                 óR   a aaa€ \         P                  ! S 4      VV VV3R  l4       pV# )c            	      óP  <€ \        SS4       \        P                  P                  4       p \	        SWS4      w  r#S! V!  pS'       g   \        VS4       \        VSVVSSR 7      \        P                  P                  4        #   \        P                  P                  4        i ; i))rH   )rg   r@   Ú_CÚ_vmapmode_increment_nestingrE   ra   rZ   Ú_vmapmode_decrement_nesting)	r6   r7   rD   rC   r%   rH   r8   r5   rG   s	   *    €€€€r   ÚwrappedÚ_vmap.<locals>.wrappedà   sŽ   ø€ ä+¨H°dÔ;Ü—X‘X×9Ñ9Ó;ˆ
ð	3Ü)?Ø˜¨4ó*Ñ&ˆNñ # NÑ3ˆOß*Ü! /°4Ô8Ü"ØØØØØØ(?ôô �H‰H×0Ñ0Õ2øŒE�H‰H×0Ñ0Õ2ús   ­9B Â B%)Ú	functoolsÚwraps)r8   r5   rG   rH   rx   s   ffff r   rp   rp   Ö   s'   û€ ô ‡_‚_�TÓ÷3ó ð3ð* €Nr   )F)r   r   )r   r   F)rz   Úcollections.abcr   Útypingr   Útyping_extensionsr   r@   r   Útorch.utils._pytreer   r	   r
   r   r&   r9   rI   r#   r+   r3   rE   rZ   ra   rg   r=   ÚFutureWarningrq   rp   r   r   r   Ú<module>r�      sŽ   ðã Ý $Ý Ý (ã Ý ß WÑ Wð �%�K€	Ø�5˜˜c˜•?Õ"€
õõ"õ	õ9A÷z#
õT
õ"

õñ ØFØôö*ó	ð*÷ ñ  r   