o ܹi@s>ddlmZddlmZmZddlmZeGdddZdS)) dataclass)ListAny)validate_block_shapec@sjeZdZUeed<eeed<eeed<eeed<dZeed<ddZ e d d edeefd d Z d S)TensorDescriptorbaseshapestrides block_shapezeropaddingcCs&t|j}t|j|ksJd|t|j|ks!Jd||dks)Jd|dks1Jdt|j}|jdvrH|jddksHJdt|j|jj j }|jdd D]}||ddksgJd qY|jd d kssJd |j d ks|j dksJd|j dkr|jj j sJddSdS)Nzrank mismatch: rzrank must not be zerozrank cannot be more than 5) FakeTensorFunctionalTensorzbase must be 16-byte alignedzstrides must be 16-byte alignedz!Last dimension must be contiguousr nanzIllegal value for paddingzAPadding option `nan` is only supported for floating point tensors) lenrr r typer__name__data_ptrrdtypeitemsizer is_floating_point)selfrankty elem_bytesstrider [/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/triton/tools/tensor_descriptor.py __post_init__s$      zTensorDescriptor.__post_init__tensorcCst||j|||S)N)rrr)r#r r r r r! from_tensor szTensorDescriptor.from_tensorN)r ) r __module__ __qualname__r__annotations__rintr strr" staticmethodr$r r r r!rs     rN) dataclassesrtypingrr triton._utilsrrr r r r!s