PTO-TILE-MODEL-STATE-DESCRIPTORS

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-STATE-DESCRIPTORS","surface":"tile","classification":["model","state","descriptors"],"depends_on":["PTO-TILE-MODEL-CAPACITY-SHARED"]}2pure func TileHandOf(index: TileIndex) => TileHand3begin4    if index < 16 then return TileHand_T;5    elsif index < 32 then return TileHand_U;6    elsif index < 48 then return TileHand_M;7    else return TileHand_N;8    end;9end;10
11pure func TileIndexWithinHand(index: TileIndex) => integer {1..16}12begin13    return ((index MOD 16) + 1) as integer {1..16};14end;15
16readonly func TileCapacityIsLegal(capacity_bytes: integer {0..262144}) => boolean17begin18    return capacity_bytes >= PTO_TILE_CELL_BYTES &&19           capacity_bytes MOD PTO_TILE_CELL_BYTES == 0 &&20           capacity_bytes <= PTO_TILE_MAX_ALLOCATION_BYTES &&21           capacity_bytes <= TileCapacityLimitBytes();22end;23
24readonly func SharedTileCapacityIsLegal(25    capacity_bytes: integer {0..262144}) => boolean26begin27    return capacity_bytes >= PTO_TILE_CELL_BYTES &&28           capacity_bytes MOD PTO_TILE_CELL_BYTES == 0 &&29           capacity_bytes <= SharedTileCapacityLimitBytes();30end;31
32pure func TileSizeCodeIsLegal(size_code: integer {0..15}) => boolean33begin34    return 1 <= size_code && size_code <= 12;35end;36
37pure func LocalTileSizeCodeIsLegal(size_code: integer {0..15}) => boolean38begin39    return 1 <= size_code && size_code <= 10;40end;41
42pure func TileSizeCodeBytes(size_code: integer {1..12})43    => integer {128,256,512,1024,2048,4096,8192,16384,32768,65536,44                131072,262144}45begin46    case size_code of47        when 1 => return 128;48        when 2 => return 256;49        when 3 => return 512;50        when 4 => return 1024;51        when 5 => return 2048;52        when 6 => return 4096;53        when 7 => return 8192;54        when 8 => return 16384;55        when 9 => return 32768;56        when 10 => return 65536;57        when 11 => return 131072;58        when 12 => return 262144;59    end;60end;61
62pure func TileElementBits(data_type: TileDataType) => integer {4,8,16,32,64}63begin64    case data_type of65        when TileDataType_E2M1X2, TileDataType_E1M2X2,66             TileDataType_HiF4X2, TileDataType_S4X2,67             TileDataType_U4X2 => return 4;68        when TileDataType_S8, TileDataType_U8, TileDataType_HiF8,69             TileDataType_E4M3, TileDataType_E5M2, TileDataType_E3M2,70             TileDataType_E2M3, TileDataType_E8M0 => return 8;71        when TileDataType_S16, TileDataType_U16, TileDataType_FP16,72             TileDataType_BF16 => return 16;73        when TileDataType_S32, TileDataType_U32,74             TileDataType_FP32, TileDataType_TF32,75             TileDataType_HF32 => return 32;76        when TileDataType_S64, TileDataType_U64,77             TileDataType_FP64 => return 64;78    end;79end;80
81pure func TileDataTypeIsFourBit(data_type: TileDataType) => boolean82begin83    return data_type == TileDataType_E2M1X2 ||84           data_type == TileDataType_E1M2X2 ||85           data_type == TileDataType_HiF4X2 ||86           data_type == TileDataType_S4X2 ||87           data_type == TileDataType_U4X2;88end;89
90// The executable payload remains bounded by PTO_MODEL_TILE_ELEMENTS. Large91// descriptors retain their architectural logical-element capacity through92// width-aware Word carriers, so a 256 KiB Shared shape and the full common93// SizeCode map remain representable without allocating a maximum Word per94// logical element. Local object legality is capped separately at 64 KiB.95readonly func TileLogicalElementCapacity(96    capacity_bytes: integer {0..262144}, data_type: TileDataType)97    => integer {1..524288}98begin99    assert capacity_bytes > 0;100    return ((capacity_bytes * 8) DIVRM TileElementBits(data_type))101        as integer {1..524288};102end;103