PTO-TILE-MODEL-LEGALITY-MATRIX-POSTPROCESS

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-LEGALITY-MATRIX-POSTPROCESS","surface":"tile","classification":["model","legality","matrix-postprocess"],"depends_on":["PTO-BLOCK-B-FPATR","PTO-BLOCK-MODEL-OPERANDS-SUBVIEW-DESCRIPTOR","PTO-BLOCK-MODEL-OPERANDS-TILE-BINDINGS","PTO-TILE-MODEL-LEGALITY-MATRIX-SHAPE"]}2// PTO-REQ-CUBE-POSTPROCESS-001: auxiliary Matrix operands are completely3// descriptor- and payload-preflighted before source snapshots or allocation.4
5readonly func BundleMatrixDestinationAt(6    ordinal: integer {0..2}) => TileIndex7begin8    var seen: integer {0..3} = 0;9    for binding = 0 to PTO_BUNDLE_TILE_BINDING_COUNT - 1 looplimit 16 do10        if _BundleTileBindings[[binding]].valid &&11           _BundleTileBindings[[binding]].destination_valid then12            if seen == ordinal then13                return _BundleTileBindings[[binding]].destination;14            end;15            seen = (seen + 1) as integer {0..3};16        end;17    end;18    return 0;19end;20
21readonly func BundleMatrixSourceAt(ordinal: integer {0..8}) => TileIndex22begin23    var seen: integer {0..9} = 0;24    for binding = 0 to PTO_BUNDLE_TILE_BINDING_COUNT - 1 looplimit 16 do25        if _BundleTileBindings[[binding]].valid then26            if _BundleTileBindings[[binding]].source0_valid then27                if seen == ordinal then28                    return BundleTileSourceIndex(29                        binding as BundleTileBindingIndex, FALSE);30                end;31                seen = (seen + 1) as integer {0..9};32            end;33            if _BundleTileBindings[[binding]].source1_valid then34                if seen == ordinal then35                    return BundleTileSourceIndex(36                        binding as BundleTileBindingIndex, TRUE);37                end;38                seen = (seen + 1) as integer {0..9};39            end;40        end;41    end;42    return 0;43end;44
45readonly func BundleMatrixArchitecturalSourceAt(46    ordinal: integer {0..8}) => TileIndex47begin48    var seen: integer {0..9} = 0;49    for binding = 0 to PTO_BUNDLE_TILE_BINDING_COUNT - 1 looplimit 16 do50        if _BundleTileBindings[[binding]].valid then51            if _BundleTileBindings[[binding]].source0_valid then52                if seen == ordinal then53                    return BundleTileArchitecturalSourceIndex(54                        binding as BundleTileBindingIndex, FALSE);55                end;56                seen = (seen + 1) as integer {0..9};57            end;58            if _BundleTileBindings[[binding]].source1_valid then59                if seen == ordinal then60                    return BundleTileArchitecturalSourceIndex(61                        binding as BundleTileBindingIndex, TRUE);62                end;63                seen = (seen + 1) as integer {0..9};64            end;65        end;66    end;67    return 0;68end;69
70readonly func TileMatrixAuxiliarySourceSchemaLegal(71    source: TileIndex,72    valid_rows: integer {1..65535},73    valid_columns: integer {1..65535},74    data_type: TileDataType) => boolean75begin76    let tile = _Tiles[[source]];77    return TileSourceContentsDefined(source) &&78           TileInfoDescriptorLegal(tile) &&79           tile.valid_rows == valid_rows &&80           tile.valid_columns == valid_columns &&81           tile.data_type == data_type &&82           tile.layout == TileLayout_RowMajor &&83           tile.location == TileLocation_Any;84end;85
86readonly func TileMatrixVectorQuantContentsLegal(87    source: TileIndex, mode: bits(6)) => boolean88begin89    let tile = _Tiles[[source]];90    for column = 0 to tile.valid_columns - 1 looplimit 65536 do91        let element = TileLogicalLinearIndex(92            tile, 0, column as integer {0..65535});93        if !BundleFPATRQuantParameterWordLegal(94               mode, TileReadLogicalElement(tile, element)) then95            return FALSE;96        end;97    end;98    return TRUE;99end;100
101readonly func TileMatrixVectorReluContentsLegal(102    source: TileIndex) => boolean103begin104    let tile = _Tiles[[source]];105    for column = 0 to tile.valid_columns - 1 looplimit 65536 do106        let element = TileLogicalLinearIndex(107            tile, 0, column as integer {0..65535});108        if !BundleFPATRReluParameterWordLegal(109               TileReadLogicalElement(tile, element)) then110            return FALSE;111        end;112    end;113    return TRUE;114end;115
116readonly func BundleMatrixPostProcessSourcesLegal(117    mathematical_sources: integer {0..6},118    m: integer {1..65535},119    n: integer {1..65535},120    accumulator_type: TileDataType) => boolean121begin122    if !BundleFPATRAccumulatorTypeLegal(123           _BundleFixedPointAttributes.pre_quant_mode,124           accumulator_type) then125        return FALSE;126    end;127    var ordinal = mathematical_sources as integer {0..8};128    if _BundleFixedPointAttributes.row_max_en &&129       _BundleFixedPointAttributes.row_max_init then130        let row_max = BundleMatrixSourceAt(ordinal);131        if !TileMatrixAuxiliarySourceSchemaLegal(132               row_max, m, 1, accumulator_type) then133            return FALSE;134        end;135        ordinal = (ordinal + 1) as integer {0..8};136    end;137
138    if BundleFPATRModeUsesVectorParameter(139           _BundleFixedPointAttributes.pre_quant_mode) then140        let quant = BundleMatrixSourceAt(ordinal);141        if !TileMatrixAuxiliarySourceSchemaLegal(142               quant, 1, n, TileDataType_U64) ||143           !TileMatrixVectorQuantContentsLegal(144               quant, _BundleFixedPointAttributes.pre_quant_mode) then145            return FALSE;146        end;147        ordinal = (ordinal + 1) as integer {0..8};148    end;149
150    if BundleFPATRReluModeUsesVectorParameter(151           _BundleFixedPointAttributes.relu_mode) then152        let relu = BundleMatrixSourceAt(ordinal);153        if !TileMatrixAuxiliarySourceSchemaLegal(154               relu, 1, n, TileDataType_U64) ||155           !TileMatrixVectorReluContentsLegal(relu) then156            return FALSE;157        end;158    end;159    return TRUE;160end;161