o h @s>ddlmZddlmZmZddlmZeGdddZdS)) dataclass)ListAny)validate_block_shapec@s\eZdZUeed<eeed<eeed<eeed<ddZededeefdd Z d S) TensorDescriptorbaseshapestrides block_shapecCst|j}t|j|ksJd|t|j|ks!Jd||dks)Jd|dks1Jdt|j}|jd|j}|dvrP|jddksPJd t |j|jj j }|jdd D]}||ddksoJd qa|jd d ks{Jd dS)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) lenrr r typer __module____name__Zdata_ptrrZdtypeitemsize)selfZrankty type_nameZ elem_bytesstriderd/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/triton/tools/tensor_descriptor.py __post_init__ s    zTensorDescriptor.__post_init__tensorcCst||j||S)N)rrr)rr rrr from_tensors zTensorDescriptor.from_tensorN) rr __qualname__r__annotations__rintr staticmethodrrrrrrs    rN) dataclassesrtypingrrZ triton._utilsrrrrrrs