
    S j8                    N    S SK Jr  S SKJrJr  SS jr\" S5      rS	S jrS
S jr	g)    )annotations)NativeFunctionsGroupNativeFunctionsViewGroupc                    [        U [        5      (       a=  [        U R                  R                  R
                  R
                  R                  5      $ [        U R                  R                  5      $ N)	
isinstancer   str
functionalfuncnamebaseview	root_name)gs    o/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/torchgen/static_runtime/config.pyfunc_name_base_strr      sL    !)**1<<$$))..3344166##$$    )'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                *    [        U 5      nU[        ;   $ r   )r   is_hand_written_ops_)r   	name_bases     r   is_hand_writtenr=   :   s    "1%I,,,r   c                
   US;  a  [        SU 35      eUS:X  a&  US:X  a  SU S'   SU S'   SU S	'   g S
U S'   SU S'   SU S	'   g US:X  a  US:X  a  SU S'   SU S'   g S
U S'   SU S'   g US:X  a  US:X  a  SU S'   g S
U S'   g US:X  a  US:X  a  SU S'   SU S'   g SU S'   SU S'   g US:X  a  US:X  a  SU S'   g SU S'   g US:X  a  US:X  a  SU S'   g SU S'   g US:X  a  US:X  a  SU S'   g SU S'   g US:X  a  US:X  a  SU S '   g S
U S '   g US!:X  a  US:X  a  SU S '   g S
U S '   g US":X  a  US:X  a  SU S#'   S$U S%'   g SU S#'   S$U S%'   g US&:X  a  US:X  a  SU S#'   S$U S%'   g SU S#'   S$U S%'   g US':X  a&  US:X  a  SU S'   S(U S)'   SU S*'   g S
U S'   S+U S)'   SU S*'   g US,:X  a  US:X  a  SU S'   S-U S)'   g S
U S'   S.U S)'   g US/:X  a&  US:X  a  SU S'   S(U S)'   SU S*'   g S
U S'   S+U S)'   SU S*'   g US0:X  a&  US:X  a  S1U S'   S2U S)'   SU S*'   g S3U S'   S4U S)'   SU S*'   g US5;   a  S6U S7'   g US8:X  a  US:X  a  S1U S'   S9U S:'   g S3U S'   S9U S:'   g US;:X  a&  US:X  a  S<U S'   S=U S>'   S<U S'   g S?U S'   S@U S>'   S?U S'   g USA:X  a  US:X  a  SBU S'   g SCU S'   g USD:X  a&  US:X  a  SEU SF'   SEU S'   SGU S'   g SHU SF'   SHU S'   SIU S'   g USJ:X  a&  US:X  a  SKU SF'   SKU S'   SLU S'   g SMU SF'   SMU S'   SNU S'   g USO:X  a  US:X  a  SPU S'   SQU S'   g SRU S'   SSU S'   g UST:X  a  US:X  a  SUU S'   SVU S'   g SWU S'   SXU S'   g USY:X  a0  US:X  a  SZU S'   S[U S\'   S]U S'   S^U S_'   g S`U S'   S[U S\'   SaU S'   S^U S_'   g USb:X  a  US:X  a  ScU S'   SdU Se'   g SfU S'   SdU Se'   g USg:X  a&  US:X  a  ScU SF'   ScU S'   SdU Se'   g SfU SF'   SfU S'   SdU Se'   g USh:X  a:  US:X  a  S<U S'   SiU S\'   SjU S'   S<U Sk'   S9U Sl'   g SmU S'   SiU S\'   SnU S'   SmU Sk'   S9U Sl'   g USo:X  a0  US:X  a  S<U S'   SiU S\'   SpU S'   S<U Sk'   g SqU S'   SiU S\'   SrU S'   SqU Sk'   g USs:X  a&  US:X  a  StU S'   StU S'   S[U S\'   g SU S'   SU S'   S[U S\'   g USu:X  aN  US:X  a$  SvU SF'   SU S'   SwU S)'   SU S*'   S[U Sx'   S[U Sy'   SvU Sz'   g SvU SF'   S{U S'   S|U S)'   S{U S*'   S[U Sx'   S[U Sy'   SvU Sz'   g US};   a9  US:X  a  S~U S'   S]U S'   S~U S'   OSU S'   SU S'   SU S'   SU ;   a  US:X  a  SOSU S'   g US:X  a  SU S'   US:X  a  S2U S'   g S4U S'   g US:X  a  US:X  a  SU S'   SU S'   g SU S'   SU S'   g US:X  a&  US:X  a  SU S'   SU S'   S^U S'   g SU S'   SU S'   S^U S'   g US:X  a&  US:X  a  SU S'   SU S'   S^U S'   g SU S'   SU S'   S^U S'   g US;   a  SiU S'   S9U S'   S[U S'   g g )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:0GFO-GFO-GFO
 	 3GFO.GFO.GFO$A:0GFO,GEN 	 3GFO-GEN(A:0GFO 	 3GFO'A:3GFO4GG 	 6GFO6GG&A:IGG 	  LGG""A:!>GI 	 "?GI/!A::GFO 	 =GFO'A: 2GH 	 !5GH'A: 2GH 	 !5GH*A:*GCL'1GO$ 	 ,GCL'1GO$-A:*GCL'1GO$ 	 ,GCL'1GO$%%A:0GFO DGH /GH
 	 3GFO FGH 0GH**A:0GFO GGH 	 3GFO JGH*A:0GFO DGH /GH
 	 3GFO FGH 0GH,A:6GFO JGH /GH
 	 ;GFO NGH 0GH   &$$A:6GFO GEN 	 ;GFO GEN'A:-GFO/GEN,GEN
 	 /GFO1GEN-GEN'A:IGFO 	 JGFO00A:%FGM"?GFO!JGI
 	 &GGM"?GFO!JGI00A:%IGM"BGFO!MGI
 	 &JGM"BGFO!MGI&&A:KGFOHGG 	 OGFOKGG''A:RGFOHGG 	 VGFOKGG(A:FGFO GENJGG%,GM" 		 GGFO GENJGG%,GM"&A:3GFO%-GM" 	 7GFO%-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 GEN
 	 6GFO6GG GEN%%A:%3GM"-GFO GGH /GH#&GK &)GN#&4GN# 	 &4GM".GFO IGH 0GH#&GK &)GN#&4GN#??A:KGFOJGGJGENKGFOJGGJGENw+26G+GWGH""$A:IGG 	  NGG. A:RGFOSGG 	 SGFOSGG44A:&IGN#%NGM"#*GK  	 'JGN#T M" $+GK 44A:@GFO"GFO#*GK 
 	 BGFO"GFO#*GK 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   