PTO-BLOCK-MODEL-DISPATCH-SORTING-SCHEMA

1// PTO-UNIT: {"id":"PTO-BLOCK-MODEL-DISPATCH-SORTING-SCHEMA","surface":"block","classification":["model","dispatch","sorting-schema"],"depends_on":["PTO-BLOCK-MODEL-DISPATCH-TILE-SCHEMA","PTO-TILE-MODEL-LEGALITY-SORTING"]}2
3pure func TileOperationUsesClosedSortingSchema(4    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean5begin6    let decoded = TileOperationOfIndex(operation);7    return decoded == TileOperation_TSORT ||8           decoded == TileOperation_TMRGSORT;9end;10
11readonly func BundleSortingSourceAt(12    ordinal: integer {0..1}) => TileIndex13begin14    var current: integer {0..2} = 0;15    var selected: TileIndex = 0;16    for binding = 0 to PTO_BUNDLE_TILE_BINDING_COUNT - 1 do17        if _BundleTileBindings[[binding]].valid then18            if _BundleTileBindings[[binding]].source0_valid then19                if current == ordinal then20                    selected = BundleTileSourceIndex(21                        binding as BundleTileBindingIndex, FALSE);22                end;23                current = (current + 1) as integer {0..2};24            end;25            if _BundleTileBindings[[binding]].source1_valid then26                if current == ordinal then27                    selected = BundleTileSourceIndex(28                        binding as BundleTileBindingIndex, TRUE);29                end;30                current = (current + 1) as integer {0..2};31            end;32        end;33    end;34    return selected;35end;36
37readonly func BundleSortingDescending(38    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean39begin40    if !_BundleScalarBindings[[0]].valid then return FALSE; end;41    let slot = BundleOperationGPRInputSlot(operation, TileOperand_flag0);42    let raw = ReadScalarRegisterOperand(43        BundleOperationGPRInputSelector(slot as integer {0..2}));44    return UInt(raw) == 1;45end;46
47readonly func BundleSortingDestinationsLegal() => boolean48begin49    for binding = 0 to PTO_BUNDLE_TILE_BINDING_COUNT - 1 do50        if _BundleTileBindings[[binding]].valid &&51           _BundleTileBindings[[binding]].destination_valid &&52           (_BundleTileBindings[[binding]].destination_allocated_by_bundle ||53            !BundleTileDestinationSizeLegal(54                binding as BundleTileBindingIndex)) then55            return FALSE;56        end;57    end;58    return TRUE;59end;60
61readonly func SelectedBundleClosedSortingSchemaLegal(62    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean63begin64    if !TileOperationUsesClosedSortingSchema(operation) then return TRUE; end;65    let decoded = TileOperationOfIndex(operation);66    let tsort = decoded == TileOperation_TSORT;67    let expected_sources: integer = if tsort then 1 else 2;68    let expected_destinations: integer = if tsort then 2 else 1;69    if BundleSharedBindingCount() != 0 ||70       BundleLocalTileSourceCount() != expected_sources ||71       BundleLocalTileDestinationCount() != expected_destinations ||72       !BundleTileBindingStreamTerminated() ||73       !BundleSortingDestinationsLegal() then74        return FALSE;75    end;76
77    if tsort then78        if _BundleDimensionPresent[[1]] || _BundleDimensionPresent[[2]] then79            return FALSE;80        end;81        if _BundleDimensionPresent[[0]] &&82           UInt(_BundleDimensions[[0]]) > 64 then83            return FALSE;84        end;85    elsif _BundleDimensionPresent[[0]] ||86          _BundleDimensionPresent[[1]] ||87          _BundleDimensionPresent[[2]] then88        return FALSE;89    end;90
91    let data_type = TileDataTypeFromEncoding(92        CurrentBundleTileOperationDataTypeCode() as TileDataTypeEncoding);93    if !TileSortDataTypeSupported(data_type) ||94       CurrentBundleTileLayout() != TileLayout_RowMajor then95        return FALSE;96    end;97
98    let source_left = BundleSortingSourceAt(0);99    if !TileDescriptorLegal(source_left) ||100       _Tiles[[source_left]].storage_kind != TileStorage_Numeric ||101       _Tiles[[source_left]].data_type != data_type ||102       _Tiles[[source_left]].layout != TileLayout_RowMajor ||103       _Tiles[[source_left]].valid_rows == 0 ||104       _Tiles[[source_left]].valid_columns == 0 ||105       !TileSortSourceValuesLegal(source_left) then106        return FALSE;107    end;108    if tsort then return TRUE; end;109
110    let source_right = BundleSortingSourceAt(1);111    if !TileDescriptorLegal(source_right) ||112       _Tiles[[source_right]].storage_kind != TileStorage_Numeric ||113       _Tiles[[source_right]].data_type != data_type ||114       _Tiles[[source_right]].layout != TileLayout_RowMajor ||115       _Tiles[[source_left]].valid_rows != 1 ||116       _Tiles[[source_right]].valid_rows != 1 ||117       _Tiles[[source_right]].valid_columns == 0 ||118       !TileSortSourceValuesLegal(source_right) then119        return FALSE;120    end;121    let output_columns =122        _Tiles[[source_left]].valid_columns +123        _Tiles[[source_right]].valid_columns;124    if output_columns > 32768 then return FALSE; end;125
126    let descending = BundleSortingDescending(operation);127    return TileSortSequenceOrdered(source_left, descending) &&128           TileSortSequenceOrdered(source_right, descending);129end;130