o Di@s>ddlmZddlmZmZddlmZeGdddZdS)) dataclass)ListAny)validate_block_shapec@sjeZdZUeed<eeed<eeed<eeed<dZeed<ddZ e d d edeefd d Z d S)TensorDescriptorbaseshapestrides block_shapezeropaddingcCsBt|j}t|j|ksJd|t|j|ks!Jd||dks)Jd|dks1Jdt|j}|jdvrH|jddksHJdt|j|jj j }|jdd D]}||ddksgJd qY|jD] }|dksuJd qk|jd d ksJd |j dks|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 alignedzshape must be positivez!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_bytesstride shape_dimr!h/var/www/addictedbytheproject.nl/epg/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