PTO-TILE-MODEL-LEGALITY-INDEXED-LAYOUT

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-LEGALITY-INDEXED-LAYOUT","surface":"tile","classification":["model","legality","indexed-layout"],"depends_on":["PTO-TILE-MODEL-LEGALITY-OPERAND-SCHEMA"]}2readonly func TileOperandsLegal_TFMA(3    destination: TileIndex, source_left: TileIndex,4    source_right: TileIndex, addend: TileIndex) => boolean5begin6    return TileElementwiseDescriptorLegal(destination) &&7           TileElementwiseSourceContentsDefined(source_left) &&8           TileElementwiseSourceContentsDefined(source_right) &&9           TileElementwiseSourceContentsDefined(addend) &&10           TileFusedMultiplyAddDataTypeSupported(11               _Tiles[[destination]].data_type) &&12           TileElementwiseShapeAndTypeMatch(destination, source_left) &&13           TileElementwiseShapeAndTypeMatch(destination, source_right) &&14           TileElementwiseShapeAndTypeMatch(destination, addend) &&15           _Tiles[[destination]].data_type == _Tiles[[source_left]].data_type &&16           _Tiles[[destination]].data_type == _Tiles[[source_right]].data_type &&17           _Tiles[[destination]].data_type == _Tiles[[addend]].data_type &&18           TileElementwiseLayoutSupported(_Tiles[[destination]].layout) &&19           (!TileDataTypeIsFloating(_Tiles[[destination]].data_type) ||20            (TileElementwiseSourceEncodingsValid(source_left) &&21             TileElementwiseSourceEncodingsValid(source_right) &&22             TileElementwiseSourceEncodingsValid(addend)));23end;24
25pure func TileHistogramSourceDataTypeSupported(26    data_type: TileDataType) => boolean27begin28    return data_type == TileDataType_U16 ||29           data_type == TileDataType_U32;30end;31
32pure func TileHistogramSelectedByteSupported(33    data_type: TileDataType,34    selected_byte: integer {0..3}) => boolean35begin36    if data_type == TileDataType_U16 then37        return selected_byte <= 1;38    end;39    return data_type == TileDataType_U32;40end;41
42readonly func TileHistogramPrefixDefined(43    filter: TileIndex,44    required_rows: integer {0..3}) => boolean45begin46    if required_rows == 0 then return TRUE; end;47    if _Tiles[[filter]].valid_rows < required_rows ||48       _Tiles[[filter]].valid_columns < 1 then49        return FALSE;50    end;51    for row = 0 to required_rows - 1 do52        if !TileElementDefined(53               filter,54               row as integer {0..65535},55               0) then56            return FALSE;57        end;58    end;59    return TRUE;60end;61
62readonly func TileHistogramInputsLegal(63    source: TileIndex,64    filter: TileIndex,65    selected_byte: integer {0..3}) => boolean66begin67    if !TileSourceContentsDefined(source) ||68       !TileDescriptorLegal(filter) ||69       _Tiles[[source]].storage_kind != TileStorage_Numeric ||70       _Tiles[[filter]].storage_kind != TileStorage_Numeric ||71       _Tiles[[source]].layout != TileLayout_RowMajor ||72       _Tiles[[filter]].data_type != TileDataType_U8 then73        return FALSE;74    end;75
76    let source_type = _Tiles[[source]].data_type;77    if !TileHistogramSelectedByteSupported(78           source_type,79           selected_byte) then80        return FALSE;81    end;82    if source_type == TileDataType_U16 then83        if selected_byte == 1 then return TRUE; end;84        if _Tiles[[filter]].valid_rows < _Tiles[[source]].valid_rows ||85           _Tiles[[filter]].valid_columns < 1 then86            return FALSE;87        end;88        for row = 0 to _Tiles[[source]].valid_rows - 1 looplimit 65536 do89            if !TileElementDefined(90                   filter,91                   row as integer {0..65535},92                   0) then93                return FALSE;94            end;95        end;96        return TRUE;97    elsif source_type == TileDataType_U32 then98        let required_rows: integer {0..3} = if selected_byte == 0 then 399            else if selected_byte == 1 then 2100            else if selected_byte == 2 then 1101            else 0;102        return TileHistogramPrefixDefined(filter, required_rows);103    end;104    return FALSE;105end;106
107readonly func TileOperandsLegal_THISTOGRAM(108    destination: TileIndex, source: TileIndex, filter: TileIndex,109    selected_byte: integer {0..3}) => boolean110begin111    if destination == source || destination == filter ||112       !TileDescriptorLegal(destination) ||113       !TileHistogramInputsLegal(source, filter, selected_byte) then114        return FALSE;115    end;116    let result = _Tiles[[destination]];117    return result.storage_kind == TileStorage_Numeric &&118           result.data_type == TileDataType_U32 &&119           result.layout == TileLayout_RowMajor &&120           result.valid_rows == _Tiles[[source]].valid_rows &&121           result.valid_columns == 256 &&122           result.rows >= result.valid_rows &&123           result.columns >= 256;124end;125
126readonly func TileOperandsLegal_GMOV(127    destination: TileIndex, source: TileIndex, peer_tid: Word) => boolean128begin129    return UInt(peer_tid) < 4 &&130           TileDescriptorLegal(destination) &&131           TileSourceContentsDefined(source) &&132           TileLogicalShapeMatch(destination, source) &&133           TileCarrierOrPackedBaselineDataTypeSupported(134               _Tiles[[source]].data_type) &&135           _Tiles[[destination]].data_type == _Tiles[[source]].data_type &&136           _Tiles[[destination]].layout == _Tiles[[source]].layout &&137           _Tiles[[destination]].location != TileLocation_Memory &&138           _Tiles[[destination]].location != TileLocation_Matrix &&139           _Tiles[[source]].location != TileLocation_Memory &&140           _Tiles[[source]].location != TileLocation_Matrix;141end;142