PTO-TILE-MODEL-EXECUTION-GENERATION

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-EXECUTION-GENERATION","surface":"tile","classification":["model","execution","generation"],"depends_on":["PTO-TILE-MODEL-EXECUTION-EXPANSION","PTO-TILE-MODEL-LEGALITY-LAYOUT-REARRANGEMENT"]}2// PTO-REQ-TEPL-GENERATE-001: generated sequences, masks, and padding.3
4pure func TileTCIDataTypeSupported(data_type: TileDataType) => boolean5begin6    return data_type == TileDataType_S32 ||7           data_type == TileDataType_S16 ||8           data_type == TileDataType_U32 ||9           data_type == TileDataType_U16;10end;11
12pure func TileTTRIDataTypeSupported(data_type: TileDataType) => boolean13begin14    return data_type == TileDataType_FP32 ||15           data_type == TileDataType_FP16 ||16           data_type == TileDataType_S32 ||17           data_type == TileDataType_S16 ||18           data_type == TileDataType_U32 ||19           data_type == TileDataType_U16;20end;21
22pure func TileTTRIOneEncoding(data_type: TileDataType) => Word23begin24    case data_type of25        when TileDataType_FP32 =>26            return Zeros{PTO_XLEN} + 0x3f800000;27        when TileDataType_FP16 =>28            return Zeros{PTO_XLEN} + 0x3c00;29        when TileDataType_BF16 =>30            return Zeros{PTO_XLEN} + 0x3f80;31        otherwise =>32            return Zeros{PTO_XLEN} + 1;33    end;34end;35
36func TCI(destination: TileIndex, start: Word, descending: boolean)37begin38    var result = _Tiles[[destination]];39    assert result.allocated;40    assert result.valid_rows == 1;41    assert TileTCIDataTypeSupported(result.data_type);42    let normalized_start = TileRawElementValue(43        start,44        result.data_type);45    for column = 0 to result.valid_columns - 1 looplimit 65536 do46        let element = TileLogicalLinearIndex(47            result,48            0,49            column as integer {0..65535});50        let offset = NaturalToWord(column as integer {0..65535});51        let value = if descending then52            normalized_start - offset53        else54            normalized_start + offset;55        result = TileInfoWithLogicalElement(result, element,56            TileRawElementValue(57            value,58            result.data_type));59    end;60    result = TileWithValidRegionDefined(result);61    result = TileWithPadding(result, TilePad_Null);62    result.location = TileLocation_Any;63    _Tiles[[destination]] = result;64end;65
66func TTRI(destination: TileIndex, upper: boolean,67          diagonal: integer {-65535..65535})68begin69    var result = _Tiles[[destination]];70    assert result.allocated;71    assert result.valid_rows >= 1;72    assert result.valid_columns >= 1;73    assert TileTTRIDataTypeSupported(result.data_type);74    let one = TileTTRIOneEncoding(result.data_type);75    for row = 0 to result.valid_rows - 1 looplimit 65536 do76        for column = 0 to result.valid_columns - 1 looplimit 65536 do77            let boundary: integer = row + diagonal;78            let selected = if upper then79                column >= boundary80            else81                column <= boundary;82            let element = TileLogicalLinearIndex(83                result,84                row as integer {0..65535},85                column as integer {0..65535});86            result = TileInfoWithLogicalElement(result, element,87                if selected then88                one89            else90                Zeros{PTO_XLEN});91        end;92    end;93    result = TileWithValidRegionDefined(result);94    result = TileWithPadding(result, TilePad_Null);95    result.location = TileLocation_Any;96    _Tiles[[destination]] = result;97end;98