+ jU#6^RIt^RIt^RIHt^RIHtHt^RIt^RI H t ^RI t ^RI Ht^RIHtHtHtHtHtHtHtHtHtHtHtHtHtHt^RIHt]]]]]]]]3t ]]]]]3t!R.]'dR.M.,t"!R R ]4t#R t$R t%R t&Rt'Rt(RRlt)Rt*R#)N)Sequence)partialwraps) TEST_CUDA)_dispatch_dtypes all_types all_types_andall_types_and_complexall_types_and_complex_andall_types_and_half complex_typesfloating_and_complex_typesfloating_and_complex_types_andfloating_typesfloating_types_andfloating_types_and_halfintegral_typesintegral_types_and)torch_to_numpy_dtype_dictcpucudac]tRt^7tRtR#)_dynamic_dispatch_dtypesN)__name__ __module__ __qualname____firstlineno____static_attributes__re/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/testing/_internal/opinfo/utils.pyrr7sr rcVR 9d\RV: 24h\'g+VR8Xd$\P!R^R7\ R 4#\ 4p\ \P\P\P4F_pV!WVR4pRpTF.pT!TP.TPO5/TPBK0 T'gKNTPT4Ka \ V4# \d#\P!RT RT 2^R7Ki;i \dRpKki;i) rrz-Expected device_type in ['cpu', 'cuda'], got zGWARNING: CUDA is not available, empty_dtypes dispatch will be returned!) stacklevelFz.WARNING: Unable to generate sample for device:z and dtype:T)rrr)AssertionErrorrwarningswarnrsetr torchboolbfloat16half RuntimeErrorinputargskwargsadd)opsample_inputs_fn device_typesupported_dtypesdtypesamples supportedsamples&&& r!get_supported_dtypesr9<s1/);K? K   9. U (++u*5::u~~uzzR &ruEG F 6<<?&++?? 9   '5S8 $$4 553  MM@ [Y^X_`   "  !  s$ C<$*D,<)D)(D), D>=D>c \P!RR4p\V4^8Xd V!RR4#\V4p\F3p\V!44V8XgKV!W3P R,4u# RpRp\ FCp\V!44pVPV4'gK,\V4pWu8gK?TpTpKE VfV!R\V44#V!\X.\\V4\V!44, 4O5!VP \\\V4\V!44, 44,4#) return_typezdispatch_fn dispatch_fn_strz()Ngr) collections namedtuplelenr'COMPLETE_DTYPES_DISPATCHrEXTENSIBLE_DTYPE_DISPATCHissubsetstrrtuple)dtypesr; set_dtypesdispatchchosen_dispatchchosen_dispatch_scoredispatch_dtypesscores& r!dtypes_dispatch_hintrKis(((8UVK 6{a2t$$VJ, xz?j (x):):T)AB B- O-hj/'' 33 O$  ($) !&O.2s6{++ @5Vs8:!>?@Cc&kC O&C DEE r cVP#N)dynamic_dtypes)r1s&r!is_dynamic_dtype_setrOs   r cRVP R\VP4P R\VP4P R2pV#)z OpInfo(z, dtypes=z, dtypesIfCUDA=z, ) )namerKrDdispatch_fn_str dtypesIfCUDA)r1fmt_strs& r!str_format_dynamic_dtyperUsYy+BII6FFGH1"//BRRST  G Nr c:aaRo\S4VV3Rl4pV#)cV\P\\P\P\P \P \P39#rM)npbool_r)uint8int8int16int32int64)r5s&r! is_integral=np_unary_ufunc_integer_promotion_wrapper..is_integralsA HH  HH GG HH HH HH   r c<\\P!4,pS!VP4'dS!VP V44#S!V4#rM)rr(get_default_dtyper5astype)xnp_dtypefnr_s& r! wrapped_fn.wrapped_fnsD-U-D-D-FG qww  ahhx() )!u r r)rfrgr_sf @r!(np_unary_ufunc_integer_promotion_wrapperrjs(   2Y r c:aa\S4RVV3Rll4pV#)aWraps a NumPy reduction operator. The wrapper function will forward dim, keepdim, mask, and identity kwargs to the wrapped function as the NumPy equivalent axis, keepdims, where, and initial kwargs, respectively. Args: f: NumPy reduction operator to wrap supports_keepdims (bool, optional): Whether the NumPy operator accepts keepdims parameter. If it does not, the wrapper will manually unsqueeze the reduced dimensions if it was called with keepdim=True. Defaults to True. Returns: Wrapped function c8V^8dQhR\P/#)rd)nptNDArray)formats"r! __annotate__/reference_reduction_numpy..__annotate__s--3;;-r c <\VP44pVPRR4pVPRR4pRV9dF\V\4'd \ V4MTpVP ^8XdVR 9dRVR&MWBR&RV9d S 'dWRR&RV9dnVPR4pVeYVP\P8wd\RVP 24hVP4P4VR&R V9dVPR 4pVepVP\PJd/VP4P\P 4pMVP4pVP4VR &S !V.VO5/VBpV'dTS 'gLVP ^8d;Vf\#\%VP 44MTp\&P(!W4pV#) dimNkeepdimFaxiskeepdimsmaskz+Expected mask.layout == torch.strided, got whereidentityinitial>rrr~)r'keyspop isinstancerrCndimlayoutr(stridedr$rnumpyr5r*tofloat32listrangerX expand_dims) rdr.r/rrtrurxrzresultfsupports_keepdimss &*, r!wrapper*reference_reduction_numpy..wrappers6;;=!jj%**Y. D= *3 9 9%*sCvv{s&::!%v!$v  !2!(:  T>::f%D;;%--/(Edkk]S#'((*"2"2"4w  zz*-H#>>U^^3'||~00?H'||~H$,NN$4y!1&t&v& ,!),$uQVV}%#C^^F0F r ri)rrrsff r!reference_reduction_numpyrs($ 1X---^ Nr cRV9d\P!VP\P4'd!VP \P 4pMT\P!VP\P 4'd VP \P4p\\P4pV!V.VO5/VB#)a9 The function will call np.prod with type as np.int64 if the input type is int or uint64 if it is uint. This is necessary because windows np.prod uses by default int32 while on linux it uses int64. This is for fixing integer overflow https://github.com/pytorch/pytorch/issues/77320 Returns: np.prod of input r5) rX issubdtyper5 signedintegerrcr^unsignedintegeruint64rprod)ar.r/rfs&*, r! prod_numpyr sf ==""2"2 3 3"A ]]177B$6$6 7 7#A "277 +B a !$ !& !!r )T)+r<r%collections.abcr functoolsrrrrX numpy.typingtypingrnr(#torch.testing._internal.common_cudar$torch.testing._internal.common_dtyperrr r r r r rrrrrrr$torch.testing._internal.common_utilsrr?r@DEVICESrr9rKrOrUrjrrrr r!rs$$ 9 K "  'fX 3 / *6Z#L !HBJ"r