+ "ݳj/P^RIt^RIHtRRlt!RR]P4tR#)N)tree_mapc<V^8dQhR\R\R\/#)sizealignment_valuereturn)int)formats"_/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchao/swizzle/swizzle_tensor.py __annotate__r s&CCSC3C3Cc>^V^, V,,V,#)rrs&&r _get_min_alignmentr s $(. /? BBr ca]tRt^toRtRV3RlRlltRRltRtRtRt Rt ] R 4t R t ]RR l4t]P"P$tR tVtR #) SwizzleTensorz A Python-only swizzled tensor subclass. Intended usage of this abstraction: Swizzle weight Tensor to avoid LDS use during GEMMs on ROCm hardware. c:<V^8dQhRS[PRS[/#)roriginalshallow)torchTensorbool)r __classdict__s"r r SwizzleTensor.__annotate__s#99,,99r cp\P!VRR7p\PPW4#)meta)device)r empty_liker_make_subclass)clsrrwrappers&&& r __new__SwizzleTensor.__new__s+ ""8F;||**388r cV'dR#VP^8XgQR4hVP^8XgVP^8XgQhVP^8Xd^ M^@pVP^8Xd^M^pVP^8XdVPwrV^pVP^8XdVPwrup\X^4p\XV4p W, p W, p \P P PV^V ^V 3R^4p VP^8Xd8V PV^,^W,^V4p V P^^^^^4p VP^8Xd:V PXV^,^W,^V4p V P^^^^^^4p V P4Vn XVn WPn W`nWnWnWnWnVPVnRVnR#)Nz"SwizzleTensor only supports ndim 2constantF)ndimitemsizeshaperrnn functionalpadviewpermute contiguousxBMKalignedMalignedKpaddedMpaddedK original_ndim is_transposed) selfrrkdivlastdimr3r4r2r5r6r7r8r1s &&& r __init__SwizzleTensor.__init__ s  }}!G#GG!  A%):):a)???&&!+r((A-!2 ==A >>DAA ==A nnGA!%a,%a.,, HH   # #Hq'1g.F TU V ==A x2~r8+;QHA !Q1a(A ==A q(b."h.>7KA !Q1a+A    %]]"r cVVPP RVP4 R2#)z (original=)) __class____name__ unswizzler;s&r __repr__SwizzleTensor.__repr__Cs(..))**T^^5E4FaHHr cBRpVP^8XdVPP^^^^^4P4pVP VP VP 4pV^VP1^VP13,pVP VPVP4pVP'd VPpVP^8XdVPP^^^^^^4P4pVP VPVP VP 4pV^VP1^VP1^VP13,pVP VPVPVP4pV#N) r9r1r/r0reshaper5r6r3r4r:Tr2)r;undones& r rDSwizzleTensor.unswizzleFs9    "VV^^Aq!Q2==?F^^DMM4==AFAJDFF 23F^^DFFDFF3F!!!    "VV^^Aq!Q15@@BF^^DFFDMM4==IFAJDFF AJ>?F^^DFFDFFDFF;F r cVP^8XdQVPPVPVP4pVP 'd VP pV#VP^8Xd\VPPVPVPVP4pVP 'd VP pV#R#)rN)r9r1rJr5r6r:rKr2)r;tmps& r as_tensorSwizzleTensor.as_tensorVs    "&&.. >C!!!eeJ    "&&.. FC!!!eeJ #r c nVP^8XdVPVP3M"VPVPVP33p\ \ P !VRVPRVPRR/R4pVPVn VPVnVPVnVPVnVPVn VPVn VPVn VPVn VPVnVP'*VnV#)rdtypelayoutrrT)r9r3r4r2rremptyrSrTr1r5r6r7r8r:)r;r*new_objs& r shallow_transposeSwizzleTensor.shallow_transposecs $ 2 2a 7TVVTVV dffdffdff=U  KK Tdjj T TV T  FF FF FF FF ====,,,, $ 2 2$($6$6 6r c\P!VP'dVPVP34#VPVP34#rI)rSizer:r4r3rEs&r r*SwizzleTensor.shapews>zzd.@.@.@466466*WWtvvtvvFVWWr c^VP'd^VP3#VP^3#r)r:r4rEs&r strideSwizzleTensor.stride{s("000466{AtvvqkAr Nc ^RIHpW9dWQ,!WV4#RpRp\Wq!\Wc4/\Wd4B4#)r)SWIZZLE_OPS_TABLEcR\V\4'dVP4#T#rI) isinstancerrDes&r unwrap0SwizzleTensor.__torch_dispatch__..unwraps$.q-$@$@1;;= Ga Gr c\\V\P4'd \V4#T#rI)rbrrrrcs&r wrap.SwizzleTensor.__torch_dispatch__..wraps"'1!U\\'B'B=# I Ir )torchao.swizzle.swizzle_opsr`r)r"functypesargskwargsr`rerhs&&&&& r __torch_dispatch__ SwizzleTensor.__torch_dispatch__~sM B  $$*4v> > H JdHV$:Whv>VWXXr ) r2r4r3r6r5r:r9r8r7r1)FrI)rC __module__ __qualname____firstlineno____doc__r$r>rFrDrPrWpropertyr*r] classmethodror_C_disabled_torch_function_impl__torch_function____static_attributes____classdictcell__)rs@r rrss99!#FI (XXB Y Y ??r r)rtorch.utils._pytreerrrrrr r r}s' (C@ELL@r