ȅi<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/james-whalen/.local/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 valuer9rcSSKJnJn U"U5n[[R U5(a3US:wa-[R R 5n[S/U-5$g)Nr)triton_compute_type TritonKernelz tl.float32rr)torch._inductor.codegen.tritonrErFr,r kerneltriton_tensor_ndimr0)rCr9rErF triton_typendims rconstant#ShapePropagationOpsHandler.constantSsP U)%0 ahh - -+2M88..0D!t$ $rnameindexNcgrr)rNrOrCs rstore_reduction*ShapePropagationOpsHandler.store_reduction`r src_dtypereduction_type.c[er)NotImplementedError)r9rTrUrCs r reduction$ShapePropagationOpsHandler.reductionds "!rmodecgrr)rNrOrCrZs rstore ShapePropagationOpsHandler.storemsruse_compute_typescUR$r)r)rCr9rTr^s rto_dtype#ShapePropagationOpsHandler.to_dtypess{{rrrc\SSKJn [[RU5(dS5eg)Nr)rFzdot supports Triton only)YBLOCKXBLOCK)rGrFr,r rH)rrrFs rdotShapePropagationOpsHandler.dot|s%?!((L11M3MM1#rexprcgrr)rgr9s r index_expr%ShapePropagationOpsHandler.index_exprsroffsetcg)Nrr)rNrks r load_seed$ShapePropagationOpsHandler.load_seedsrvarsizecheckwrap_negcgrr)rorprqrrs rindirect_indexing,ShapePropagationOpsHandler.indirect_indexingsrcS$)Nc[U5$r)r?)r2kwargss r8ShapePropagationOpsHandler.__getattr__..s '@'Frr)rrNs r __getattr__&ShapePropagationOpsHandler.__getattr__sFFrcondmsgcgrr)r}r~s rdevice_assert_async.ShapePropagationOpsHandler.device_assert_asyncrSrrr)NT)TT)"rrrr__doc__ staticmethodr8typesNumberr9rrLr%r/ShapeArgrQrr0rXrr\r boolr`sympyExprrerirmrtrr{rrrrrrArANs  ** 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>8LLr