Ë ÄÓ^j@(ã ó—UdZddlmZddlmZddlmZddlmZddlm Z e edzZee d <d edee fd„Zd edeefd „Zd edeedffd„Zdededefd„Zdedee dffd„Zde de fd„Zdede fd„Zdedede fd„Zdede fd„Zdededefd„Zdededefd„Zd*dededefd„Zd+deded edzdefd!„Z d+d"edzded edzde fd#„Z d+d"ed$ed%edzdefd&„Zd"ee zdzd'ee zdee zfd(„Zdee zdzdefd)„Z y),z& Functions for manipulating IntTuples é)Úreduce)Úchain)Ú TypeAlias)ÚTypeIsé)ÚInteger)ÚIntTuple.r ÚxÚreturncó"—t|t«S©N)Ú isinstancer©r s ún/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/distributed/_pycute/int_tuple.pyÚis_intr1s€ÜaœÓ!Ð!ócó"—t|t«Sr )rÚtuplers rÚis_tupler5s€ÜaœÓÐr.có"—t|«r|fS|Sr )rrs rÚas_tupler9s€Ü ˆa„yØˆtˆØ€HrÚaÚbcóÈ—t|«rt|«ryt|«r@t|«r5t|«t|«k(xrtd„t ||«D««Sy)NTc3ó:K—|]\}}t||«–—Œywr )Úmatch_structure©Ú.0r Úys rú z"match_structure..Csèø€Ò'TÁ$À!ÀQ¬¸¸1×(=Ñ'Tùó‚F)rrÚlenÚallÚzip©rrs rrr?sL€Ü ˆa„y”V˜A”YØÜ„{”x ”{Ü1‹vœ˜Q›ÑÒT¤CÑ'TÌ#ÈaÐQRË)Ô'TÓ$TÐTØrÚtcó`—t|«r!t|«dk(rytd„|D««S|fS)Nr©c3ó@K—|]}t|«D]}|–—ŒŒywr )Úflatten)rrÚis rr zflatten..Ls èø€Ò:˜q¬w°q«zÒ:¨!œÐ:˜Ñ:ùs‚)rr"r)r&s rr*r*Gs/€Ü„{Üˆq‹6QŠ;ØäÑ: AÔ:Ó:Ð:àˆtˆrcó<—t|dkD«t|dk«z S)Nr)Úbool©rs rÚsignumr/Qs€ÜA‘‹;œ˜a !™e›Ñ$Ð$rcó8—t|«rtd„|d«S|S)Ncó—|t|«zSr )Úproduct)ÚvalÚelems rúzproduct..Ws€¨¬g°d«mÑ(;€rr)rrr.s rr2r2Us€Ü„{ÜÑ;¸QÀÓBÐBàˆrcóâ—t|«rDt|«r9t|«t|«k7rt‚td„t ||«D««St|«st|«rt‚||zS)Nc3ó:K—|]\}}t||«–—Œywr )Ú inner_productrs rr z inner_product..`sèø€Ò=©4¨1¨a”= A×&Ñ=ùr!)rr"ÚAssertionErrorÚsumr$r%s rr8r8\sX€Ü„{”x ”{Üˆq‹6”S˜“VÒÜ Ð ÜÑ=´3°q¸!³9Ô=Ó=Ð=äAŒ;œ( 1œ+Ü Ð Ø1‰uˆrcó@—t|«rtd„|D««S|S)Nc3ó2K—|]}t|«–—Œywr )Ú tuple_max)rr s rr ztuple_max..isèø€Ò+ A”9˜Q—<Ñ+ùó‚)rÚmaxr.s rr=r=gs€Ü„{ÜÑ+¨Ô+Ó+Ð+àˆrcó—t|«rOt|«r9t|«t|«k7rt‚td„t ||«D««Std«‚t|«rt|t |««S||zS)Nc3ó:K—|]\}}t||«–—Œywr )Ú elem_scalers rr zelem_scale..ssèø€Ò@©d¨a°œ A q×)Ñ@ùr!z#Invalid combination: tuple with int)rr"r9rr$rBr2r%s rrBrBnsj€Ü„{ÜAŒ;Ü1‹vœ˜Q›ÒÜ$Ð$ÜÑ@´c¸!¸Q³iÔ@Ó@Ð@ä Ð!FÓGÐGäAŒ;Ü˜a¤¨£Ó,Ð,àq‘5ˆLrcóª—t|«rˆt|«r9t|«t|«k7rt‚td„t ||«D««Sg}|D]2}|jt ||««t |t|««}Œ4t|«St|«rt |t|««S||zdk(s||zdk(st‚||zdz |zS)Nc3ó:K—|]\}}t||«–—Œywr )Ú shape_divrs rr zshape_div..ƒsèø€Ò?©T¨Q°œ 1 aŸÑ?ùr!rr)rr"r9rr$ÚappendrEr2)rrÚrÚvs rrErE~sÆ€Ü„{ÜAŒ;Ü1‹vœ˜Q›ÒÜ$Ð$ÜÑ?´S¸¸A³YÔ?Ó?Ð?ðˆAØò -Ø—‘œ 1 a›Ô)Ü˜a¤¨£Ó,‘ð -ô˜“8ˆOäAŒ;Ü˜Q¤¨£ Ó+Ð+à˜‘E˜Q’J ! a¡%¨1¢*Ü$Ð$Ø˜‘E˜A‘I !Ñ#Ð#rÚinitcó¤—t|«r®t|«r9t|«t|«k7rt‚td„t ||«D««Sg}tt|«dz dd«D]1}|j t|||««|t||«z}Œ3|j«t|«St|«rtd«‚|S)Nc3ó:K—|]\}}t||«–—Œywr )Úsuffix_product)rr r+s rr z!suffix_product..›sèø€ÒG±$°!°Qœ¨¨1×-ÑGùr!réÿÿÿÿz(Invalid combination: int with tuple init) rr"r9rr$ÚrangerFrLr2Úreverse)rrIrGr+s rrLrL•s»€ä„{ÜDŒ>Ü1‹vœ˜T›Ò"Ü$Ð$ÜÑG¼#¸aÀ»,ÔGÓGÐGð ˆAôœ3˜q›6 A™: r¨2Ó.ò ,Ø—‘œ¨¨!©¨dÓ3Ô4Øœg a¨¡d›mÑ+‘ð ,ð I‰IŒKÜ˜“8ˆOäDŒ>Ü Ð!KÓLÐLàˆKrNÚidxÚshapeÚstridecó‡—|€t|«}t‰«rrt|«r\t|«rQt‰«t|«k(rt|«t|«k(st‚t d„t‰||«D««Std«‚t|«rFt|«r;t|«t|«k7rt‚t ˆfd„t||«D««St|«st|«rt‚‰|z|zS)Nc3ó>K—|]\}}}t|||«–—Œywr ©Úidx2crd)rr+ÚsÚds rr zidx2crd..¹sèø€ÒQ©g¨a°°Aœ A q×)ÑQùó‚z*Invalid combination: tuple with int stridec3ó>•K—|]\}}t‰||«–—Œywr rU)rrWrXrPs €rr zidx2crd..Àsøèø€ÒK±°°1œ a¨×+ÑKùsƒ)rLrr"r9rr$)rPrQrRs` rrVrV±sÑø€Ø €~Ü Ó&ˆä„}ÜEŒ?œx¨Ô/Ü˜“H¤ E£ Ò*¬s°6«{¼cÀ%»jÒ/HÜ$Ð$ÜÑQ¼¸SÀ%ÈÓ9PÔQÓQÐQä Ð!MÓNÐNäEŒ?œx¨Ô/Ü5‹zœS ›[Ò(Ü$Ð$ÜÓK¼¸EÀ6Ó8JÔKÓKÐKä˜Œ¤(¨6Ô"2Ü$Ð$Ø˜6‘M UÑ*Ð*rÚcrdc óÊ—|€t|«}t|«rxt|«r\t|«rQt|«t|«k(rt|«t|«k(st‚t d„t|||«D««Std|›d|›«‚|€d}t|«r¤t|«r™t|«t|«k7rt‚d}t t|«dz dd«D]8}|t|t||«z||||«z }|t||«z}Œ:t|«dkDr|t||d|d«z }|St|«st|«rt‚||zS)Nc3ó>K—|]\}}}t|||«–—Œywr )Úcrd2idx)rÚcrWrXs rr zcrd2idx..Ñsèø€ÒO©G¨A¨q°!”w˜q ! Q×'ÑOùrYzInvalid combination: crd=z, shape=rrrM) rLrr"r9r:r$rNr^r2)r[rQrRÚresultr+s rr^r^ÇsZ€ð€~Ü Ó&ˆä„}ÜEŒ?œx¨Ô/Ü˜“H¤ E£ Ò*¬s°6«{¼cÀ%»jÒ/HÜ$Ð$ÜÑO´s¸3ÀÀvÓ7NÔOÓOÐOä Ð#<¸S¸EÀÈ%ÈÐ!QÓRÐRàˆ;ØˆCäEŒ?œx¨Ô/Ü5‹zœS ›[Ò(Ü$Ð$ØˆFäœ3˜u›:¨™>¨1¨bÓ1ò /Øœ' #¬°°a±Ó(9Ñ"9¸5À¹8ÀVÈAÁYÓOÑOØœW U¨1¡XÓ.Ñ.‘ð /ô5‹z˜AŠ~Øœ' # u¨Q¡x°¸±Ó;Ñ;ØˆMä˜Œ¤(¨6Ô"2Ü$Ð$Ø˜‘<ÐrÚ dst_shapeÚ src_shapecó—t|«rXt|«r9t|«t|«k7rt‚td„t ||«D««S|€t‚t||«St|«rt ||«S||k\rt‚|S)Nc3ó:K—|]\}}t||«–—Œywr )Úcrd2crdrs rr zcrd2crd..ñsèø€ÒG©4¨1¨aœ AŸÑGùr!)rr"r9rr$r^rV)r[rarbs rrereês„€ô„}ÜIÔÜ3‹xœ3˜y›>Ò)Ü$Ð$ÜÑG´3°s¸IÓ3FÔGÓGÐGðÐ Ü$Ð$Ü˜3 Ó*Ð*äIÔÜ˜3 Ó*Ð*àiÒÜ$Ð$ØˆJrÚtrgcó—t|«rvt|«r`t|«t|«k7rt‚tt td„t ||«Dcgc]\}}t||«‘Œc}}«Ž«Std«‚|€|fSycc}}w)Ncó—|dk7S)Nr(r(rs rr5zslice_.. s € ! r¡'€rz+Invalid combination: tuple crd with int trgr()rr"r9rrÚfilterr$Úslice_)r[rfr_rWs rrjrjs‡€Ü„}ÜCŒ=Ü3‹xœ3˜s›8Ò#Ü$Ð$äÜÜÙ)Ü25°c¸3³-×@©$¨!¨Qœ 1Ó@óðóð ô!Ð!NÓOÐOØ ˆàˆvˆ àùóAsÁBcóD—t|«rtd„|D««S|duS)Nc3ó2K—|]}t|«–—Œywr )Úhas_none)rrHs rr zhas_none..sèø€Ò* 1”8˜A—;Ñ*ùr>)rÚanyr.s rrmrms$€Ü„{ÜÑ*¨Ô*Ó*Ð*àDˆyÐr)rr )!Ú__doc__Ú functoolsrÚ itertoolsrÚtypingrÚtyping_extensionsrrÚintrr Ú__annotations__Úobjectrrrr-rr*r/r2r8r=rBrErLrVr^rerjrmr(rrúrws#ðòBõÝÝÝ$åð˜E /Ñ2Ñ2€ˆ)Ó2ð"ˆfð"˜ ™ó"ð ð ˜6 %™=ó ð ð ˜U 8¨S =Ñ1ó ðxð Hð°óðˆxð˜E # s (™Oóð%ˆcð%có%ðˆxð˜CóðXð (ð¨sóðð˜cóð(ð˜xð¨Hóð $ð$˜hð$¨8ó$ñ.hð hð°xóñ8+ð+ (ð+°H¸t±Oð+Èxó+ð.FJñ Ø D‰ð Ø!)ð Ø3;¸d±?ð àó ðHFJñØ ðØ&ðØ3;¸d±?ðà óð.˜‘˜dÑ"ð¨°©ð¸À¹óð0˜‘˜dÑ"ð tôr