K idZddlmZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtddlmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~mZmZmZmZmZmZmZddlmZmZmZmZmZmZmZmZmZmZddlmZgd Zd Zy ) zisort:skip_file)math)extra)argmaxargmin bitonic_mergecdivcumprodcumsumflip interleavemaxminravel reduce_orsigmoidsoftmaxsortsum swizzle2dtopkxor_sumzeros zeros_like)Z PropagateNanTRITON_MAX_TENSOR_NUMELload_tensor_descriptorstore_tensor_descriptormake_tensor_descriptortensor_descriptortensor_descriptor_typeaddadvancearangeassociative_scanassume async_task atomic_add atomic_and atomic_cas atomic_max atomic_min atomic_or atomic_xchg atomic_xorbfloat16 block_type broadcast broadcast_tocatcastclamp conditionconst constexprconstexpr_type debug_barrier device_assert device_printdot dot_scaleddtype expand_dimsfloat16float32float64 float8e4b15 float8e4nv float8e4b8float8e5 float8e5b16fullgather histograminline_asm_elementwiseint1int16int32int64int8joinloadmake_block_ptrmap_elementwise max_constancymax_contiguousmaximumminimum multiple_of num_programspermutepi32_t pointer_type program_idrangereducereshapeslicesplit static_assert static_print static_rangestoretensortranstuple tuple_typeuint16uint32uint64uint8viewvoidwhere)umulhiexpexp2fmaloglog2cosrsqrtsinsqrtsqrt_rnabsfdivdiv_rnerffloorceil) pair_uniform_to_normalphilox philox_implrandrand4xrandint randint4xrandnrandn4xuint_to_uniform_float) target_info)rrrrrrrr!r"r#rrr$r%r&r'r(r)r*r+r,r-r.r/rr0r1r2r3r4rrr5r6r7r8r9rzr r r:r;r<rr=r>r?rrurvr@rrr rArBrCrDrErFrGrHrrwrIrJrKrLr rMrNrOrPrQrRrSrxryrTrUrr rVrWrXrrYrZr[rr\rrr]r^r_rrrrrrr`rrarrbr{rcrr|rrrdr}r~rerfrgrhrrrrirrjrkrmrnrorprrtrqrrrsrrrc :ddlm}t||rIt|jj dd}t |Dcgc]}t||c}|S|ddk(r/|dd}d}|ddk(r|dd}d}t||}t|| S|jd rB|jd djd }|jd d\}} | jdd\} } | jdjdD cgc]} t| j} } | jd} t| } t|d}t| }t t g|z}t t"g|z}t%|| }| r;ddlm}ddlm}t/| t1|} t| |sJ||||| St-|||S|jdr t3|Sidt4dt6dt8dt:dt<dt>dt@dtBdtDdtFdtHdtJd t d!t"d"tFd#tLd$tNtPtRtFd%}||Scc}wcc} w)&N)rk_fields*rFkT) element_tyr7 tensordesc<>[)maxsplit],)NVMMASharedLayout)r r8fp8e4nvfp8e4b8fp8e5fp8e5b16fp8e4b15fp16bf16fp32fp64i1i8i16i32i64u1u8u16)u32u64B)*builtinsrk isinstancetype__dict__getrl str_to_tyr^ startswithrdrstripintstriplstriplenrOrPr0+triton.experimental.gluon.language._layoutsr4triton.experimental.gluon.language.nvidia.hopper.tmar evaldictr9rErFrGrHrDrAr/rBrCrMrQrNrprmrnro)namecrkfieldsxr7tyinnerr?rest block_shapeslayoutis_gluonndim shape_type stride_typeblockrgluon_tensor_descriptor_typetyss ^/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/triton/language/__init__.pyrrs$d$$((D9D9q9Q?96BB Aw#~ABx 7c>8DE tQ r77 |$ 3"))#.kk#k2 t JJsQJ7 T/:/A/A#/F/L/LS/QR!s1779~R RS!v;%&;$/  5'D.2 5+.  U D&$9J"KLFf&78 88/z;PVW W%eZEE {#a   : :    K   K            d  d  u  u  u  d  e! " v# $ ) C, t9s:Ss J JN)__doc__rrstandardrrrrr r r r r rrrrrrrrrrrrcorerrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrPrQrRrSrTrUrVrWrXrYrZr[r\r]r^r_r`rarbrcrdrerfrgrhrirjrkrlrmrnrorprqrrrsrtrurvrwrxryrzr{r|r}r~rrrrrrrandomrrrrrrrrrrr__all__rrrs.[[[[[[[[[[[[[[[[[[[[[[[x   L ^>r