PTO-BLOCK-MODEL-DISPATCH-SCALAR-SCHEMA

1// PTO-UNIT: {"id":"PTO-BLOCK-MODEL-DISPATCH-SCALAR-SCHEMA","surface":"block","classification":["model","dispatch","scalar-schema"],"depends_on":["PTO-BLOCK-MODEL-DISPATCH-DESCRIPTOR-LEGALITY","PTO-TILE-MODEL-NUMERIC-FORMATS"]}2pure func BundleOperationConsumesScalarSource0(3    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean4begin5    return TileOperandPresent(operation, TileOperand_address) ||6           TileOperandPresent(operation, TileOperand_scalar0);7end;8
9pure func BundleOperationConsumesScalarSource1(10    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean11begin12    return (TileOperandPresent(operation, TileOperand_address) &&13            (TileOperandPresent(operation, TileOperand_scalar0) ||14             TileOperandPresent(operation, TileOperand_scalar1))) ||15           (TileOperandPresent(operation, TileOperand_scalar0) &&16            TileOperandPresent(operation, TileOperand_scalar1));17end;18
19// Complete-bundle GPR inputs are packed in the architectural order20// address, scalar0, scalar1, diagonal, flag0.  Tile operands that are not21// present in the selected operation do not consume a B.IOR source slot.22readonly func BundleOperationGPRInputCount(23    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => integer {0..7}24begin25    var count: integer {0..7} = 0;26    if TileOperandPresent(operation, TileOperand_address) then27        count = (count + 1) as integer {0..7};28    end;29    if TileOperandPresent(operation, TileOperand_scalar0) then30        count = (count + 1) as integer {0..7};31    end;32    if TileOperandPresent(operation, TileOperand_scalar1) then33        count = (count + 1) as integer {0..7};34    end;35    if TileOperandPresent(operation, TileOperand_diagonal) then36        count = (count + 1) as integer {0..7};37    end;38    if TileOperandPresent(operation, TileOperand_flag0) then39        count = (count + 1) as integer {0..7};40    end;41    if _BundleOperation.valid &&42       _BundleOperation.operation_class == BundleOperation_TileMatrix &&43       _BundleFixedPointAttributes.valid then44        if BundleFPATRModeUsesScalarParameter(45               _BundleFixedPointAttributes.pre_quant_mode) then46            count = (count + 1) as integer {0..7};47        end;48        if BundleFPATRReluModeUsesScalarParameter(49               _BundleFixedPointAttributes.relu_mode) then50            count = (count + 1) as integer {0..7};51        end;52    end;53    return count;54end;55
56pure func BundleOperationGPRInputSlot(57    operation: integer {0..PTO_TILE_OPERATION_COUNT-1},58    field: TileOperandField) => integer {0..5}59begin60    var slot: integer {0..5} = 0;61    if TileOperandPresent(operation, TileOperand_address) then62        if field == TileOperand_address then return slot; end;63        slot = (slot + 1) as integer {0..5};64    end;65    if TileOperandPresent(operation, TileOperand_scalar0) then66        if field == TileOperand_scalar0 then return slot; end;67        slot = (slot + 1) as integer {0..5};68    end;69    if TileOperandPresent(operation, TileOperand_scalar1) then70        if field == TileOperand_scalar1 then return slot; end;71        slot = (slot + 1) as integer {0..5};72    end;73    if TileOperandPresent(operation, TileOperand_diagonal) then74        if field == TileOperand_diagonal then return slot; end;75        slot = (slot + 1) as integer {0..5};76    end;77    if TileOperandPresent(operation, TileOperand_flag0) then78        if field == TileOperand_flag0 then return slot; end;79        slot = (slot + 1) as integer {0..5};80    end;81    return 3;82end;83
84readonly func BundleOperationGPRInputSelector(85    slot: integer {0..2}) => Reg5Selector86begin87    if !_BundleScalarBindings[[0]].valid then return 0; end;88    case slot of89        when 0 => return _BundleScalarBindings[[0]].source0;90        when 1 => return _BundleScalarBindings[[0]].source1;91        when 2 => return _BundleScalarBindings[[0]].source2;92    end;93    unreachable;94end;95
96readonly func BundleOperationGPRBindingValuesLegal(97    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean98begin99    if !_BundleScalarBindings[[0]].valid then return TRUE; end;100    let input_count = BundleOperationGPRInputCount(operation);101    if input_count > 3 then return FALSE; end;102    let decoded = TileOperationOfIndex(operation);103    if decoded == TileOperation_TQUANT ||104       decoded == TileOperation_TDEQUANT then105        let scale = ReadScalarRegisterOperand(106            BundleOperationGPRInputSelector(0));107        let zero_point = ReadScalarRegisterOperand(108            BundleOperationGPRInputSelector(1));109        if !TileQuantizationScaleLegal(scale) then return FALSE; end;110        let zero_point_type = if decoded == TileOperation_TQUANT then111            if _BundleDataAttributes.data_type_present &&112               BundleDataTypeConcrete(_BundleDataAttributes.data_type) then113                BundleTileDataType(_BundleDataAttributes.data_type)114            else TileDataType_FP64115        else if _BundleOperation.data_type_valid &&116                BundleDataTypeConcrete(_BundleOperation.data_type) then117            BundleTileDataType(_BundleOperation.data_type)118        else TileDataType_FP64;119        if !TileQuantizationZeroPointLegal(120               zero_point,121               zero_point_type) then122            return FALSE;123        end;124    end;125    if TileOperandPresent(operation, TileOperand_flag0) then126        let slot = BundleOperationGPRInputSlot(operation, TileOperand_flag0);127        let raw = ReadScalarRegisterOperand(128            BundleOperationGPRInputSelector(slot as integer {0..2}));129        let value = UInt(raw);130        if value != 0 && value != 1 then return FALSE; end;131    end;132    if TileOperandPresent(operation, TileOperand_diagonal) then133        let slot = BundleOperationGPRInputSlot(operation, TileOperand_diagonal);134        let raw = ReadScalarRegisterOperand(135            BundleOperationGPRInputSelector(slot as integer {0..2}));136        let value = SInt(raw);137        if value < -65535 || value > 65535 then return FALSE; end;138    end;139    if _BundleOperation.valid &&140       _BundleOperation.operation_class == BundleOperation_TileMatrix &&141       _BundleFixedPointAttributes.valid then142        var post_slot: integer {0..5} = 0;143        if TileOperandPresent(operation, TileOperand_address) then144            post_slot = (post_slot + 1) as integer {0..5};145        end;146        if TileOperandPresent(operation, TileOperand_scalar0) then147            post_slot = (post_slot + 1) as integer {0..5};148        end;149        if TileOperandPresent(operation, TileOperand_scalar1) then150            post_slot = (post_slot + 1) as integer {0..5};151        end;152        if TileOperandPresent(operation, TileOperand_diagonal) then153            post_slot = (post_slot + 1) as integer {0..5};154        end;155        if TileOperandPresent(operation, TileOperand_flag0) then156            post_slot = (post_slot + 1) as integer {0..5};157        end;158
159        if BundleFPATRModeUsesScalarParameter(160               _BundleFixedPointAttributes.pre_quant_mode) then161            let raw = ReadScalarRegisterOperand(162                BundleOperationGPRInputSelector(163                    post_slot as integer {0..2}));164            if !BundleFPATRQuantParameterWordLegal(165                   _BundleFixedPointAttributes.pre_quant_mode, raw) then166                return FALSE;167            end;168            post_slot = (post_slot + 1) as integer {0..5};169        end;170        if BundleFPATRReluModeUsesScalarParameter(171               _BundleFixedPointAttributes.relu_mode) then172            let raw = ReadScalarRegisterOperand(173                BundleOperationGPRInputSelector(174                    post_slot as integer {0..2}));175            if !BundleFPATRReluParameterWordLegal(raw) then176                return FALSE;177            end;178        end;179    end;180    return TRUE;181end;182
183readonly func BundleOperationScalarBindingSchemaLegal(184    operation: integer {0..PTO_TILE_OPERATION_COUNT-1}) => boolean185begin186    if !_BundleScalarBindings[[0]].valid then return TRUE; end;187    let input_count = BundleOperationGPRInputCount(operation);188    if input_count < 3 && _BundleScalarBindings[[0]].source2 != 0 then189        return FALSE;190    end;191    if input_count < 2 && _BundleScalarBindings[[0]].source1 != 0 then192        return FALSE;193    end;194    if input_count < 1 && _BundleScalarBindings[[0]].source0 != 0 then195        return FALSE;196    end;197    return _BundleScalarBindings[[0]].destination == 0;198end;199