Ë
    ùÿæiÒ8  ã                  óH   — d dl mZ d dlmZmZ dd„Z ed«      Zd	d„Zd
d„Z	y)é    )Úannotations)ÚNativeFunctionsGroupÚNativeFunctionsViewGroupc                óÚ   — t        | t        «      r=t        | j                  j                  j
                  j
                  j                  «      S t        | j                  j                  «      S ©N)	Ú
isinstancer   ÚstrÚ
functionalÚfuncÚnameÚbaseÚviewÚ	root_name)Úgs    ús/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torchgen/static_runtime/config.pyÚfunc_name_base_strr      sI   € Ü�!Ô)Ô*Ü�1—<‘<×$Ñ$×)Ñ)×.Ñ.×3Ñ3Ó4Ð4ä�1—6‘6×#Ñ#Ó$Ð$ó    )'ÚabsÚaddÚaddmmÚallÚanyÚargminÚbmmÚclampÚ	clamp_minÚcumsumÚdivÚfmodÚindex_selectÚ
leaky_reluÚlinearÚlogÚmatmulÚmulÚnarrow_copyÚnonzeroÚpowÚ	remainderÚsigmoidÚsignÚsubÚtanhÚdetachÚ	expand_asÚflattenÚnarrowÚ
reshape_asÚselectÚsliceÚsoftmaxÚsplitÚsqueezeÚ	transposer   Úwherec                ó(   — t        | «      }|t        v S r   )r   Úis_hand_written_ops_)r   Ú	name_bases     r   Úis_hand_writtenr=   :   s   € Ü" 1Ó%€IØÔ,Ð,Ð,r   c                óL
  — |dvrt        d|› �«      ‚|dk(  r%|dk(  rd| d<   d| d<   d| d	<   y d
| d<   d| d<   d| d	<   y |dk(  r|dk(  rd| d<   d| d<   y d
| d<   d| d<   y |dk(  r|dk(  rd| d<   y d
| d<   y |dk(  r|dk(  rd| d<   d| d<   y d| d<   d| d<   y |dk(  r|dk(  rd| d<   y d| d<   y |dk(  r|dk(  rd| d<   y d| d<   y |dk(  r|dk(  rd| d<   y d| d<   y |dk(  r|dk(  rd| d <   y d
| d <   y |d!k(  r|dk(  rd| d <   y d
| d <   y |d"k(  r|dk(  rd| d#<   d$| d%<   y d| d#<   d$| d%<   y |d&k(  r|dk(  rd| d#<   d$| d%<   y d| d#<   d$| d%<   y |d'k(  r%|dk(  rd| d<   d(| d)<   d| d*<   y d
| d<   d+| d)<   d| d*<   y |d,k(  r|dk(  rd| d<   d-| d)<   y d
| d<   d.| d)<   y |d/k(  r%|dk(  rd| d<   d(| d)<   d| d*<   y d
| d<   d+| d)<   d| d*<   y |d0k(  r%|dk(  rd1| d<   d2| d)<   d| d*<   y d3| d<   d4| d)<   d| d*<   y |d5v rd6| d7<   y |d8k(  r|dk(  rd1| d<   d9| d:<   y d3| d<   d9| d:<   y |d;k(  r%|dk(  rd<| d<   d=| d><   d<| d<   y d?| d<   d@| d><   d?| d<   y |dAk(  r|dk(  rdB| d<   y dC| d<   y |dDk(  r%|dk(  rdE| dF<   dE| d<   dG| d<   y dH| dF<   dH| d<   dI| d<   y |dJk(  r%|dk(  rdK| dF<   dK| d<   dL| d<   y dM| dF<   dM| d<   dN| d<   y |dOk(  r|dk(  rdP| d<   dQ| d<   y dR| d<   dS| d<   y |dTk(  r|dk(  rdU| d<   dV| d<   y dW| d<   dX| d<   y |dYk(  r/|dk(  rdZ| d<   d[| d\<   d]| d<   d^| d_<   y d`| d<   d[| d\<   da| d<   d^| d_<   y |dbk(  r|dk(  rdc| d<   dd| de<   y df| d<   dd| de<   y |dgk(  r%|dk(  rdc| dF<   dc| d<   dd| de<   y df| dF<   df| d<   dd| de<   y |dhk(  r9|dk(  rd<| d<   di| d\<   dj| d<   d<| dk<   d9| dl<   y dm| d<   di| d\<   dn| d<   dm| dk<   d9| dl<   y |dok(  r/|dk(  rd<| d<   di| d\<   dp| d<   d<| dk<   y dq| d<   di| d\<   dr| d<   dq| dk<   y |dsk(  r%|dk(  rdt| d<   dt| d<   d[| d\<   y d| d<   d| d<   d[| d\<   y |duk(  rM|dk(  r$dv| dF<   d| d<   dw| d)<   d| d*<   d[| dx<   d[| dy<   dv| dz<   y dv| dF<   d{| d<   d|| d)<   d{| d*<   d[| dx<   d[| dy<   dv| dz<   y |d}v r5|dk(  rd~| d<   d]| d<   d~| d<   nd€| d<   d�| d<   d€| d<   d‚| v r|dƒk(  rd„nd…| d‚<   y |d†k(  rd‡| d‚<   |dk(  rd2| d<   y d4| d<   y |dˆk(  r|dk(  rd‰| d<   d‰| d<   y dŠ| d<   dŠ| d<   y |d‹k(  r%|dk(  rdŒ| d�<   dŽ| d�<   d^| d�<   y d‘| d�<   d’| d�<   d^| d�<   y |d“k(  r%|dk(  rd”| d<   d•| d–<   d^| d�<   y d—| d<   d˜| d–<   d^| d�<   y |d™v rdi| dš<   d9| d›<   d[| dœ<   y y )�N)r   é   zindex must be 0 or 1, got Úaddrr   zat::rand({6, 6})Úselfzat::rand({6})Úvec1Úvec2zat::rand({22, 22})zat::rand({22})ÚmvÚvecÚaddbmmÚcrosszat::rand({3, 3, 3})Úotherzat::rand({22, 3, 22})Útakez(at::randint(0, 216, {20}, torch::kInt64)Úindexz*at::randint(0, 1000, {100}, torch::kInt64)Útake_along_dimzat::argsort(self0, 1, true)Úindiceszat::argsort(self1, 1, true)Úmasked_selectzat::randn({6, 6, 6}) > 0.5Úmaskzat::rand({22, 22, 22}) > 0.5ÚorgqrÚinput2ÚormqrÚquantileÚqz"linear"ÚinterpolationÚnanquantileÚmulti_margin_lossz"at::randint(6, {6}, torch::kInt64)ÚtargetÚweightz$at::randint(22, {22}, torch::kInt64)Úmultilabel_margin_lossz%at::randint(6, {6, 6}, torch::kInt64)z(at::randint(22, {22, 22}, torch::kInt64)Únll_lossÚ
nll_loss2dzat::rand({6, 6, 6, 6})z(at::randint(6, {6, 6, 6}, torch::kInt64)zat::rand({22, 22, 22, 22})z,at::randint(22, {22, 22, 22}, torch::kInt64))Úfft_fftÚfft_ifftÚfft_rfftÚ	fft_irfftÚfft_hfftÚ	fft_ihfftz	"forward"ÚnormÚlinalg_tensorinvÚ2ÚindÚaddmvzat::rand({2})zat::rand({2, 2})Úmatzat::rand({35})zat::rand({35, 35})Úacoshz)at::rand({2, 2, 2}) + at::ones({2, 2, 2})z)at::rand({5, 5, 5}) + at::ones({5, 5, 5})Úadaptive_max_pool2d_backwardzat::rand({2, 2, 2}, at::kFloat)Úgrad_outputz'at::randint(0, 1, {2, 2, 2}, at::kLong)zat::rand({3, 3, 3}, at::kFloat)z'at::randint(0, 1, {3, 3, 3}, at::kLong)Úadaptive_max_pool3d_backwardz"at::rand({2, 2, 2, 2}, at::kFloat)z*at::randint(0, 1, {2, 2, 2, 2}, at::kLong)z"at::rand({3, 3, 3, 3}, at::kFloat)z*at::randint(0, 1, {3, 3, 3, 3}, at::kLong)Úbitwise_left_shiftz+at::randint(1, 1 << 4, {6, 6, 6}, at::kInt)z'at::randint(1, 26, {6, 6, 6}, at::kInt)z.at::randint(1, 1 << 4, {22, 22, 22}, at::kInt)z*at::randint(1, 26, {22, 22, 22}, at::kInt)Úbitwise_right_shiftz2at::randint(1 << 21, 1 << 30, {6, 6, 6}, at::kInt)z'at::randint(1, 22, {6, 6, 6}, at::kInt)z5at::randint(1 << 21, 1 << 30, {22, 22, 22}, at::kInt)z*at::randint(1, 22, {22, 22, 22}, at::kInt)Úgatherz&at::randint(1, 100, {2,2,2}, at::kInt)Ú1Údimz)at::randint(0, 1, {2,2,2}, torch::kInt64)ÚfalseÚsparse_gradz&at::randint(1, 100, {5,5,5}, at::kInt)z)at::randint(0, 4, {5,5,5}, torch::kInt64)Úgeluzat::rand({6, 6, 6})z"tanh"Úapproximatezat::rand({22, 22, 22})Úgelu_backwardÚ	index_addÚ0z at::randint(0, 1, {2}, at::kInt)ÚsourceÚalphazat::rand({16})z"at::randint(0, 10, {16}, at::kInt)Ú
index_copyz!at::randint(0, 1, {2}, at::kLong)zat::rand({32})z#at::randint(0, 10, {32}, at::kLong)Úlinalg_crosszat::rand({6, 3, 6})Únll_loss_backwardzat::rand({})z%at::randint(0, 5, {6}, torch::kInt64)Ú	reductionÚignore_indexÚtotal_weightzat::rand({36})z'at::randint(0, 11, {36}, torch::kInt64))ÚscatterÚscatter_addÚ_scatter_reducez+at::randint(1, 100, {2,2,2}, torch::kInt64)Úsrcz+at::randint(1, 100, {5,5,5}, torch::kInt64)z)at::randint(0, 1, {5,5,5}, torch::kInt64)Úreducer‚   z"sum"z"add"Úscatter_reducez"mean"Úspecial_zetaz2at::rand({2,2,2}, at::kDouble) + at::ones({2,2,2})z2at::rand({5,5,5}, at::kDouble) + at::ones({5,5,5})Ú _convert_indices_from_csr_to_cooz!torch::tensor({1}, torch::kInt32)Úcrow_indicesz'torch::tensor({0, 1, 0}, torch::kInt32)Úcol_indicesÚ	out_int32z!torch::tensor({0}, torch::kInt32)zBtorch::tensor({0, 1, 0, 2, 1, 2, 0, 1, 0, 2, 1, 2}, torch::kInt32)Ú _convert_indices_from_coo_to_csrz at::randint(0, 3, {2}, at::kInt)Ú10Úsizez!at::randint(0, 3, {12}, at::kInt)Ú24)ÚdiagonalÚlinalg_diagonalÚoffsetÚdim1Údim2)ÚAssertionError)Úarg_mapÚop_namerJ   s      r   Úoverride_test_valuesr—   ?   s2
  € Ø�FÑÜÐ9¸%¸ÐAÓBÐBØ�&ÒØ�AŠ:Ø0ˆG�F‰OØ-ˆG�F‰OØ-ˆG�F‰Oð
 	ð 3ˆG�F‰OØ.ˆG�F‰OØ.ˆG�F‰OØØ�$‚Ø�AŠ:Ø0ˆG�F‰OØ,ˆG�E‰Nð 	ð 3ˆG�F‰OØ-ˆG�E‰NØØ�(ÒØ�AŠ:Ø0ˆG�F‰Oð 	ð 3ˆG�F‰OØØ�'ÒØ�AŠ:Ø3ˆG�F‰OØ4ˆG�GÑð 	ð 6ˆG�F‰OØ6ˆG�GÑØØ�&ÒØ�AŠ:ØIˆG�GÑð 	ð  LˆG�GÑØØÐ"Ò"Ø�AŠ:Ø!>ˆG�IÑð 	ð "?ˆG�IÑØØ�/Ò!Ø�AŠ:Ø:ˆG�F‰Oð 	ð =ˆG�F‰OØØ�'ÒØ�AŠ:Ø 2ˆG�HÑð 	ð !5ˆG�HÑØØ�'ÒØ�AŠ:Ø 2ˆG�HÑð 	ð !5ˆG�HÑØØ�*ÒØ�AŠ:Ø*ˆG�C‰LØ'1ˆG�OÑ$ð 	ð ,ˆG�C‰LØ'1ˆG�OÑ$ØØ�-ÒØ�AŠ:Ø*ˆG�C‰LØ'1ˆG�OÑ$ð 	ð ,ˆG�C‰LØ'1ˆG�OÑ$ØØÐ%Ò%Ø�AŠ:Ø0ˆG�F‰OØ DˆG�HÑØ /ˆG�HÑð
 	ð 3ˆG�F‰OØ FˆG�HÑØ 0ˆG�HÑØØÐ*Ò*Ø�AŠ:Ø0ˆG�F‰OØ GˆG�HÑð 	ð 3ˆG�F‰OØ JˆG�HÑØØ�*ÒØ�AŠ:Ø0ˆG�F‰OØ DˆG�HÑØ /ˆG�HÑð
 	ð 3ˆG�F‰OØ FˆG�HÑØ 0ˆG�HÑØØ�,ÒØ�AŠ:Ø6ˆG�F‰OØ JˆG�HÑØ /ˆG�HÑð
 	ð ;ˆG�F‰OØ NˆG�HÑØ 0ˆG�HÑØØð ñ ð &ˆ�‰ØØÐ$Ò$Ø�AŠ:Ø6ˆG�F‰OØ ˆG�E‰Nð 	ð ;ˆG�F‰OØ ˆG�E‰NØØ�'ÒØ�AŠ:Ø-ˆG�F‰OØ/ˆG�E‰NØ,ˆG�E‰Nð
 	ð /ˆG�F‰OØ1ˆG�E‰NØ-ˆG�E‰NØØ�'ÒØ�AŠ:ØIˆG�F‰Oð 	ð JˆG�F‰OØØÐ0Ò0Ø�AŠ:Ø%FˆG�MÑ"Ø?ˆG�F‰OØ!JˆG�IÑð
 	ð &GˆG�MÑ"Ø?ˆG�F‰OØ!JˆG�IÑØØÐ0Ò0Ø�AŠ:Ø%IˆG�MÑ"ØBˆG�F‰OØ!MˆG�IÑð
 	ð &JˆG�MÑ"ØBˆG�F‰OØ!MˆG�IÑØØÐ&Ò&Ø�AŠ:ØKˆG�F‰OØHˆG�GÑð 	ð OˆG�F‰OØKˆG�GÑØØÐ'Ò'Ø�AŠ:ØRˆG�F‰OØHˆG�GÑð 	ð VˆG�F‰OØKˆG�GÑØØ�(ÒØ�AŠ:ØFˆG�F‰OØ ˆG�E‰NØJˆG�GÑØ%,ˆG�MÑ"ð 	ð	 GˆG�F‰OØ ˆG�E‰NØJˆG�GÑØ%,ˆG�MÑ"ØØ�&ÒØ�AŠ:Ø3ˆG�F‰OØ%-ˆG�MÑ"ð 	ð 7ˆG�F‰OØ%-ˆG�MÑ"ØØ�/Ò!Ø�AŠ:Ø%:ˆG�MÑ"Ø3ˆG�F‰OØ%-ˆG�MÑ"ð
 	ð &>ˆG�MÑ"Ø6ˆG�F‰OØ%-ˆG�MÑ"ØØ�+ÒØ�AŠ:Ø-ˆG�F‰OØ ˆG�E‰NØAˆG�GÑØ /ˆG�HÑØ"ˆG�GÑð 	ð /ˆG�F‰OØ ˆG�E‰NØCˆG�GÑØ 0ˆG�HÑØ"ˆG�GÑØØ�,ÒØ�AŠ:Ø-ˆG�F‰OØ ˆG�E‰NØBˆG�GÑØ /ˆG�HÑð 	ð	 /ˆG�F‰OØ ˆG�E‰NØDˆG�GÑØ 0ˆG�HÑØØ�.Ò Ø�AŠ:Ø3ˆG�F‰OØ4ˆG�GÑØ ˆG�E‰Nð
 	ð 6ˆG�F‰OØ6ˆG�GÑØ ˆG�E‰NØØÐ%Ò%Ø�AŠ:Ø%3ˆG�MÑ"Ø-ˆG�F‰OØ GˆG�HÑØ /ˆG�HÑØ#&ˆG�KÑ Ø&)ˆG�NÑ#Ø&4ˆG�NÑ#ð 	ð &4ˆG�MÑ"Ø.ˆG�F‰OØ IˆG�HÑØ 0ˆG�HÑØ#&ˆG�KÑ Ø&)ˆG�NÑ#Ø&4ˆG�NÑ#ØØÐ?Ñ?Ø�AŠ:ØKˆG�F‰OØJˆG�GÑØJˆG�EŠNàKˆG�F‰OØJˆG�GÑØJˆG�E‰NØ�wÑØ+2Ð6GÒ+G¡ÈWˆG�HÑØØÐ"Ò"Ø$ˆ�ÑØ�AŠ:ØIˆG�GÑð 	ð  NˆG�GÑØØ�.Ò Ø�AŠ:ØRˆG�F‰OØSˆG�GÑð 	ð SˆG�F‰OØSˆG�GÑØØÐ4Ò4Ø�AŠ:Ø&IˆG�NÑ#Ø%NˆG�MÑ"Ø#*ˆG�KÑ ð 	ð 'JˆG�NÑ#àTð �MÑ"ð $+ˆG�KÑ ØØÐ4Ò4Ø�AŠ:Ø@ˆG�F‰OØ"ˆG�F‰OØ#*ˆG�KÑ ð
 	ð BˆG�F‰OØ"ˆG�F‰OØ#*ˆG�KÑ ØØÐ1Ñ1Øˆ�ÑØˆ�‰Øˆ�‰Øð	 2r   N)r   ú/NativeFunctionsGroup | NativeFunctionsViewGroupÚreturnr	   )r   r˜   r™   Úbool)r•   zdict[str, str]r–   r	   rJ   Úintr™   ÚNone)
Ú
__future__r   Útorchgen.modelr   r   r   Ú	frozensetr;   r=   r—   © r   r   Ú<module>r¡      s.   ðÝ "ç Ió%ñ !ð(ó*Ð óZ-ô
Fr   