biSSKJr SSKJrJrJr SSKJr SSKJ s J s J s J r SSKJrJr SSKJrJr \(a SSKJr SSKJr /S Qr\"S S 9"S S \ R.55r\"SS\ R255r\SSSjj5r\SSSjj5r\SSSjj5r\SSSjj5rg)) annotations)ListTuple TYPE_CHECKING) dataclassN)PaddedSharedLayoutSwizzledSharedLayout)builtin_unwrap_if_constexpr)ir)shared_memory_descriptor) async_load async_waitmake_tensor_descriptortensor_descriptortensor_descriptor_typeT)eqct\rSrSr%SrS\S'S\S'S\S'S\S 'SS jrSS jrSS jrSS jr SSjr Sr g)rz!The type for a tensor descriptor.zttgl.block_type block_typezttgl.tuple_type shape_type strides_type)PaddedSharedLayout | SwizzledSharedLayoutlayoutc<SURSURS3$)Nztensor_descriptor)rrselfs l/home/james-whalen/.local/lib/python3.13/site-packages/triton/experimental/gluon/language/amd/gfx1250/tdm.py__str__tensor_descriptor_type.__str__s #DOO#4Bt{{m1EEcXnUS- nURRX5upBURRX5upR[X4XP5nXb4$)N)r _unflatten_irrr)rhandlescursorhandleshapestridesvalues rr%$tensor_descriptor_type._unflatten_irsT! 55gF ++99'J!&?}r"cURRR5nURURR U5UUR R U55$N)r element_ty is_int_signed!get_tensor_descriptor_layout_typeto_irr_to_ir)rbuilder is_signeds rr3tensor_descriptor_type._to_ir$sTOO..<<> 88 OO ! !' *  KK  w '  r"cURURU55 URRX5 URRX5 gr.)appendr3r_flatten_ir_typesr)rr4outs rr9(tensor_descriptor_type._flatten_ir_types,s= 4;;w'( ))'7 ++G9r"c SURR5SURR5SURR5SURR5S3 $)NTD_)rmanglerrrrs rr?tensor_descriptor_type.mangle1sbDOO**,-Qt/E/E/G.H$J[J[JbJbJdIeefgkgrgrgygyg{f||~r"N)returnstr)r&List[ir.value]r'intrBzTuple[tensor_descriptor, int])r4 ir.builderrBzir.type)r4rFr:z List[ir.type]rBNone) __name__ __module__ __qualname____firstlineno____doc____annotations__r r%r3r9r?__static_attributes__rAr"rrrs8+!! 55F : @r"rc\rSrSr%SrS\S'S\S'S\S'S\S 'SS jr\S 5r\S 5r \S 5r \S5r Sr g)r5z4A descriptor representing a tensor in global memory.zir.valuer(z ttgl.tupler)r*rtypecURUR5 URRU5 URRU5 gr.)r8r(r) _flatten_irr*)rr&s rrStensor_descriptor._flatten_ir>s6t{{# w'   )r"c.URR$r.)rQrrs rrtensor_descriptor.block_typeCsyy###r"cBURRR$r.)rQrr)rs r block_shapetensor_descriptor.block_shapeGsyy##)))r"cBURRR$r.)rQrr/rs rdtypetensor_descriptor.dtypeKsyy##...r"c.URR$r.)rQrrs rrtensor_descriptor.layoutOsyyr"rAN)r&rDrBrG) rHrIrJrKrLrMrSpropertyrrXr[rrNrAr"rrr5sr>     * $$**//  r"rc[U5nSUs=::aS::d O SUS35e[U5U:XdSUS[U535e[U5U:XdSUS[U535e[UR[R5(dS 5e[ U5n[U[ [45(dS 5e[U[5(aURS:XdS 5eURnURUS S 9nURUSS 9n [R"U5n[R"U5n[R"URRU5n [XRURU5n UR!S5n UR"R%U R'UR"5XxX5n [)XX+5$)aMake a tensor descriptor object. Args: base (tensor): base pointer of the tensor in global memory. shape (List[int]): shape of the tensor. strides (List[int]): strides of the tensor. block_shape (List[int]): block shape of the tensor. layout (PaddedSharedLayout | SwizzledSharedLayout): the layout of the tensor in shared memory. Returns: tensor_descriptor: the created tensor descriptor object r$z Expected 1 <= ndim <= 5 but got z dimensionsz Expected z strides but got zExpected block_shape to have z dimensions but got zExpected base to be a pointerzBExpected layout to be a PaddedSharedLayout or SwizzledSharedLayoutz3Expected max_phase to be 1 for SwizzledSharedLayoutF require_i64Tzero)len isinstancer[ttgl pointer_typer rr max_phaser(_convert_to_ir_valuestuplerrQr/r_str_to_padding_optionr4create_make_tensor_descriptorr3r)baser)r*rXr _semanticndim base_handle shape_handlesstride_handlesrrQpaddingr(s rrrTs u:D >>O=dV;OO> w<4 R9TF2CCL>!RR  { t #m'DTFJ^_bcj_k^l%mm # djj$"3"3 4 4U6UU 4 !& )F f13GH I IMLM I&.//1$[&[[$++K33Eu3MM44W$4ON JJu Ejj!G!5!5{CJ !*jj',, OD..v6G    < rs"--!77`RQ o d!@T^^!@!@H     < Y](;$F(;Ui(;#L(;ar(; (;V W[MM,DM`dM M(  e#' e  e = =r"