PTO-TILE-MODEL-LEGALITY-INDEXED-REARRANGEMENT

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-LEGALITY-INDEXED-REARRANGEMENT","surface":"tile","classification":["model","legality","indexed-rearrangement"],"depends_on":["PTO-TILE-MODEL-LEGALITY-DTYPE-LAYOUT"]}2
3pure func InstructionContractValueDataTypeLegal_TGATHER(4    data_type: TileDataType) => boolean5begin6    return IndexedTLSUTransferDataTypeLegal(data_type);7end;8
9pure func InstructionContractIndexDataTypeLegal_TGATHER(10    data_type: TileDataType) => boolean11begin12    return data_type == TileDataType_S32 ||13           data_type == TileDataType_U32 ||14           data_type == TileDataType_S16 ||15           data_type == TileDataType_U16 ||16           data_type == TileDataType_S64 ||17           data_type == TileDataType_U64;18end;19
20pure func InstructionContractTypePairLegal_TSCATTER(21    value_type: TileDataType,22    index_type: TileDataType) => boolean23begin24    return IndexedTLSUTransferDataTypeLegal(value_type) &&25           (index_type == TileDataType_S16 ||26            index_type == TileDataType_U16 ||27            index_type == TileDataType_S32 ||28            index_type == TileDataType_U32 ||29            index_type == TileDataType_S64 ||30            index_type == TileDataType_U64);31end;32
33pure func TileIndexedRowIsNegative(34    value: Word,35    data_type: TileDataType) => boolean36begin37    if data_type == TileDataType_S16 then38        return SInt(value[15:0]) < 0;39    end;40    if data_type == TileDataType_S32 then41        return SInt(value[31:0]) < 0;42    end;43    if data_type == TileDataType_S64 then44        return SInt(value[63:0]) < 0;45    end;46    return FALSE;47end;48
49pure func TileIndexedRowValue(50    value: Word,51    data_type: TileDataType) => integer52begin53    if data_type == TileDataType_S16 ||54       data_type == TileDataType_U16 then55        return UInt(value[15:0]);56    end;57    if data_type == TileDataType_S32 ||58       data_type == TileDataType_U32 then59        return UInt(value[31:0]);60    end;61    assert data_type == TileDataType_S64 ||62           data_type == TileDataType_U64;63    return UInt(value[63:0]);64end;65
66readonly func TileGatherReferencesLegal(67    source: TileIndex,68    indices: TileIndex) => boolean69begin70    let source_tile = _Tiles[[source]];71    let index_tile = _Tiles[[indices]];72    for row = 0 to index_tile.valid_rows - 1 looplimit 65536 do73        for column = 0 to index_tile.valid_columns - 1 looplimit 65536 do74            let index_element = TileLogicalLinearIndex(75                index_tile,76                row as integer {0..65535},77                column as integer {0..65535});78            let raw_index = TileReadLogicalElement(index_tile, index_element);79            if TileIndexedRowIsNegative(raw_index, index_tile.data_type) then80                return FALSE;81            end;82            let source_row = TileIndexedRowValue(83                raw_index,84                index_tile.data_type);85            if source_row >= source_tile.valid_rows then86                return FALSE;87            end;88            let source_element = TileLogicalLinearIndex(89                source_tile,90                source_row as integer {0..65535},91                column as integer {0..65535});92            if !TileLogicalElementDefined(source_tile, source_element) then93                return FALSE;94            end;95        end;96    end;97    return TRUE;98end;99
100readonly func TileScatterReferencesLegal(101    destination: TileIndex,102    indices: TileIndex) => boolean103begin104    let destination_tile = _Tiles[[destination]];105    let index_tile = _Tiles[[indices]];106    var selected = Zeros{524288};107    for row = 0 to index_tile.valid_rows - 1 looplimit 65536 do108        for column = 0 to index_tile.valid_columns - 1 looplimit 65536 do109            let index_element = TileLogicalLinearIndex(110                index_tile,111                row as integer {0..65535},112                column as integer {0..65535});113            let raw_index = TileReadLogicalElement(index_tile, index_element);114            if TileIndexedRowIsNegative(raw_index, index_tile.data_type) then115                return FALSE;116            end;117            let destination_row = TileIndexedRowValue(118                raw_index,119                index_tile.data_type);120            if destination_row >= destination_tile.valid_rows then121                return FALSE;122            end;123            let destination_element = TileLogicalLinearIndex(124                destination_tile,125                destination_row as integer {0..65535},126                column as integer {0..65535});127            if selected[destination_element] == '1' then128                return FALSE;129            end;130            selected[destination_element] = '1';131        end;132    end;133    return TRUE;134end;135
136readonly func TileOperandsLegal_TGATHER(137    destination: TileIndex,138    source: TileIndex,139    indices: TileIndex) => boolean140begin141    if !TileDescriptorLegal(destination) ||142       !TileDescriptorLegal(source) ||143       !TileDescriptorLegal(indices) then144        return FALSE;145    end;146    let destination_tile = _Tiles[[destination]];147    let source_tile = _Tiles[[source]];148    let index_tile = _Tiles[[indices]];149    if destination_tile.storage_kind != TileStorage_Numeric ||150       source_tile.storage_kind != TileStorage_Numeric ||151       index_tile.storage_kind != TileStorage_Numeric ||152       !InstructionContractValueDataTypeLegal_TGATHER(153           source_tile.data_type) ||154       !InstructionContractIndexDataTypeLegal_TGATHER(155           index_tile.data_type) ||156       destination_tile.data_type != source_tile.data_type ||157       destination_tile.valid_rows == 0 ||158       destination_tile.valid_columns == 0 ||159       destination_tile.valid_rows != index_tile.valid_rows ||160       destination_tile.valid_columns != index_tile.valid_columns ||161       source_tile.valid_columns < destination_tile.valid_columns ||162       !TileSourceContentsDefined(indices) ||163       !TileSourceEncodingsValid(indices) then164        return FALSE;165    end;166    return TileGatherReferencesLegal(source, indices);167end;168
169readonly func TileOperandsLegal_TSCATTER(170    destination: TileIndex,171    source: TileIndex,172    indices: TileIndex) => boolean173begin174    if !TileDescriptorLegal(destination) ||175       !TileDescriptorLegal(source) ||176       !TileDescriptorLegal(indices) then177        return FALSE;178    end;179    let destination_tile = _Tiles[[destination]];180    let source_tile = _Tiles[[source]];181    let index_tile = _Tiles[[indices]];182    if destination_tile.storage_kind != TileStorage_Numeric ||183       source_tile.storage_kind != TileStorage_Numeric ||184       index_tile.storage_kind != TileStorage_Numeric ||185       destination_tile.data_type != source_tile.data_type ||186       !InstructionContractTypePairLegal_TSCATTER(187           source_tile.data_type,188           index_tile.data_type) ||189       source_tile.valid_rows == 0 ||190       source_tile.valid_columns == 0 ||191       source_tile.valid_rows != index_tile.valid_rows ||192       source_tile.valid_columns != index_tile.valid_columns ||193       destination_tile.valid_rows == 0 ||194       destination_tile.valid_columns != source_tile.valid_columns ||195       !TileSourceContentsDefined(source) ||196       !TileSourceContentsDefined(indices) ||197       !TileSourceEncodingsValid(indices) then198        return FALSE;199    end;200    return TileScatterReferencesLegal(destination, indices);201end;202