Ë
    øÿæiá  ã                   ó.   — d dl Z d dlmZ 	  G d„ d«      Zy)é    Nc                   óv   — e Zd ZdZdZg d¢Zg d¢Zeez   Zdd„Z	 d„ Z		 d„ Z
	 d	„ Z	 d
„ Z	 d„ Z	 d„ Z	 d„ Z	 d„ Zy)ÚPythonDispatcherÚ__test__Úfoo)ÚCPUÚAutogradCPUÚFPGAÚAutogradOtherÚXLAÚAutogradXLAÚLazyÚAutogradLazy)ÚCompositeExplicitAutogradÚAutogradÚCompositeImplicitAutogradNc                 óÄ   — t        j                  | j                  «       t        j                  d| j                  d«      | _        | j
                  j                  d«       y )NÚFRAGMENTÚ zfoo(Tensor x) -> Tensor)ÚCÚ_dispatch_check_invariantsÚnameÚ_dispatch_libraryÚ	namespaceÚrefÚdef_©Úselfs    úm/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torch/_python_dispatcher.pyÚ__init__zPythonDispatcher.__init__I   s?   € Ü	×$Ñ$ T§Y¡YÔ/Ü×&Ñ& z°4·>±>À2ÓFˆŒØ�‰�‰Ð/Õ0ó    c                 ó   — | j                   S ©N)Úsupported_keysr   s    r   ÚkeyszPythonDispatcher.keysS   s   € Ø×"Ñ"Ð"r    c                 ó*  — t        t        |«      «      t        |«      k7  rt        d|› d�«      ‚d|v rd|v rt        d«      ‚|D ]L  }|| j                  vrt        |› d| j                  › d�«      ‚| j                  j                  d|d|z   ¬	«       ŒN y )
Nz2Overridden is not allowed but found duplicates in Ú.r   r   z\Registration to both CompositeImplicitAutograd and CompositeExplicitAutograd is not allowed.z3 is not supported, please select a dispatch key in r   Úfn_)ÚdispatchÚdebug)ÚlenÚsetÚRuntimeErrorr#   r   Úimpl_t_t)r   ÚdispatchKeysÚkeys      r   ÚregisterzPythonDispatcher.register^   s´   € äŒs�<Ó Ó!¤S¨Ó%6Ò6ÜØDÀ\ÀNÐRSÐTóð ð
 (¨<Ñ7Ø+¨|Ñ;äØnóð ó  ˆCØ˜$×-Ñ-Ñ-Ü"Ø�eÐNÈt×ObÑObÐNcÐcdÐeóð ð �H‰H×Ñ˜e¨c¸À¹ÐÕEñ  r    c                 ó   — |d›d|› d�S )Nz<15Ú Ú
© )r   r/   Úkernels      r   Ú_format_linezPythonDispatcher._format_linew   s   € Ø�c�˜!˜F˜8 2Ð&Ð&r    c                 óF   — d|› d�}|| j                  dd«      z  }|dz  }|S )Nr3   r/   r5   z---------------------------
)r6   )r   ÚheaderÚss      r   Ú_format_headerzPythonDispatcher._format_header~   sA   € ðØ€ð 	ðˆð 	
ˆT×Ñ˜u hÓ/Ñ/ˆØ	Ð,Ñ,ˆØˆr    c                 ó^   — t        j                  | j                  › d| j                  › �«      S ©Nú::)r   Ú_dispatch_dumpr   r   r   s    r   ÚrawRegistrationsz!PythonDispatcher.rawRegistrations‹   s'   € Ü×Ñ 4§>¡>Ð"2°"°T·Y±Y°KÐ @ÓAÐAr    c                 ó^   — t        j                  | j                  › d| j                  › �«      S r<   )r   Ú_dispatch_dump_tabler   r   r   s    r   ÚrawDispatchTablez!PythonDispatcher.rawDispatchTable“   s'   € Ü×%Ñ%¨¯©Ð(8¸¸4¿9¹9¸+Ð&FÓGÐGr    c                 óV  ‡— | j                  d«      }| j                  «       }|j                  d«      }|D ]p  }|j                  d«      d   Št        ˆfd„| j                  D «       «      sŒ6|j                  d«      d   j                  d«      d   }|| j                  ‰|«      z  }Œr |S )	NzRegistered Kernelsr3   Ú:r   c              3   ó@   •K  — | ]  }‰j                  |«      –— Œ y ­wr"   )Ú
startswith)Ú.0ÚkÚfirsts     €r   Ú	<genexpr>z1PythonDispatcher.registrations.<locals>.<genexpr>¡   s   øè ø€ ÐDÑ0C¨1�5×#Ñ# A×&Ñ0Cùs   ƒr=   r2   é   )r:   r?   ÚsplitÚanyr#   r6   )r   ÚoutputÚstateÚstate_entriesÚliner5   rI   s         @r   ÚregistrationszPythonDispatcher.registrations›   s¡   ø€ Ø×$Ñ$Ð%9Ó:ˆØ×%Ñ%Ó'ˆØŸ™ DÓ)ˆÛ!ˆDØ—J‘J˜s“O AÑ&ˆEÜÓD°×0CÒ0CÓDÕDØŸ™ DÓ)¨!Ñ,×2Ñ2°3Ó7¸Ñ:�Ø˜$×+Ñ+¨E°6Ó:Ñ:‘ð	 "ð
 ˆr    c                 óZ  — | j                  d«      }| j                  «       }|j                  d«      }t        j                  d«      }|D ]^  }|j                  d«      d   }|| j
                  v sŒ&|j                  d|«      }|| j                  ||j                  d«      d   «      z  }Œ` |S )	NzComputed Dispatch Tabler3   z)registered at .*FallbackKernel\.cpp.*(\[)rD   r   Ú[z: rK   )r:   rB   rL   ÚreÚcompileÚruntime_keysÚsubr6   )r   rN   ÚtableÚtable_entriesÚregexrQ   rH   Úentrys           r   ÚdispatchTablezPythonDispatcher.dispatchTable¬   s¤   € Ø×$Ñ$Ð%>Ó?ˆØ×%Ñ%Ó'ˆØŸ™ DÓ)ˆÜ—
‘
ÐGÓHˆÛ!ˆDØ—
‘
˜3“ Ñ"ˆAØ�D×%Ñ%Ò%ØŸ	™	 # tÓ,�Ø˜$×+Ñ+¨A¨u¯{©{¸4Ó/@ÀÑ/CÓDÑD‘ð	 "ð
 ˆr    )ÚreturnN)Ú__name__Ú
__module__Ú__qualname__r   r   rW   Ú
alias_keysr#   r   r$   r0   r6   r:   r?   rB   rR   r]   r4   r    r   r   r   7   s~   „ Ø€IØ€Dò€Lò€Jð
 " JÑ.€Nó1ð
ò
#ðòFð*ò'ðòðò
Bðò
Hðò
	ðó
r    r   )rU   Útorch._CÚ_Cr   r   r4   r    r   Ú<module>re      s   ðã 	å ð-÷`ò r    