
    aj8                    P    d dl mZ d dlmZmZ ddZ ed          Zdd
ZddZ	dS )    )annotations)NativeFunctionsGroupNativeFunctionsViewGroupg/NativeFunctionsGroup | NativeFunctionsViewGroupreturnstrc                    t          | t                    r(t          | j        j        j        j        j                  S t          | j        j                  S N)	
isinstancer   r	   
functionalfuncnamebaseview	root_name)r   s    c/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/torchgen/static_runtime/config.pyfunc_name_base_strr      sD    !)** %1<$).344416#$$$    )'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   whereboolc                2    t          |           }|t          v S r   )r   is_hand_written_ops_)r   	name_bases     r   is_hand_writtenr@   :   s    "1%%I,,,r   arg_mapdict[str, str]op_nameindexintNonec                &   |dvrt          d|           |dk    r'|dk    rd| d<   d| d<   d| d	<   nd
| d<   d| d<   d| d	<   d S |dk    r|dk    rd| d<   d| d<   n
d
| d<   d| d<   d S |dk    r|dk    rd| d<   nd
| d<   d S |dk    r|dk    rd| d<   d| d<   n
d| d<   d| d<   d S |dk    r|dk    rd| d<   nd| d<   d S |dk    r|dk    rd| d<   nd| d<   d S |dk    r|dk    rd| d<   nd| d<   d S |dk    r|dk    rd| d <   nd
| d <   d S |d!k    r|dk    rd| d <   nd
| d <   d S |d"k    r|dk    rd| d#<   d$| d%<   n
d| d#<   d$| d%<   d S |d&k    r|dk    rd| d#<   d$| d%<   n
d| d#<   d$| d%<   d S |d'k    r'|dk    rd| d<   d(| d)<   d| d*<   nd
| d<   d+| d)<   d| d*<   d S |d,k    r|dk    rd| d<   d-| d)<   n
d
| d<   d.| d)<   d S |d/k    r'|dk    rd| d<   d(| d)<   d| d*<   nd
| d<   d+| d)<   d| d*<   d S |d0k    r'|dk    rd1| d<   d2| d)<   d| d*<   nd3| d<   d4| d)<   d| d*<   d S |d5v rd6| d7<   d S |d8k    r|dk    rd1| d<   d9| d:<   n
d3| d<   d9| d:<   d S |d;k    r'|dk    rd<| d<   d=| d><   d<| d<   nd?| d<   d@| d><   d?| d<   d S |dAk    r|dk    rdB| d<   ndC| d<   d S |dDk    r'|dk    rdE| dF<   dE| d<   dG| d<   ndH| dF<   dH| d<   dI| d<   d S |dJk    r'|dk    rdK| dF<   dK| d<   dL| d<   ndM| dF<   dM| d<   dN| d<   d S |dOk    r|dk    rdP| d<   dQ| d<   n
dR| d<   dS| d<   d S |dTk    r|dk    rdU| d<   dV| d<   n
dW| d<   dX| d<   d S |dYk    r1|dk    rdZ| d<   d[| d\<   d]| d<   d^| d_<   nd`| d<   d[| d\<   da| d<   d^| d_<   d S |dbk    r|dk    rdc| d<   dd| de<   n
df| d<   dd| de<   d S |dgk    r'|dk    rdc| dF<   dc| d<   dd| de<   ndf| dF<   df| d<   dd| de<   d S |dhk    r;|dk    rd<| d<   di| d\<   dj| d<   d<| dk<   d9| dl<   ndm| d<   di| d\<   dn| d<   dm| dk<   d9| dl<   d S |dok    r1|dk    rd<| d<   di| d\<   dp| d<   d<| dk<   ndq| d<   di| d\<   dr| d<   dq| dk<   d S |dsk    r'|dk    rdt| d<   dt| d<   d[| d\<   nd| d<   d| d<   d[| d\<   d S |duk    rO|dk    r$dv| dF<   d| d<   dw| d)<   d| d*<   d[| dx<   d[| dy<   dv| dz<   n#dv| dF<   d{| d<   d|| d)<   d{| d*<   d[| dx<   d[| dy<   dv| dz<   d S |d}v r8|dk    rd~| d<   d]| d<   d~| d<   nd| d<   d| d<   d| d<   d| v r|dk    rdnd| d<   d S |dk    rd| d<   |dk    rd2| d<   nd4| d<   d S |dk    r|dk    rd| d<   d| d<   n
d| d<   d| d<   d S |dk    r'|dk    rd| d<   d| d<   d^| d<   nd| d<   d| d<   d^| d<   d S |dk    r'|dk    rd| d<   d| d<   d^| d<   nd| d<   d| d<   d^| d<   d S |dv rdi| d<   d9| d<   d[| d<   d S d S )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)rD   z*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)rA   rC   rD   s      r   override_test_valuesr   ?   s
   FA%AABBB&A::0GFO-GFO-GFOO2GFO.GFO.GFO$A::0GFO,GENN2GFO-GEN(A::0GFOO2GFO'A::3GFO4GG5GFO6GG&A::IGGKGG"""A::!>GI!>GI/!!A:::GFOO<GFO'A:: 2GH 4GH'A:: 2GH 4GH*A::*GCL'1GO$$+GCL'1GO$-A::*GCL'1GO$$+GCL'1GO$%%%A::0GFO DGH /GH2GFO FGH 0GH***A::0GFO GGH2GFO JGH*A::0GFO DGH /GH2GFO FGH 0GH,A::6GFO JGH /GH:GFO NGH 0GH    &$$$A::6GFO GENN:GFO GEN'A::-GFO/GEN,GENN.GFO1GEN-GEN'A::IGFOOIGFO000A::%FGM"?GFO!JGI%FGM"?GFO!JGI000A::%IGM"BGFO!MGI%IGM"BGFO!MGI&&&A::KGFOHGGNGFOKGG'''A::RGFOHGGUGFOKGG(A::FGFO GENJGG%,GM""FGFO GENJGG%,GM"&A::3GFO%-GM""6GFO%-GM"/!!A::%:GM"3GFO%-GM""%=GM"6GFO%-GM"+A::-GFO GENAGG /GH"GG.GFO GENCGG 0GH"GG,A::-GFO GENBGG /GH.GFO GENDGG 0GH.  A::3GFO4GG GENN5GFO6GG GEN%%%A::%3GM"-GFO GGH /GH#&GK &)GN#&4GN##%3GM".GFO IGH 0GH#&GK &)GN#&4GN#???A::KGFOJGGJGENNKGFOJGGJGENw+26G+G+GWGH"""$A::IGGMGG.  A::RGFOSGGRGFOSGG444A::&IGN#%NGM"#*GK  &IGN#T M" $+GK 444A::@GFO"GFO#*GK  AGFO"GFO#*GK 111	 21r   N)r   r   r   r	   )r   r   r   r<   )rA   rB   rC   r	   rD   rE   r   rF   )

__future__r   torchgen.modelr   r   r   	frozensetr>   r@   r    r   r   <module>r      s    " " " " " " I I I I I I I I% % % % !y(* * Z- - - -
F F F F F Fr   