Ei<SSKrSSKJrJr SSKJrJrJr SSKrSSK r SSK J r J r \\\\ \4r"SS\5r\\\ R"R$\\ \ R&4r\R*"S5S\S \S \4S j5rS \\S \4S jr"SS5rg)N)CallableSequence)OptionalProtocolUnion)OpsValueVc,\rSrSr\S\4Sj5rSrg)ShapeVarreturncgN)selfs ^/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_inductor/shape_propagation.pyshapeShapeVar.shapes'*rN)__name__ __module__ __qualname____firstlineno__propertyBlockShapeTyper__static_attributes__rrrr r s *~**rr abrc $^[U[5(de[U[5(de[U5[U5:a*[U/S/[U5[U5- -QUQ75$[U5[U5:a,Xp[U/S/[U5[U5- -QUQ75$S[[ [ 4S[[ [ 4S[[ [ 44Sjm[U4Sj[X555$)Nrd1d2rc~[U5S:XaU$[U5S:XaU$[U5[U5:XdeU$)N1)str)r!r"s r_get_broadcasted_dim3get_broadcasted_shape.._get_broadcasted_dim%s?2w#~ RC r7c"g% %%Irc38># UHupT"X5v M g7frr).0r!r"r&s r (get_broadcasted_shape../sJ fb)"11 s) isinstancerlenget_broadcasted_shaperintr%tuplezip)rrr&s @rr.r.s a " "" " a " "" " 1vA$Q(F1#Q#a&*A(FA(FGG Q#a&1$Q(F1#Q#a&*A(FA(FGG c3h %*38_  38_ JA JJJrargscSnUHn[US5(a8URnUc gUc [U5nM5[U[U55nML[ U[ [ 45(a UcSnMnMp[ U[R5(aMSSK J nJ n [ X%U[45(a g[S[U535e U$)Nrrr)LoopBody LoopBodyBlockzUnknown type: )hasattrrr0r.r,r/floattorchdtypetorch._inductor.loop_bodyr4r5r TypeErrortype)r2 result_shapeargrr4r5s rbroadcast_shapes_for_argsr?2s#'L 3 IIE}%$U| 4\5<P c5\ * *#! $ U[[ ) )  I#xBCCnT#YK89 9), rc\rSrSrSr\S\RRS\RS\ 4Sj5r \S\ S\ S\SS 4S j5r\S\RS \RS \ S\\\\S 44S\\ \\ S 444 Sj5r\S$S\ S\ S\S\\ SS 4 Sjj5r\S%S\S\RS \\RS\S\ 4 Sjj5r\S\R2S\R2S\ 4Sj5r\S\R2S\RS\ 4Sj5r\S\ S\ S\ 4Sj5r\S&S\S\\R2\ 4S\S\SS 4 Sjj5rS\ S\S \ 44Sjr\S \S!\ SS 4S"j5r S#r!g )'ShapePropagationOpsHandlerNz% Propagate shape from args to output valuer9rcSSKJn [[RU5(a-[RR 5n[ S/U-5$g)Nr TritonKernelrr)torch._inductor.codegen.tritonrFr,r kerneltriton_tensor_ndimr0)rCr9rFndims rconstant#ShapePropagationOpsHandler.constantSs? @ ahh - -88..0D!t$ $rnameindexNcgrr)rMrNrCs rstore_reduction*ShapePropagationOpsHandler.store_reduction^r src_dtypereduction_type.c[er)NotImplementedError)r9rSrTrCs r reduction$ShapePropagationOpsHandler.reductionbs "!rmodecgrr)rMrNrCrYs rstore ShapePropagationOpsHandler.storeksruse_compute_typescUR$r)r)rCr9rSr]s rto_dtype#ShapePropagationOpsHandler.to_dtypeqs{{rrrc\SSKJn [[RU5(dS5eg)NrrEzdot supports Triton only)YBLOCKXBLOCK)rGrFr,r rH)rrrFs rdotShapePropagationOpsHandler.dotzs%?!((L11M3MM1#rexprcgrr)rfr9s r index_expr%ShapePropagationOpsHandler.index_exprsroffsetcg)Nrr)rMrjs r load_seed$ShapePropagationOpsHandler.load_seedsrvarsizecheckwrap_negcgrr)rnrorprqs rindirect_indexing,ShapePropagationOpsHandler.indirect_indexingsrcS$)Nc[U5$r)r?)r2kwargss r8ShapePropagationOpsHandler.__getattr__..s '@'Frr)rrMs r __getattr__&ShapePropagationOpsHandler.__getattr__sFFrcondmsgcgrr)r|r}s rdevice_assert_async.ShapePropagationOpsHandler.device_assert_asyncrRrrr)NT)TT)"rrrr__doc__ staticmethodr8typesNumberr9rrKr%r/ShapeArgrPrr0rWrr[r boolr_sympyExprrdrhrlrsrrzrrrrrrArANs **5;;>c#h4"{{";;""XuXs]334 " ~u^S%899 : ""FJ&.6>sm  ,0"& {{EKK(   $uzz$ejj$^$$ EKKNS^  EJJO$   GGn1D(EG(rrA) functoolscollections.abcrrtypingrrrrr8 virtualizedr r r/r%rr rrr9r lru_cacher.r?rArrrrs.,, $(5c?34+x+ 5;;--sHekkI J  TK^KK>KK.HX$6>8JJr