PTO-TILE-MODEL-CAPACITY-LOCAL

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-CAPACITY-LOCAL","surface":"tile","classification":["model","capacity","local"],"depends_on":["PTO-TILE-MODEL-STATE-SHARED-REGISTERS"]}2readonly func TileCapacityLimitBytes() => integer {0..262144}3begin4    assert UInt(_SystemRegisters.tile_capacity) <=5        PTO_MODEL_MAX_TILE_CAPACITY_BYTES;6    return UInt(_SystemRegisters.tile_capacity) as integer {0..262144};7end;8
9readonly func TileCapacityInUseExcept(excluded: TileIndex) => integer10begin11    var total: integer = 0;12    for index = 0 to PTO_TILE_REGISTER_COUNT - 1 do13        if index != excluded && _Tiles[[index]].allocated then14            total = total + TileCoreAllocationBytes(15                _TileAllocationMasks[[index]],16                _Tiles[[index]].capacity_bytes);17        end;18    end;19    return total;20end;21
22readonly func TileCapacityInUseForPE(23    pe_identity: integer {0..3}) => integer24begin25    var total: integer = 0;26    let mask_bit = PTOPEMaskBitOfPEIdentity(pe_identity);27    for index = 0 to PTO_TILE_REGISTER_COUNT - 1 do28        if _Tiles[[index]].allocated &&29           _TileAllocationMasks[[index]][mask_bit] == '1' then30            total = total + _Tiles[[index]].capacity_bytes;31        end;32    end;33    return total;34end;35
36readonly func TileCapacityInUseExceptForPE(37    excluded: TileIndex, pe_identity: integer {0..3}) => integer38begin39    var total: integer = 0;40    let mask_bit = PTOPEMaskBitOfPEIdentity(pe_identity);41    for index = 0 to PTO_TILE_REGISTER_COUNT - 1 do42        if index != excluded && _Tiles[[index]].allocated &&43           _TileAllocationMasks[[index]][mask_bit] == '1' then44            total = total + _Tiles[[index]].capacity_bytes;45        end;46    end;47    return total;48end;49
50readonly func LocalTileAllocationFitsExcept(51    excluded: TileIndex, pe_mask: bits(4), per_pe_bytes: integer) => boolean52begin53    for pe = 0 to PTO_MODEL_MEMORY_AGENTS - 1 do54        let mask_bit = PTOPEMaskBitOfPEIdentity(pe);55        if pe_mask[mask_bit] == '1' &&56           TileCapacityInUseExceptForPE(excluded, pe) + per_pe_bytes >57               TileCapacityLimitBytes() then58            return FALSE;59        end;60    end;61    return TRUE;62end;63
64readonly func LocalTileAllocationFits(65    pe_mask: bits(4), per_pe_bytes: integer) => boolean66begin67    for pe = 0 to PTO_MODEL_MEMORY_AGENTS - 1 do68        let mask_bit = PTOPEMaskBitOfPEIdentity(pe);69        if pe_mask[mask_bit] == '1' &&70           TileCapacityInUseForPE(pe) + per_pe_bytes >71               TileCapacityLimitBytes() then72            return FALSE;73        end;74    end;75    return TRUE;76end;77
78readonly func TileCapacityInUse() => integer79begin80    var total: integer = 0;81    for index = 0 to PTO_TILE_REGISTER_COUNT - 1 do82        if _Tiles[[index]].allocated then83            total = total + TileCoreAllocationBytes(84                _TileAllocationMasks[[index]],85                _Tiles[[index]].capacity_bytes);86        end;87    end;88    return total;89end;90