a kh @s>ddlmZddlmZmZddlmZeGdddZdS)) dataclass)ListAny)validate_block_shapec@sZeZdZUeed<eeed<eeed<eeed<ddZeeeeddd Z d S) TensorDescriptorbaseshapestrides block_shapecCst|j}t|j|ks&Jd|t|j|ksBJd||dksRJd|dksbJdt|j}|jd|j}|dvr|jddksJd t |j|jj j }|jdd D]}||ddksJd q|jd d ksJd 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_bytesstriderL/var/www/auris/lib/python3.9/site-packages/triton/tools/tensor_descriptor.py __post_init__ s    zTensorDescriptor.__post_init__Ztensorr cCst||j||S)N)rrrrrrr from_tensors zTensorDescriptor.from_tensorN) rr __qualname__r__annotations__rintr staticmethodrrrrrrs    rN)Z dataclassesrtypingrrZ triton._utilsrrrrrrs