PTO-TILE-MODEL-LEGALITY-DTYPE-LAYOUT

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-LEGALITY-DTYPE-LAYOUT","surface":"tile","classification":["model","legality","dtype-layout"],"depends_on":["PTO-TILE-MODEL-LEGALITY-DESCRIPTOR-SHAPE"]}2pure func TileTeplRawCarrierTypeSupported(data_type: TileDataType) => boolean3begin4    // PTO-v0 TEPL operates over the raw XLEN carrier for every architectural5    // tile type. Target numeric interpretation, rounding, saturation, and6    // exceptional values remain Stage 5 profile obligations.7    case data_type of8        when TileDataType_FP64, TileDataType_FP32, TileDataType_TF32,9             TileDataType_HF32, TileDataType_FP16, TileDataType_BF16,10             TileDataType_HiF8, TileDataType_E4M3, TileDataType_E5M2,11             TileDataType_E3M2, TileDataType_E2M3,12             TileDataType_E2M1X2, TileDataType_E1M2X2,13             TileDataType_E8M0, TileDataType_HiF4X2,14             TileDataType_S64, TileDataType_S32, TileDataType_S16,15             TileDataType_S8, TileDataType_S4X2,16             TileDataType_U64, TileDataType_U32, TileDataType_U16,17             TileDataType_U8, TileDataType_U4X2 => return TRUE;18        otherwise => return FALSE;19    end;20end;21
22// Stage 4 carrier-only operations use the concrete dtype's physical byte23// width.  Packed X2 formats have a one-byte storage class but retain their24// baseline nibble semantics and are deliberately excluded here.25pure func TileCarrierOnlyDataTypeSupported(26    data_type: TileDataType) => boolean27begin28    return !TileDataTypeIsFourBit(data_type) &&29           TileElementBytes(data_type) <= 4;30end;31
32// These operations already have a packed-X2 baseline.  Preserve that33// baseline while admitting only the new non-packed B8/B16/B32 carrier set;34// B64 remains outside the Stage 4 extension.35pure func TileCarrierOrPackedBaselineDataTypeSupported(36    data_type: TileDataType) => boolean37begin38    return TileCarrierOnlyDataTypeSupported(data_type) ||39           TileDataTypeIsFourBit(data_type);40end;41
42// Move24 operations accept every assigned Tile DataType except HiF4X2.43// Keep that exact architectural exclusion independent of the narrower44// Stage-4 carrier helper used by other raw-carrier operations.45pure func TileCarrierOrMove24BaselineDataTypeSupported(46    data_type: TileDataType) => boolean47begin48    return TileMove24DataTypeSupported(data_type);49end;50
51pure func TileRegularTLSUDataTypeSupported(52    data_type: TileDataType) => boolean53begin54    return TileTeplRawCarrierTypeSupported(data_type);55end;56
57pure func TileVecArithmeticDataTypeSupported(58    data_type: TileDataType) => boolean59begin60    case data_type of61        when TileDataType_FP64, TileDataType_FP32, TileDataType_TF32,62             TileDataType_HF32, TileDataType_FP16, TileDataType_BF16,63             TileDataType_E4M3, TileDataType_E5M2,64             TileDataType_S64, TileDataType_S32, TileDataType_S16,65             TileDataType_S8, TileDataType_U64, TileDataType_U32,66             TileDataType_U16, TileDataType_U8 => return TRUE;67        otherwise => return FALSE;68    end;69end;70
71pure func TileA9DataTypeSupported(72    data_type: TileDataType) => boolean73begin74    case data_type of75        when TileDataType_S32, TileDataType_U32,76             TileDataType_FP32, TileDataType_S16,77             TileDataType_U16, TileDataType_FP16,78             TileDataType_BF16, TileDataType_S8,79             TileDataType_U8 => return TRUE;80        otherwise => return FALSE;81    end;82end;83
84pure func TileA7DataTypeSupported(85    data_type: TileDataType) => boolean86begin87    case data_type of88        when TileDataType_S32, TileDataType_U32,89             TileDataType_FP32, TileDataType_S16,90             TileDataType_U16, TileDataType_FP16,91             TileDataType_BF16 => return TRUE;92        otherwise => return FALSE;93    end;94end;95
96pure func TileF3DataTypeSupported(97    data_type: TileDataType) => boolean98begin99    return data_type == TileDataType_FP16 ||100           data_type == TileDataType_FP32 ||101           data_type == TileDataType_BF16;102end;103
104pure func TileFloatingElementwiseDataTypeSupported(105    data_type: TileDataType) => boolean106begin107    return data_type == TileDataType_FP64 ||108           data_type == TileDataType_FP32 ||109           data_type == TileDataType_TF32 ||110           data_type == TileDataType_HF32 ||111           data_type == TileDataType_FP16 ||112           data_type == TileDataType_BF16 ||113           data_type == TileDataType_E4M3 ||114           data_type == TileDataType_E5M2;115end;116
117pure func TileI6DataTypeSupported(118    data_type: TileDataType) => boolean119begin120    case data_type of121        when TileDataType_S32, TileDataType_U32,122             TileDataType_S16, TileDataType_U16,123             TileDataType_S8, TileDataType_U8 => return TRUE;124        otherwise => return FALSE;125    end;126end;127
128pure func TileTNegDataTypeSupported(129    data_type: TileDataType) => boolean130begin131    return TileVecArithmeticDataTypeSupported(data_type);132end;133
134pure func TileTReluDataTypeSupported(135    data_type: TileDataType) => boolean136begin137    return TileVecArithmeticDataTypeSupported(data_type);138end;139
140pure func TileArgReductionSourceDataTypeSupported(141    data_type: TileDataType) => boolean142begin143    return TileA9DataTypeSupported(data_type);144end;145
146pure func TileFusedMultiplyAddDataTypeSupported(147    data_type: TileDataType) => boolean148begin149    return TileVecArithmeticDataTypeSupported(data_type);150end;151
152pure func TileMove24DataTypeSupported(153    data_type: TileDataType) => boolean154begin155    return data_type != TileDataType_HiF4X2;156end;157
158pure func TileFillPadDataTypeSupported(159    data_type: TileDataType) => boolean160begin161    return TileVecArithmeticDataTypeSupported(data_type);162end;163
164pure func TileImg2ColDataTypeSupported(165    data_type: TileDataType) => boolean166begin167    case data_type of168        when TileDataType_FP32, TileDataType_FP16,169             TileDataType_BF16, TileDataType_S32,170             TileDataType_S16, TileDataType_S8,171             TileDataType_U32, TileDataType_U16,172             TileDataType_U8 => return TRUE;173        otherwise => return FALSE;174    end;175end;176
177pure func TileBinaryUsesClosedElementwiseContract(178    operation: TileBinaryOperation) => boolean179begin180    return operation == TileBinary_ADD ||181           operation == TileBinary_SUB ||182           operation == TileBinary_MUL ||183           operation == TileBinary_DIV ||184           operation == TileBinary_REM ||185           operation == TileBinary_MAX ||186           operation == TileBinary_MIN ||187           operation == TileBinary_AND ||188           operation == TileBinary_OR ||189           operation == TileBinary_XOR ||190           operation == TileBinary_SHL ||191           operation == TileBinary_SHR;192end;193
194// The closed elementwise family is also defined for Local CUBE M16/M32.195// CUBE_N8 remains a matrix/transport layout and is not an elementwise class.196pure func TileElementwiseLayoutSupported(layout: TileLayout) => boolean197begin198    return layout == TileLayout_RowMajor ||199           layout == TileLayout_CUBE_M16 ||200           layout == TileLayout_CUBE_M32;201end;202
203pure func TileVecScalarIntegerDataTypeSupported(204    data_type: TileDataType) => boolean205begin206    case data_type of207        when TileDataType_S64, TileDataType_S32, TileDataType_S16,208             TileDataType_S8, TileDataType_U64, TileDataType_U32,209             TileDataType_U16, TileDataType_U8 => return TRUE;210        otherwise => return FALSE;211    end;212end;213
214pure func TileBinaryDataTypeSupported(215    operation: TileBinaryOperation,216    data_type: TileDataType) => boolean217begin218    if operation == TileBinary_AND ||219       operation == TileBinary_OR ||220       operation == TileBinary_XOR then221        return TileVecScalarIntegerDataTypeSupported(data_type);222    end;223    if operation == TileBinary_SHL || operation == TileBinary_SHR then224        return TileVecScalarIntegerDataTypeSupported(data_type);225    end;226    return TileVecArithmeticDataTypeSupported(data_type);227end;228
229readonly func TileLogicalShapeMatch(left: TileIndex, right: TileIndex) => boolean230begin231    return TileDescriptorLegal(left) && TileDescriptorLegal(right) &&232           _Tiles[[left]].rows == _Tiles[[right]].rows &&233           _Tiles[[left]].columns == _Tiles[[right]].columns &&234           _Tiles[[left]].valid_rows == _Tiles[[right]].valid_rows &&235           _Tiles[[left]].valid_columns == _Tiles[[right]].valid_columns &&236           _Tiles[[left]].layout == _Tiles[[right]].layout;237end;238
239readonly func TileShapeAndTypeMatch(left: TileIndex, right: TileIndex) => boolean240begin241    return TileLogicalShapeMatch(left, right) &&242           _Tiles[[left]].storage_kind == _Tiles[[right]].storage_kind &&243           _Tiles[[left]].data_type == _Tiles[[right]].data_type;244end;245