
    R jk                     X   S SK r S SKJr  S SKJr  S SKrSSKJr  SSKJ	r	J
r
  \R                  R                  R                  5       (       a?     SS\\
   S\S	\\\      S
\S\\
   4
S jjrS\\
   S\\
   4S jrS\\
   S\\
   4S jrg   SS\\
   S\S	\\\      S
\S\\
   4
S jjrg)    N)Sequence)Optional   )MACOS_VERSION)DecorateInfoOpInfoopsdevice_typexfail_exclusionsparsereturnc                 -  ^ Uc  / n1 SknS[         R                  /0n0 SS _SS _SS _SS _SS _SS _S	S _S
S _SS _SS _SS _SS _SS _SS _SS _SS _SS _0 SS _SS _SS _SS _SS _SS _SS _S[         R                  /_S[         R                  /_S[         R                  /_SS _SS _S S _S!S _S"[         R                  [         R                  [         R
                  [         R                  [         R                  [         R                  /_S#[         R                  /_S$S _E0 S%[         R                  /_S&[         R                  [         R                  [         R                  [         R                  [         R
                  /_S'S _S([         R                  [         R
                  [         R                  [         R                  [         R                  /_S)[         R                  [         R
                  [         R                  [         R                  [         R                  /_S*[         R                  [         R
                  [         R                  [         R                  [         R                  /_S+[         R                  [         R                  [         R                  [         R                  [         R
                  /_S,S _S-S _S.[         R                  [         R
                  /_S/[         R
                  [         R                  [         R                  [         R                  [         R                  /_S0[         R                  [         R                  [         R                  [         R                  [         R
                  /_S1S _S2[         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_S3[         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_S4S _S5S _E0 S6[         R                  /_S7S _S8[         R                  [         R                  [         R
                  [         R                  [         R                  /_S9[         R                  [         R
                  [         R                  [         R                  [         R                  /_S:[         R                  [         R                  [         R                  [         R                  [         R
                  [         R                  /_S;[         R                  /_S<[         R                  /_S=[         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_S>[         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_S?[         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_S@[         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_SA[         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_SB[         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_SC[         R
                  [         R                  [         R                  [         R                  [         R                  /_SDS _SE[         R                  [         R                  [         R
                  [         R                  [         R                  /_SF[         R                  [         R                  [         R
                  [         R                  /_E0 SG[         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_SH[         R                  /_SI[         R                  /_SJ[         R                  /_SKS _SL[         R                  /_SMS _SN[         R                  [         R
                  [         R                  [         R                  [         R                  /_SO[         R
                  [         R                  [         R                  [         R                  [         R                  /_SPS _SQ[         R                  /_SRS _SSS _ST[         R                  /_SUS _SV[         R                  /_SW[         R                  [         R                  [         R                  [         R
                  [         R                  /_E0 SX[        SY:  a   [         R
                  [         R                  /O[         R                  /_SZ[        SY:  a   [         R
                  [         R                  /O[         R                  /_S[[         R                  /_S\S _S]S _S^S _S_S _S`S _SaS _SbS _ScS _SdS _SeS _SfS _SgS _ShS _SiS _E0 SjS _SkS _Sl[         R                  [         R                  /_SmS _SnS _SoS _SpS _Sq[         R                  /_Sr[         R                  [         R                  [         R
                  [         R                  [         R                  /_Ss[         R                  [         R                  [         R
                  [         R                  [         R                  /_St[         R                  [         R                  [         R
                  [         R                  [         R                  /_Su[         R                  [         R                  [         R
                  [         R                  [         R                  /_SvS _Sw[         R                  /_Sx[         R                  /_SyS _SzS _E0 S{[         R                  [         R                  [         R
                  [         R                  [         R                  [         R                  /_S|S _S}S _S~S _SS _SS _SS _SS _S[         R                  /_S[         R                  /_S[         R                  /_S[         R                  /_S[         R                  [         R                  /_S[         R                  [         R                  [         R                  /_S[         R                  [         R                  /_S[         R                  [         R
                  [         R                  [         R                  [         R                  /_S[         R                  [         R
                  [         R                  [         R                  [         R                  /_ES[         R                  [         R
                  [         R                  [         R                  [         R                  /0EnS S S S S S.n[        SY:  a  UR                  S S S.5        U(       a  UR                  U5        0 S[         R                  [         R
                  [         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_SS _SS _SS _SS _S[         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  [         R
                  [         R                  [         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_S[         R                  [         R                  [         R                  /_[         R                  [         R                  [         R                  /[         R                  [         R                  [         R                  [         R                  /[         R                  [         R                  [         R                  /[         R                  [         R                  [         R                  /[         R                  [         R                  [         R                  /[         R                  [         R                  [         R
                  [         R                  /[         R                  [         R                  [         R                  /[         R                  [         R                  [         R                  /[         R                  [         R                  [         R
                  [         R                  [         R                  [         R                  /S [         R                  [         R                  [         R                  /S.EnS [         R                  /[         R                  [         R                  [         R                  [         R                  [         R                  /[         R                  [         R                  [         R                  [         R                  [         R                  /S.n	S S S S S S S.n
SS 0nS[        S[        SS 4U4S jjnU  GH  nUR                  UR                   -   nU" U[        ["        R$                  [         R&                  [         R(                  /S95        U(       a4  U" U[        ["        R*                  " S5      [         R,                  /S95        X;   a'  U" U[        ["        R*                  " S5      X   S95        X;   a'  U" U[        ["        R*                  " S5      X   S95        UUU	4 H2  nX;   d  M
  X;  d  M  U" U[        ["        R$                  X   S95        M4     X;   a0  X;  a+  [        S:  a!  U" U[        ["        R$                  X^   S95        X;  d  GMe  U" U[        ["        R$                  [         R.                  [         R                  /S95        GM     U $ )N>  HTteqmHmTmmmvneabsaddallanybmmcatcoscovdivdotexpeyeintlogmulnegsgnsinsubsumsvdtanacosasinatanboolbytecharconjcoshdiagdiffexp2fillflipfullhalfimagitemkronlerplog2longmeanonesprodrealrollrsubsincsinhsqrtstfttanhtriltriuvdotviewacoshaddmvangleasinhatanhchalfchunkcloneemptyequalexpm1floatinnerisinfisnanldexplog10log1pouterrandnravelrot90rsqrtshortslicesplitstacktracetrapzwherezero_zerosaddbmmcfloatdsplitdstackeinsumexpandfliplrflipudhsplithstackisrealmatmulnarrowselectsquaret_copyunbindunfoldvsplitvstackaddcdivaddcmulbaddbmmflattenisclosemovedimnonzeropermutereshapesigmoidsqueezeview_as__radd____rdiv____rmul____rsub__allcloseargwherebfloat16corrcoefdiagflatdiagonalgradientisfinitelinspacenew_fullnew_onespinversepositive	expand_as	full_like	index_add	index_put	logaddexp	logsumexp	new_zeros	ones_like	tensordot	transpose	trapezoid	unflatten	unsqueeze	view_copy
_chunk_cat
alias_copy
as_strided
atleast_1d
atleast_2d
atleast_3d
block_diag
contiguous
diag_embed
index_copy
logaddexp2
logical_or
reciprocal
reshape_as
zeros_like__getitem____rmatmul__expand_copylogical_andlogical_notlogical_xormasked_fillnarrow_copyresolve_negsum_to_sizetrue_divideunbind_copyunfold_copybroadcast_tocolumn_stackcombinationsindex_selectpermute_copyresolve_conjsqueeze_copytensor_splitunsafe_chunkunsafe_splitview_as_realconj_physicalcount_nonzerodiagonal_copyempty_stridedmasked_selectscalar_tensorcartesian_prodempty_permutedmasked_scattersplitlist_argstranspose_copyunsqueeze_copyaddmmdecomposedas_strided_copyconstant_pad_ndsqueezemultiplediagonal_scattersplit_with_sizesbroadcast_tensorsrepeat_interleavesoftmaxwith_dtypeas_strided_scatterdivno_rounding_mode_unsafe_masked_indexsplit_with_sizes_copylinspacetensor_overloadmeshgridlist_of_tensorsmeshgridvariadic_tensors#_unsafe_masked_index_put_accumulatefft.fftfft.fft2fft.fftnfft.hfftfft.ifft	fft.hfft2	fft.hfftn	fft.ifft2	fft.ifftn	fft.irfft
fft.irfft2
fft.irfftn
linalg.svd
masked.std
masked.sum
masked.varlinalg.pinvmasked.meanmasked.prodfft.fftshiftspecial.entrspecial.zetafft.ifftshiftlinalg.vecdotlinalg.diagonalspecial.xlog1pylinalg.multi_dotmasked.logsumexpnn.functional.conv1dnn.functional.conv2dnn.functional.unfoldnn.functional.rms_normnn.functional.softsignnn.functional.padreflectnn.functional.tanhshrinknn.functional.padcircularnn.functional.padconstantlinalg.householder_productnn.functional.padreplicatenn.functional.pixel_shufflespecial.spherical_bessel_j0nn.functional.pixel_unshufflenn.functional.conv_transpose1dnn.functional.conv_transpose2dnn.functional.conv_transpose3dnn.functional.softminwith_dtype0nn.functional.feature_alpha_dropoutwithout_trainr   logspacelogspacetensor_overload
linalg.eiglinalg.eigvalsputcholesky_solvefrexpgeqrfznn.functional.grid_samplehash_tensor	heavisideindex_reduceprodindex_reducemeanindex_reduceamaxindex_reduceaminlcmzlinalg.condzlinalg.eighzlinalg.eigvalshzlinalg.ldl_factorzlinalg.ldl_factor_exzlinalg.ldl_solvezlinalg.lstsqzlinalg.lstsqgrad_orientedzlinalg.matrix_normzlinalg.normzlinalg.normsubgradients_at_zeroz	linalg.qrzlinalg.svdvalszmasked.median
matrix_exp max_pool2d_with_indices_backwardmedianmode	nanmediannative_batch_normnormnucznn.functional.avg_pool1dznn.functional.avg_pool2dznn.functional.avg_pool3dznn.functional.batch_normz#nn.functional.fractional_max_pool2dz#nn.functional.fractional_max_pool3dznn.functional.group_normznn.functional.gluznn.functional.huber_lossz!nn.functional.adaptive_avg_pool3dz!nn.functional.adaptive_max_pool1dz!nn.functional.adaptive_max_pool2dz!nn.functional.adaptive_max_pool3dznn.functional.interpolateareaz nn.functional.interpolatebicubicznn.functional.ctc_lossz!nn.functional.local_response_normznn.functional.logsigmoidznn.functional.max_pool1dznn.functional.max_pool2dznn.functional.max_pool3dznn.functional.max_unpool1dznn.functional.max_unpool1dgradznn.functional.max_unpool2dznn.functional.max_unpool2dgradznn.functional.max_unpool3dznn.functional.max_unpool3dgradznn.functional.mishznn.functional.multi_margin_lossz$nn.functional.multilabel_margin_lossz)nn.functional.multilabel_soft_margin_lossznn.functional.nll_lossr  r  z#nn.functional.padreplicate_negativeznn.functional.pdistznn.functional.reluznn.functional.rreluznn.functional.siluznn.functional.softplusznn.functional.normznn.functional.thresholdormqrpca_lowrankpowqr	remainderrounddecimals_0scatter_reduceamaxg      .@scatter_reduceaminscatter_reducemeansegment_reducez_segment.reducezsegment.reducesegment_reduce_offsets_segment_reduce_offsets_segment_reduce_lengths_segment_reducelengths_segment_reduceoffsetsz	sparse.mmzsparse.sampled_addmmsparse.mmreducezspecial.airy_aizspecial.laguerre_polynomial_lzspecial.legendre_polynomial_pzspecial.log_ndtrzspecial.ndtrirL   svd_lowranksymeigtaketotruncvar_meanunbiasedvar_meanstd_meanstd_meanunbiasedsegment_reduce__upsample_bilinear2d_aa_upsample_bicubic2d_aacdoubledoublelog_softmaxwith_dtypefloat_powerzlinalg.matrix_rankhermitianzlinalg.pinvhermitianzlinalg.pinvsingularnonzero_staticz!nn.functional.adaptive_avg_pool1dz!nn.functional.adaptive_avg_pool2dr  r  nn.functional.conv3dr  r   r!  histcznn.functional.linearmatbincount)r$  r%  r&  r'  r(  )quantilenanquantiletopkmultinomialuniform	rand_likerandintrandint_likere   
randn_like	bernoulliexponential
log_normalcauchy	geometricz-nn.functional.feature_alpha_dropoutwith_trainnormalnormalin_placenormalnumber_mean)znn.functional.alpha_dropoutznn.functional.dropoutznn.functional.dropout2dznn.functional.dropout3dz*nn.functional.multi_head_attention_forward__rpow__resize_
resize_as_addras_stridedpartial_viewsz*nn.functional.scaled_dot_product_attention)linalg.matrix_rankarangeargsortsort)	new_emptynew_empty_stridedr   rZ   
empty_liker   opdr   c                 D   > Tb  TUl         U R                  U4-   U l        g N)r
   
decorators)r|  r}  r
   s     s/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/torch/testing/_internal/common_mps.pyaddDecorator&mps_ops_modifier.<locals>.addDecorator;  s"    & +MMQD0BM    dtypesz4Skipped due to MPS not supporting complex128 tensorszSkipping empty ops.zSkipped!g,@)torch	complex64float32int8int16int32int64uint8r1   r   float16r   updater   r   namevariant_test_nameunittestexpectedFailurerW  rV  skip
complex128	complex32)r	   r
   r   r   SUPPORTED_COMPLEX_OPSMACOS_BEFORE_14_4_XFAILLISTUNIMPLEMENTED_XFAILLISTUNIMPLEMENTED_XFAILLIST_SPARSEUNDEFINED_XFAILLISTON_MPS_XFAILLISTEMPTY_OPS_SKIPLISTSKIPLISTr  r|  key	xfaillists    `              r  mps_ops_modifierr     sM    " OV!
t %//*'
#^>
^>
 &t^>
 $	^>

 d^>
 4^>
 d^>
 T^>
 T^>
 (^>
 4^>
 ^>
 ^>
 ^>
 ^>
  !^>
$ 4%^>
& 4'^>
( 4)^>
* t+^>
,  -^>
. #D/^>
0 1^>
2 D3^>
4 (5^>
6 !5==/7^>
8 EMM?9^>
: .;^>
< =^>
> d?^>
@ TA^>
B $C^>
D /



1E^>
T uzzlU^>
V DW^>
X %**Y^>
Z  



"[^>
h ti^>
j '



)k^>
x '



)y^>
F '



)G^>
T '



)U^>
b 24c^>
d 24e^>
f 'ekk(Bg^>
h  



"i^>
v '



)w^>
D 0E^>
F 0



2G^>
V 0



2W^>
f 0g^>
h ,Ti^>
j /k^>
l %dm^>
n 0



2o^>
| '



)}^>
J '



)K^>
Z '[^>
\ ']^>
^ )



+_^>
n -



/o^>
~ )



+^>
N -



/O^>
^ )



+_^>
n -



/o^>
~ !



#^>
L .tM^>
N 3

5O^>
\ 8

	:]^>
h %



'i^>
x 'y^>
z )5::,{^>
| 2EJJ<}^>
~ "4^>
@ !5::,A^>
B "4C^>
D !



#E^>
R %



'S^>
` !$a^>
b &

|c^>
d Te^>
f 4g^>
h EJJ<i^>
j $k^>
l %**m^>
n 

 o^>
| !t# $);;"<++A^>
B !t# $);;"<++G^>
H !5::,I^>
J dK^>
L tM^>
N dO^>
P %dQ^>
R &tS^>
T &tU^>
V %dW^>
X %dY^>
Z [^>
\ #D]^>
^ t_^>
` ta^>
b ,Tc^>
d ,Te^>
f g^>
h Ti^>
j U]]ENN3k^>
l 4m^>
n do^>
p Dq^>
r $s^>
t ejj\u^>
v 



!w^>
D	 ejj%++u{{EJJWE	^>
F	 ejj%++u{{EJJWG	^>
H	 



!I	^>
V	 tW	^>
X	 &}Y	^>
Z	 %u{{m[	^>
\	 t]	^>
^	 d_	^>
`	 $

&a	^>
p	 4q	^>
r	 *4s	^>
t	 #Du	^>
v	 "4w	^>
x	 dy	^>
|	 0}	^>
~	 0	^>
B
 #U[[MC
^>
D
 #U[[ME
^>
F
 #U[[MG
^>
H
 -u{{mI
^>
J
 -u{{ENN.KK
^>
L
 -/M
^>
X
 emmU^^4Y
^>
\
 #

%]
^>
j
 EKKekk5;;

Sk
^>
n
 

o
^>
@ '+"E
& 4#** $#' #**+IJ\:
 

\:
 \:
" u}}ennE#\:
$ %--G%\:
& t'\:
( D)\:
* T+\:
, $-\:
. %--G/\:
0 EMM5==%..I1\:
2 5==%--H3\:
4 u}}emmU^^D5\:
6 

	7\:
J <>K\:
T u}}emmU^^DU\:
V u}}emmU^^LW\:
X  %--!OY\:
\ , 	& ).u}}enn'U(-u}}enn'U ; U[[%++u{{Ku}}ennE ==%--H 



 (, ;o\:
B #'{{m 



 



3 7
L !%! "
$ #D
	1V 	1 	1 	1 B''B000C,,	    R ",, 	
 (  &;<16 X]]:%>x}U (# 	
 #(B $X%=%=inU 2."T)  00:? /  00 %AA P 
r  c                 H
   0 S[         R                  [         R                  /_S[         R                  [         R                  /_SS _SS _S[         R                  /_SS _S[         R                  [         R                  /_S[         R                  [         R                  /_S	S _S
[         R                  [         R                  /_SS _SS _SS _SS _S[         R                  [         R                  /_S[         R                  [         R                  /_S[         R                  [         R                  /_0 S[         R                  [         R                  /_S[         R                  /_S[         R                  /_S[         R                  /_S[         R                  [         R                  /_S[         R                  [         R                  /_S[         R                  [         R                  /_S[         R                  [         R                  /_SS _SS _S[         R                  [         R                  /_S[         R                  [         R                  /_S[         R                  [         R                  /_S[         R                  [         R                  /_S [         R                  [         R                  /_S![         R                  [         R                  /_S"[         R                  [         R                  /_E[         R                  [         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  [         R                  /S S S#.En[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /[         R                  /S$.nS [         R                  /S%.nS&[        S'[        S(S 4S) jnU  H  nUR
                  UR                  -   nXa;   a!  U" U[	        [        R                  X   S*95        Xb;   a!  U" U[	        [        R                  X&   S*95        Xc;   d  Mo  U" U[	        [        R                  X6   S*95        M     U $ )+N_segment_reducer   rT  rU  rI  r  unique_consecutiver   cdistzmasked.scattergrid_sampler_2dgrid_sampler_3digammaigammaczlinalg.solvezlinalg.solve_exzlinalg.tensorsolveaminmaxz
special.i1zspecial.i1emsortri  rj  rk  rl  r
  	nextafterfloor_divider   histogramdd	histogramr   r[   r>   )re   zsignal.windows.bartlettzsignal.windows.blackmanzsignal.windows.cosinezsignal.windows.exponentialzsignal.windows.gaussianzsignal.windows.general_cosinezsignal.windows.general_hammingzsignal.windows.hammingzsignal.windows.hannzsignal.windows.kaiserzsignal.windows.nuttallr#   	to_sparseunique)znn.functional.pairwise_distancer  r  r[  r  r   r!  )ru  r[  r|  r}  r   c                 .    U R                   U4-   U l         g r  r  r|  r}  s     r  r  +mps_ops_grad_modifier.<locals>.addDecorator      MMQD0BMr  r  )
r  r  r  r   r   r  r  r  r  r  )r	   XFAILLIST_GRADSKIPLIST_GRADr  r  r|  r  s          r  mps_ops_grad_modifierr    s   K
u}}=K
 5==%--8K
 &t	K

 %dK
 K
 )$K
 !5==%--"@K
 emmU]];K
 TK
 u}}emm<K
 tK
 tK
 dK
 tK
  U]]EMM:!K
"  #K
* !#+K
2 u}}53K
4 5==/5K
6 EMM?7K
B emm_CK
F EMM5==9GK
H 5==%--8IK
J u}}emm4KK
L %--7MK
R DSK
V WK
Z U]]EMM:[K
^ EMM5==9_K
b EMM5==9cK
f %--7gK
j 6kK
l emmU]]3mK
p U]]EMM2qK
t mmU]]3(-(-&+mm_+0==/(-.3mm_/4}}o',}}o$)MM?&+mm_',}}oMM5==1UK
\ 16%*]]O%*]]O%*]]O/4}}o/4}}o/4}}o	
 #'%*]]O
	1V 	1 	1 	1 B''B000C$ !9!9.BUV
 #RhmmMDV!WX&  009I9N & 
r  c                     1 SknS[         S[        SS 4S jnU  HC  nUR                  UR                  -   nXA;   d  M#  U" U[        [        R
                  5      5        ME     U $ )N>   amaxamingatherr  scatter__rmod__rp  r   	clamp_max	clamp_minrb  scatter_addr   r|  r}  r   c                 .    U R                   U4-   U l         g r  r  r  s     r  r  3mps_ops_error_inputs_modifier.<locals>.addDecorator  r  r  )r   r   r  r  r  r  )r	   	XFAILLISTr  r|  r  s        r  mps_ops_error_inputs_modifierr    se    
	&	1V 	1 	1 	1 B''B000CRh.F.F!GH 
 
r  c                     U $ r   )r	   r
   r   r   s       r  r  r  $  s	     
r  )mpsNF)r  collections.abcr   typingr   r  common_utilsr   opinfo.corer   r   backendsr  is_availablestrlistr1   r  r  r  r  r  r  <module>r     s    $   ' - 	>>""$$ !/3	|f|| "$s),| 	|
 
&	||w8F#3 w8H wr8F+; @P D !/3	f "$s), 	
 
&	r  