PTO-BLOCK-MODEL-DISPATCH-COMPARISON-SCHEMA

1// PTO-UNIT: {"id":"PTO-BLOCK-MODEL-DISPATCH-COMPARISON-SCHEMA","surface":"block","classification":["model","dispatch","comparison-schema"],"depends_on":["PTO-BLOCK-MODEL-DISPATCH-TILE-SCHEMA"]}2pure func TileOperationUsesClosedTCMPSchema(3    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean4begin5    return TileOperationOfIndex(operation) == TileOperation_TCMP;6end;7
8pure func TileOperationUsesClosedTSELSchema(9    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean10begin11    return TileOperationOfIndex(operation) == TileOperation_TSEL;12end;13
14pure func TileOperationUsesClosedComparisonSchema(15    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean16begin17    return TileOperationUsesClosedTCMPSchema(operation) ||18           TileOperationUsesClosedTSELSchema(operation);19end;20
21readonly func SelectedBundleComparisonDimensionsLegal() => boolean22begin23    if !_BundleDimensionPresent[[0]] ||24       UInt(_BundleDimensions[[0]]) < 1 ||25       UInt(_BundleDimensions[[0]]) > 65535 then26        return FALSE;27    end;28    for dimension = 1 to 2 looplimit 2 do29        if _BundleDimensionPresent[[dimension]] &&30           (UInt(_BundleDimensions[[dimension]]) < 1 ||31            UInt(_BundleDimensions[[dimension]]) > 65535) then32            return FALSE;33        end;34    end;35    return TRUE;36end;37
38readonly func SelectedBundleComparisonShapeMatches(39    source: TileIndex) => boolean40begin41    let valid_columns = UInt(_BundleDimensions[[0]]);42    let valid_rows = if _BundleDimensionPresent[[1]] then43        UInt(_BundleDimensions[[1]]) else 1;44    let columns = if _BundleDimensionPresent[[2]] then45        UInt(_BundleDimensions[[2]]) else valid_columns;46    return _Tiles[[source]].valid_rows == valid_rows &&47           _Tiles[[source]].valid_columns == valid_columns &&48           _Tiles[[source]].columns == columns;49end;50
51readonly func SelectedBundleClosedTCMPSchemaLegal(52    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean53begin54    if !TileOperationUsesClosedTCMPSchema(operation) then return TRUE; end;55    if BundleTileBindingCount() != 1 ||56       BundleSharedBindingCount() != 0 ||57       _BundleScalarBindings[[0]].valid ||58       !SelectedBundleComparisonDimensionsLegal() then59        return FALSE;60    end;61
62    let binding = _BundleTileBindings[[0]];63    if !binding.destination_valid ||64       binding.destination_allocated_by_bundle ||65       !BundleTileDestinationSizeLegal(0) ||66       !binding.source0_valid ||67       !binding.source1_valid ||68       !binding.last then69        return FALSE;70    end;71    if UInt(_BundleDataAttributes.comparison_mode) > 5 then72        return FALSE;73    end;74
75    let source_left = BundleTileSourceIndex(0, FALSE);76    let source_right = BundleTileSourceIndex(0, TRUE);77    let data_type = TileDataTypeFromEncoding(78        CurrentBundleTileOperationDataTypeCode() as TileDataTypeEncoding);79    return TileCompareDataTypeSupported(data_type) &&80           TileShapeAndTypeMatch(source_left, source_right) &&81           _Tiles[[source_left]].storage_kind == TileStorage_Numeric &&82           _Tiles[[source_left]].data_type == data_type &&83           _Tiles[[source_left]].layout == TileLayout_RowMajor &&84           TileSourceContentsDefined(source_left) &&85           TileSourceContentsDefined(source_right) &&86           TileSourceEncodingsValid(source_left) &&87           TileSourceEncodingsValid(source_right) &&88           SelectedBundleComparisonShapeMatches(source_left);89end;90
91readonly func SelectedBundleClosedTSELSchemaLegal(92    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean93begin94    if !TileOperationUsesClosedTSELSchema(operation) then return TRUE; end;95    if BundleTileBindingCount() != 2 ||96       BundleSharedBindingCount() != 0 ||97       _BundleScalarBindings[[0]].valid ||98       !SelectedBundleComparisonDimensionsLegal() then99        return FALSE;100    end;101
102    let inputs = _BundleTileBindings[[0]];103    let result = _BundleTileBindings[[1]];104    if inputs.destination_valid ||105       !inputs.source0_valid ||106       !inputs.source1_valid ||107       inputs.last ||108       !result.destination_valid ||109       result.destination_allocated_by_bundle ||110       !BundleTileDestinationSizeLegal(1) ||111       !result.source0_valid ||112       result.source1_valid ||113       !result.last then114        return FALSE;115    end;116
117    let mask = BundleTileSourceIndex(0, FALSE);118    let source_true = BundleTileSourceIndex(0, TRUE);119    let source_false = BundleTileSourceIndex(1, FALSE);120    let data_type = TileDataTypeFromEncoding(121        CurrentBundleTileOperationDataTypeCode() as TileDataTypeEncoding);122    return TileSelectDataTypeSupported(data_type) &&123           TilePredicateValuesLegal(mask) &&124           TileShapeAndTypeMatch(source_true, source_false) &&125           _Tiles[[source_true]].storage_kind == TileStorage_Numeric &&126           _Tiles[[source_true]].data_type == data_type &&127           _Tiles[[source_true]].layout == TileLayout_RowMajor &&128           TileSourceContentsDefined(source_true) &&129           TileSourceContentsDefined(source_false) &&130           TileLogicalShapeMatch(mask, source_true) &&131           SelectedBundleComparisonShapeMatches(source_true);132end;133