h JddlmZddlmZmZddlmZeGddZy)) dataclass)ListAny)validate_block_shapecneZdZUeed<eeed<eeed<eeed<dZededeefdZ y) TensorDescriptorbaseshapestrides block_shapect|j}t|j|k(s Jd|t|j|k(s Jd||dkDsJd|dksJdt |j }|j d|j}|dvr'|j jdzdk(sJd t|j|j jj}|jdd D]}||zdzdk(rJd |jd d k(sJd y)Nzrank mismatch: rzrank must not be zerozrank cannot be more than 5.)ztorch.FakeTensorztorch.FunctionalTensorzbase must be 16-byte alignedzstrides must be 16-byte alignedz!Last dimension must be contiguous) lenr r r typer __module____name__data_ptrrdtypeitemsize)selfrankty type_name elem_bytesstrides \/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/triton/tools/tensor_descriptor.py __post_init__zTensorDescriptor.__post_init__ sG4::4<< D(BOD6*BB(4##$,Fv.FF,ax000xqy666y $))_}}oQr{{m4 J J99%%'",1 Q3Q Q1T--.YY__-- ll3B' VFZ'2-2 U4U U2 V||B1$I&II$tensorcNt||j|j|S)N)rr r)r#r s r from_tensorzTensorDescriptor.from_tensors%  LL MMO    r"N) rr __qualname__r__annotations__rintr! staticmethodr%r"r rrsQ I 9 #YcJ  C d3i  r"rN) dataclassesrtypingrr triton._utilsrrr*r"r r.s'!.     r"