Source code for spinn_machine.chip

# Copyright (c) 2014 The University of Manchester
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from spinn_utilities.ordered_set import OrderedSet
from .processor import Processor

standard_processors = {}


[docs]class Chip(object): """ Represents a SpiNNaker chip with a number of cores, an amount of SDRAM shared between the cores, and a router. The chip is iterable over the processors, yielding ``(processor_id, processor)`` where: * ``processor_id`` is the ID of a processor * ``processor`` is the :py:class:`Processor` with ``processor_id`` """ # tag 0 is reserved for stuff like IO STD _IPTAG_IDS = OrderedSet(range(1, 8)) __slots__ = ( "_x", "_y", "_p", "_router", "_sdram", "_ip_address", "_tag_ids", "_nearest_ethernet_x", "_nearest_ethernet_y", "_n_user_processors", "_parent_link", "_v_to_p_map" ) # pylint: disable=too-many-arguments def __init__(self, x, y, n_processors, router, sdram, nearest_ethernet_x, nearest_ethernet_y, ip_address=None, tag_ids=None, down_cores=None, parent_link=None, v_to_p_map=None): """ :param int x: the x-coordinate of the chip's position in the two-dimensional grid of chips :param int y: the y-coordinate of the chip's position in the two-dimensional grid of chips :param int n_processors: the number of processors including monitor processors. :param ~spinn_machine.Router router: a router for the chip :param ~spinn_machine.SDRAM sdram: an SDRAM for the chip :param ip_address: the IP address of the chip, or ``None`` if no Ethernet attached :type ip_address: str or None :param tag_ids: IDs to identify the chip for SDP can be empty to define no tags or `None` to allocate tag automatically based on if there is an ip_address :type tag_ids: iterable(int) or None :param nearest_ethernet_x: the nearest Ethernet x coordinate :type nearest_ethernet_x: int or None :param nearest_ethernet_y: the nearest Ethernet y coordinate :type nearest_ethernet_y: int or None :param down_cores: Ids of cores that are down for this Chip :type down_cores: iterable(int) or None :param parent_link: The link down which the parent chips is found in the tree of chips towards the root (boot) chip :type parent_link: int or None :param v_to_p_map: An array indexed by virtual core which gives the physical core number or 0xFF if no entry for the core :type v_to_p_map: bytearray or None :raise ~spinn_machine.exceptions.SpinnMachineAlreadyExistsException: If processors contains any two processors with the same ``processor_id`` """ self._x = x self._y = y self._p = self.__generate_processors(n_processors, down_cores) self._router = router self._sdram = sdram self._ip_address = ip_address if tag_ids is not None: self._tag_ids = tag_ids elif self._ip_address is None: self._tag_ids = [] else: self._tag_ids = self._IPTAG_IDS self._nearest_ethernet_x = nearest_ethernet_x self._nearest_ethernet_y = nearest_ethernet_y self._parent_link = parent_link self._v_to_p_map = v_to_p_map def __generate_processors(self, n_processors, down_cores): if down_cores is None: if n_processors not in standard_processors: processors = dict() processors[0] = Processor.factory(0, True) for i in range(1, n_processors): processors[i] = Processor.factory(i) standard_processors[n_processors] = processors self._n_user_processors = n_processors - 1 return standard_processors[n_processors] else: processors = dict() if 0 in down_cores: raise NotImplementedError( "Declaring core 0 as down is not supported") processors[0] = Processor.factory(0, True) for i in range(1, n_processors): if i not in down_cores: processors[i] = Processor.factory(i) self._n_user_processors = n_processors - 1 - len(down_cores) return processors
[docs] def is_processor_with_id(self, processor_id): """ Determines if a processor with the given ID exists in the chip. Also implemented as ``__contains__(processor_id)`` :param int processor_id: the processor ID to check for :return: Whether the processor with the given ID exists :rtype: bool """ return processor_id in self._p
[docs] def get_processor_with_id(self, processor_id): """ Return the processor with the specified ID, or ``None`` if the processor does not exist. :param int processor_id: the ID of the processor to return :return: the processor with the specified ID, or ``None`` if no such processor :rtype: Processor or None """ if processor_id in self._p: return self._p[processor_id] return None
@property def x(self): """ The X-coordinate of the chip in the two-dimensional grid of chips. :rtype: int """ return self._x @property def y(self): """ The Y-coordinate of the chip in the two-dimensional grid of chips. :rtype: int """ return self._y @property def processors(self): """ An iterable of available processors. :rtype: iterable(Processor) """ return iter(self._p.values()) @property def n_processors(self): """ The total number of processors. :rtype: int """ return len(self._p) @property def n_user_processors(self): """ The total number of processors that are not monitors. :rtype: int """ return self._n_user_processors @property def router(self): """ The router object associated with the chip. :rtype: Router """ return self._router @property def sdram(self): """ The SDRAM associated with the chip. :rtype: SDRAM """ return self._sdram @property def ip_address(self): """ The IP address of the chip, or ``None`` if there is no Ethernet connected to the chip. :rtype: str or None """ return self._ip_address @property def nearest_ethernet_x(self): """ The X-coordinate of the nearest Ethernet chip. :rtype: int """ return self._nearest_ethernet_x @property def nearest_ethernet_y(self): """ The Y-coordinate of the nearest Ethernet chip. :rtype: int """ return self._nearest_ethernet_y @property def tag_ids(self): """ The tag IDs supported by this chip. :rtype: iterable(int) """ return self._tag_ids
[docs] def get_first_none_monitor_processor(self): """ Get the first processor in the list which is not a monitor core. :rtype: Processor or None """ for processor in self.processors: if not processor.is_monitor: return processor return None
@property def parent_link(self): """ The link down which the parent is found in the tree of chips rooted at the machine root chip (probably 0, 0 in most cases). This will be ``None`` if the chip information didn't contain this value. :rtype: int or None """ return self._parent_link
[docs] def get_physical_core_id(self, virtual_p): """ Get the physical core ID from a virtual core ID. :param int virtual_p: The virtual core ID :rtype: int or None if core not in map """ if (self._v_to_p_map is None or virtual_p >= len(self._v_to_p_map) or self._v_to_p_map[virtual_p] == 0xFF): return None return self._v_to_p_map[virtual_p]
[docs] def get_physical_core_string(self, virtual_p): """ Get a string that can be appended to a core to show the physical core, or an empty string if not possible. :param int virtual_p: The virtual core ID :rtype: str """ physical_p = self.get_physical_core_id(virtual_p) if physical_p is None: return "" return f" (ph: {physical_p})"
def __iter__(self): """ Get an iterable of processor identifiers and processors :return: An iterable of ``(processor_id, processor)`` where: * ``processor_id`` is the ID of a processor * ``processor`` is the processor with the ID :rtype: iterable(tuple(int,Processor)) """ return iter(self._p.items()) def __len__(self): """ The number of processors associated with this chip. :return: The number of items in the underlying iterator. :rtype: int """ return len(self._p) def __getitem__(self, processor_id): if processor_id in self._p: return self._p[processor_id] # Note difference from get_processor_with_id(); this is to conform to # standard Python semantics raise KeyError(processor_id) def __contains__(self, processor_id): return self.is_processor_with_id(processor_id) def __str__(self): return ( f"[Chip: x={self._x}, y={self._y}, sdram={self.sdram}, " f"ip_address={self.ip_address}, router={self.router}, " f"processors={list(self._p.values())}, " f"nearest_ethernet={self._nearest_ethernet_x}:" f"{self._nearest_ethernet_y}]") def __repr__(self): return self.__str__()