PTO-TILE-MODEL-ORDERING-SORTING

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-ORDERING-SORTING","surface":"tile","classification":["model","ordering","sorting"],"depends_on":["PTO-TILE-MODEL-EXECUTION-COMPARISON"]}2
3pure func TileSortDataTypeSupported(4    data_type: TileDataType) => boolean5begin6    return data_type == TileDataType_FP32 ||7           data_type == TileDataType_FP16;8end;9
10pure func TileSortValueIsSignalingNaN(11    data_type: TileDataType,12    value: Word) => boolean13begin14    return TileNumericValueClass(data_type, value) ==15        NumericValue_SignalingNaN;16end;17
18pure func TileSortLeftBefore(19    left: Word,20    right: Word,21    descending: boolean,22    data_type: TileDataType) => boolean23begin24    let left_class = TileNumericValueClass(data_type, left);25    let right_class = TileNumericValueClass(data_type, right);26    let left_nan = NumericValueClassIsNaN(left_class);27    let right_nan = NumericValueClassIsNaN(right_class);28
29    // Every numeric value precedes every quiet NaN in both directions.30    // Two NaNs retain their incoming order.31    if left_nan then32        return right_nan;33    elsif right_nan then34        return TRUE;35    end;36
37    let both_zero =38        NumericValueClassIsZero(left_class) &&39        NumericValueClassIsZero(right_class);40    if both_zero || left == right then41        return TRUE;42    end;43
44    let left_key = TileFloatingOrderKey(data_type, left);45    let right_key = TileFloatingOrderKey(data_type, right);46    if descending then47        return UInt(left_key) > UInt(right_key);48    end;49    return UInt(left_key) < UInt(right_key);50end;51
52readonly func TileSortSourceValuesLegal(53    source: TileIndex) => boolean54begin55    return TileSourceContentsDefined(source) &&56           TileSourceEncodingsValid(source);57end;58
59readonly func TileSortSourceHasSignalingNaN(60    source: TileIndex) => boolean61begin62    let source_tile = _Tiles[[source]];63    for row = 0 to source_tile.valid_rows - 1 looplimit 65536 do64        for column = 0 to source_tile.valid_columns - 1 looplimit 65536 do65            let element = TileLogicalLinearIndex(66                source_tile,67                row as integer {0..65535},68                column as integer {0..65535});69            if TileSortValueIsSignalingNaN(70                   source_tile.data_type,71                   TileReadLogicalElement(source_tile, element)) then72                return TRUE;73            end;74        end;75    end;76    return FALSE;77end;78
79readonly func TileSortSequenceOrdered(80    source: TileIndex,81    descending: boolean) => boolean82begin83    let source_tile = _Tiles[[source]];84    if source_tile.valid_columns <= 1 then85        return TRUE;86    end;87
88    for column = 0 to source_tile.valid_columns - 2 looplimit 65536 do89        let left_element = TileLogicalLinearIndex(90            source_tile,91            0,92            column as integer {0..65535});93        let right_element = TileLogicalLinearIndex(94            source_tile,95            0,96            (column + 1) as integer {0..65535});97        if !TileSortLeftBefore(98               TileReadLogicalElement(source_tile, left_element),99               TileReadLogicalElement(source_tile, right_element),100               descending,101               source_tile.data_type) then102            return FALSE;103        end;104    end;105    return TRUE;106end;107