(Thj8SSKJr SSKJr SSKJr SSKJr SSj555r\R\\R,"SSS9S?Sj555r\S 5r\S!5r\S"5r\S#5r\R\\R,"S$SSS9S>S%j555r\R\\R,"S&SS9S?S'j555r\S(5rS@S*jr \R\\R,"S+S)S,9SASBS-jj555r!\S.5r"\R\\R,"S/5SCS0j555r#\R\\RH"S15SDS2j555r%\S35r&\R\\RH"S45SDS5j555r'\SES6j5r(\SFS7j5r)\R\S\RT4SGS8jj55r+S9r,\R\SHS:j55r-\S;5r.g)I) annotations)jit)core)mathczSnURnUS:aUS-nUS- nUS:aM[R"U5$)Nrrvaluer constexpr)ilog2ns P/var/www/auris/envauris/lib/python3.13/site-packages/triton/language/standard.py_log2r sC D A a% a   a% >>$ cnURn[R"XS- -S:H=(a US:g5$)Nrrr )r rs r_is_power_of_twors. A >>1A;1,7a 88rcX-S- U-$)z Computes the ceiling division of :code:`x` by :code:`div` :param x: the input number :type x: Block :param div: the divisor :type div: Block r)xdivs rcdivrs GaKC rsigmoidc<SS[R"U*5-- $)Nr)rexp)rs rrr+s DHHaRL !!rsoftmaxcU[US5- n[R"U5n[US5n[R"X4U5$)Nr)maxrrsumfdiv)r ieee_roundingznumdens rrr2s< C1I A ((1+C c1+C 99S} --rcB[R"XR/US9$)z^ Returns a contiguous flattened view of :code:`x`. :param x: the input tensor :type x: Block ) can_reorder)rreshapenumel)rr's rravelr*<s <<GG9+ >>rc~X-U-nXC-nXV-nXt-n[R"X(- U5nXV-nXU--n XT-n X4$)a Transforms the indices of a row-major `size_i * size_j` matrix into the indices of a column-major matrix for each group of `size_g` rows. For example, for :code:`size_i = size_j = 4` and :code:`size_g = 2`, it will transform :: [[0 , 1 , 2 , 3 ], [4 , 5 , 6 , 7 ], [8 , 9 , 10, 11], [12, 13, 14, 15]] into :: [[0, 2, 4 , 6 ], [1, 3, 5 , 7 ], [8, 10, 12, 14], [9, 11, 13, 15]] rminimum) r jsize_isize_jsize_gijsize_gjgroup_idoff_inew_inew_js r swizzle2dr8Hs[, aBoG}H  E \\&.& 1F B K E LE <rc2[R"USU5$)a Returns a tensor filled with the scalar value 0 for the given :code:`shape` and :code:`dtype`. :param shape: Shape of the new array, e.g., (8, 16) or (8, ) :type shape: tuple of ints :param dtype: Data-type of the new array, e.g., :code:`tl.float16` :type dtype: DType r)rfull)shapedtypes rzerosr=ps 99UAu %%rcB[URUR5$)z{ Returns a tensor of zeros with the same shape and type as a given tensor. :param input: input tensor :type input: Tensor )r=r;r<)inputs r zeros_liker@}s ekk **rcU(aX:H=(a X:nOSnX:=(d Un[R"X`U5n[R"XaU5nXx4$NFrwhere) value1index1value2index2tie_break_lefttiegtv_reti_rets r_argmax_combinerNsM26?  CB JJr6 *E JJr6 *E <rc[XX#S5$NTrNrErFrGrHs r_argmax_combine_tie_break_leftrS 664 @@rc[XX#S5$rBrQrRs r_argmax_combine_tie_break_fastrV 665 AArc.[R"X5$N)rmaximumabs r_elementwise_maxr^ << rrZreturn_indicesreturn_indices_tie_break_left)return_indices_arg tie_break_argNc[R"U5nU(a;U(a[R"X[US9$[R"X[US9$[R "UR R5[R "S5:a[R "UR R55(a UR[R5nOEUR R5(dS5eUR[R5n[R"X[US9$N keep_dims z"Expecting input to be integer type)r_promote_bfloat16_to_float32_reduce_with_indicesrSrVr r<primitive_bitwidth is_floatingtofloat32is_intint32reducer^r?axisr`rargs rrrs  - -e 4E (,,U:Xdmn n,,U:Xdmn n >>%++88 9DNN2>%++88 9B >~~ekk55788.{{))++Q-QQ+,{{5(8INNrz minimum indexc [XSX#S9upEU$ru)rrvs rargminrsTQ_ uFA Jrc X-$rYrr[s r _sum_combiner 5Lrr<c$[R"U5nUbU$SnUR5(a&URS:a[RnU$SnU$UR 5(a"URS:a[R OSnU$)Nrh)r_unwrap_if_constexpr is_int_signed int_bitwidthrpis_int_unsigneduint32)in_dtyper< out_dtypes r_pick_sum_dtypers  % %e ,E  I"*"7"7""># rc 8[R"[UR[ XR555 [R"[UR 55 [ UR 5n[ UR 5[ UR[ XR55- n[R"URRSS9n[R"URUSS9S/U-5n[R"XS5n[R"SS5SS2S4S[R"SS5- :Hn[R"X25HhnUn[R"SUS-5H*n X:wdM XS-:wdM[R"X5nM, [XX-US-SURS9nMj [R"XPR5RURSS9nU$) z Flips a tensor `x` along the dimension `dim`. :param x: the first input tensor :type x: Block :param dim: the dimension to flip along (currently only final dimension supported) :type dim: int TrrrrNr)rgr<)rrrr;rr)rrr<rkr(rm expand_dimsrrr ) rrstepsstartrrrr flip2r.s rrrs ' c770K(LMN'01"!''NE!!''NU177=gg;V3W-XXE   )C)CD QF QTT&$T/!u=A "A KK1 ag &!dkk!Q.?*? ?D   u ,""1eai0Av!1u*((21  1q5D @ - Q ##AGGT#:A Hrc[R"X5n[UR5S:XaU$[R"X"RSSSURS-/-5$)a Interleaves the values of two tensors along their last dimension. The two tensors must have the same shape. Equivalent to `tl.join(a, b).reshape(a.shape[:-1] + [2 * a.shape[-1]])` :param a: The first input tensor. :type a: Tensor :param b: The second input tensor. :type b: Tensor rNr)rjoinrr;r()r\r]cs r interleaversU !A 177|q ||Awws|q1772;.??@@r)r core.constexpr)F)NFTF)TF)rrr<r)NFN)r<rrB)rF)r rrr)rrrrrr)rrrrrY)/ __future__r runtime.jitrrrrr_tensor_member_fnr_add_math_1arg_docstrrrr*r8r=r@rNrSrVr^_add_reduction_docstrrryr~rrrrrrrr rr_add_scan_docstrrrrrr CONSTEXPR_0rrrrrrrrsS"  9    I&"'"I&.'.??$$N & &++AABBI:J*IKOKO"O;KLM  AABBI:J*IKOKO"O;KLM  EW5G6GI&G'Gx E!Ey!F"F))&  0"&TEUEU  0  <AAr