ó
    "Eñii	  ã                   ót  • % S SK Jr  S SKrS SKJr  S SKJr  S\\   4S jrS\\   4S jr	\\\S-        r
\\S'   \\   r\\S	'   \R                  /r\" 5        SS
\
S\S\\\R$                  \R&                  4   \\
\4   4   4S jj5       rS\R$                  S\4S jrS\\   S\R$                  S\4S jrg)é    )Ú	TypeAliasN)ÚTensor)Úno_gradÚreturnc                  óF   • SSS[         R                  R                  5       /$ )z:Return the device type list that supports foreach kernels.ÚcudaÚxpuÚmtia©ÚtorchÚ_CÚ_get_privateuse1_backend_name© ó    ÚW/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/utils/_foreach_utils.pyÚ&_get_foreach_kernels_supported_devicesr      s   € à�E˜6¤5§8¡8×#IÑ#IÓ#KÐLÐLr   c                  óL   • SSSSSS[         R                  R                  5       /$ )zEReturn the device type list that supports fused kernels in optimizer.Úmpsr   r	   ÚhpuÚcpur
   r   r   r   r   Ú$_get_fused_kernels_supported_devicesr      s0   € ð 	ØØØØØÜ�‰×.Ñ.Ó0ðð r   ÚTensorListListÚIndicesÚtensorlistlistÚwith_indicesc                 ó@   • [         R                  R                  X5      $ ©N)r   r   Ú"_group_tensors_by_device_and_dtype)r   r   s     r   r   r   *   s   € ô
 �8‰8×6Ñ6°~ÓTÐTr   Údevicec                 óŒ   • U R                   [        5       S/-   ;   =(       a#    [        R                  R	                  5       (       + $ )Nr   )Útyper   r   ÚjitÚis_scripting)r   s    r   Ú_device_has_foreach_supportr$   2   s6   € à�‰Ô>Ó@ÀEÀ7ÑJÑK÷ 	)Ü—	‘	×&Ñ&Ó(Ô(ðr   Útensorsc                 óJ   • [        U5      =(       a    [        S U  5       5      $ )Nc              3   óZ   #   • U  H!  oS L =(       d    [        U5      [        ;   v •  M#     g 7fr   )r!   Ú_foreach_supported_types)Ú.0Úts     r   Ú	<genexpr>Ú'_has_foreach_support.<locals>.<genexpr>:   s'   é € ð 7ÚBI¸QˆTˆ	×8”T˜!“WÔ 8Ñ8Ô8Â'ùs   ‚)+)r$   Úall)r%   r   s     r   Ú_has_foreach_supportr.   9   s(   € Ü& vÓ.÷ ´3ñ 7ÙBIó7ó 4ð r   )F)Útypingr   r   r   Útorch.autograd.grad_moder   ÚlistÚstrr   r   r   Ú__annotations__Úintr   r(   ÚboolÚdictÚtupler   Údtyper   r$   r.   r   r   r   Ú<module>r9      s   ðÞ ã Ý Ý ,ðM°°S±	ô Mð

¨d°3©iô 
ð !  f¨t¡mÑ!4Ñ5€�	Ó 5Ø˜#‘Y€ˆÓ Ø!ŸL™L˜>Ð ñ 	ƒð ñUØ"ðUàðUð 
ˆ%�—‘˜eŸk™kÐ)Ñ
*¨E°.À'Ð2IÑ,JÐ
JÑKôUó ðUð¨¯©ð ¸ô ð $ v¡,ð ¸¿¹ð Èõ r   