PTO-BLOCK-MODEL-DISPATCH-EXPANSION-SCHEMA

1// PTO-UNIT: {"id":"PTO-BLOCK-MODEL-DISPATCH-EXPANSION-SCHEMA","surface":"block","classification":["model","dispatch","expansion-schema"],"depends_on":["PTO-BLOCK-MODEL-DISPATCH-COMPARISON-SCHEMA","PTO-TILE-MODEL-LEGALITY-REDUCTION-AND-EXPANSION"]}2
3pure func TileOperationUsesClosedRowExpansionSchema(4    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean5begin6    let decoded = TileOperationOfIndex(operation);7    return decoded == TileOperation_TROWEXPAND ||8           decoded == TileOperation_TROWEXPANDADD ||9           decoded == TileOperation_TROWEXPANDSUB ||10           decoded == TileOperation_TROWEXPANDMUL ||11           decoded == TileOperation_TROWEXPANDDIV ||12           decoded == TileOperation_TROWEXPANDMAX ||13           decoded == TileOperation_TROWEXPANDMIN ||14           decoded == TileOperation_TROWEXPANDEXPDIF;15end;16
17pure func TileOperationUsesClosedColumnExpansionSchema(18    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean19begin20    let decoded = TileOperationOfIndex(operation);21    return decoded == TileOperation_TCOLEXPAND ||22           decoded == TileOperation_TCOLEXPANDADD ||23           decoded == TileOperation_TCOLEXPANDSUB ||24           decoded == TileOperation_TCOLEXPANDMUL ||25           decoded == TileOperation_TCOLEXPANDDIV ||26           decoded == TileOperation_TCOLEXPANDMAX ||27           decoded == TileOperation_TCOLEXPANDMIN ||28           decoded == TileOperation_TCOLEXPANDEXPDIF;29end;30
31pure func TileOperationUsesClosedExpansionSchema(32    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean33begin34    return TileOperationUsesClosedRowExpansionSchema(operation) ||35           TileOperationUsesClosedColumnExpansionSchema(operation);36end;37
38pure func TileExpansionOperationIsCopy(39    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean40begin41    let decoded = TileOperationOfIndex(operation);42    return decoded == TileOperation_TROWEXPAND ||43           decoded == TileOperation_TCOLEXPAND;44end;45
46pure func TileExpansionOperationIsExponentialDifference(47    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean48begin49    let decoded = TileOperationOfIndex(operation);50    return decoded == TileOperation_TROWEXPANDEXPDIF ||51           decoded == TileOperation_TCOLEXPANDEXPDIF;52end;53
54readonly func SelectedBundleExpansionExponentialDifferenceTypes()55    => (boolean, TileDataType, TileDataType)56begin57    let default_type = TileDataType_FP64;58    if !_BundleOperation.data_type_valid ||59       !BundleDataTypeConcrete(_BundleOperation.data_type) then60        return (FALSE, default_type, default_type);61    end;62    let source_type = BundleTileDataType(_BundleOperation.data_type);63    var destination_type = source_type;64    if _BundleDataAttributesPresent then65        if _BundleDataAttributes.data_type == DTYPE_NONE then66            destination_type = source_type;67        elsif !BundleDataTypeConcrete(_BundleDataAttributes.data_type) then68            return (FALSE, default_type, default_type);69        else70            destination_type = BundleTileDataType(71                _BundleDataAttributes.data_type);72        end;73    end;74    return (75        TileExpandExpdifTypePairLegal(source_type, destination_type),76        source_type,77        destination_type);78end;79
80readonly func SelectedBundleExpansionBroadcastShapeMatches(81    operation: integer {0..PTO_TILE_OPERATION_COUNT-1},82    broadcast: TileIndex) => boolean83begin84    let valid_columns = UInt(_BundleDimensions[[0]]);85    let valid_rows = if _BundleDimensionPresent[[1]] then86        UInt(_BundleDimensions[[1]]) else 1;87    let columns = if _BundleDimensionPresent[[2]] then88        UInt(_BundleDimensions[[2]]) else valid_columns;89    if TileOperationUsesClosedRowExpansionSchema(operation) then90        return _Tiles[[broadcast]].valid_rows == valid_rows &&91               _Tiles[[broadcast]].valid_columns == 1 &&92               _Tiles[[broadcast]].columns == 1;93    end;94    return _Tiles[[broadcast]].valid_rows == 1 &&95           _Tiles[[broadcast]].valid_columns == valid_columns &&96           _Tiles[[broadcast]].columns == columns;97end;98
99readonly func SelectedBundleClosedExpansionSchemaLegal(100    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean101begin102    if !TileOperationUsesClosedExpansionSchema(operation) then103        return TRUE;104    end;105    if BundleTileBindingCount() != 1 ||106       BundleSharedBindingCount() != 0 ||107       _BundleScalarBindings[[0]].valid ||108       !SelectedBundleComparisonDimensionsLegal() then109        return FALSE;110    end;111
112    let binding = _BundleTileBindings[[0]];113    let copy = TileExpansionOperationIsCopy(operation);114    if !binding.destination_valid ||115       binding.destination_allocated_by_bundle ||116       !BundleTileDestinationSizeLegal(0) ||117       !binding.source0_valid ||118       binding.source1_valid != !copy ||119       !binding.last then120        return FALSE;121    end;122
123    let broadcast = if copy then124        binding.source0 else binding.source1;125    var data_type = TileDataTypeFromEncoding(126        CurrentBundleTileOperationDataTypeCode()127            as TileDataTypeEncoding);128    let expdif = TileExpansionOperationIsExponentialDifference(operation);129    var source_data_type = data_type;130    if expdif then131        let (types_legal, selected_source_type, selected_destination_type) =132            SelectedBundleExpansionExponentialDifferenceTypes();133        if !types_legal then134            return FALSE;135        end;136        source_data_type = selected_source_type;137        data_type = selected_destination_type;138    end;139    if (!expdif && !TileVecArithmeticDataTypeSupported(data_type)) ||140       !TileDescriptorLegal(broadcast) ||141       _Tiles[[broadcast]].storage_kind != TileStorage_Numeric ||142       _Tiles[[broadcast]].data_type != source_data_type ||143       _Tiles[[broadcast]].layout != TileLayout_RowMajor ||144       !TileSourceContentsDefined(broadcast) ||145       !TileSourceEncodingsValid(broadcast) ||146       !SelectedBundleExpansionBroadcastShapeMatches(147           operation,148           broadcast) then149        return FALSE;150    end;151
152    if copy then153        return TRUE;154    end;155    return TileDescriptorLegal(binding.source0) &&156           _Tiles[[binding.source0]].storage_kind == TileStorage_Numeric &&157           _Tiles[[binding.source0]].data_type == source_data_type &&158           _Tiles[[binding.source0]].layout == TileLayout_RowMajor &&159           TileSourceContentsDefined(binding.source0) &&160           TileSourceEncodingsValid(binding.source0) &&161           SelectedBundleComparisonShapeMatches(binding.source0) &&162           ((TileOperationOfIndex(operation) != TileOperation_TROWEXPANDDIV &&163             TileOperationOfIndex(operation) != TileOperation_TCOLEXPANDDIV) ||164            !TileDataTypeIsInteger(data_type) ||165            TileBroadcastPayloadNonzero(166                if TileOperationUsesClosedRowExpansionSchema(operation) then167                    TileAxis_Row168                else169                    TileAxis_Column,170                binding.source0,171                binding.source1));172end;173