Apparatus and mechanisms for processing neural network tasks using a single chip package with multiple identical dies

摘要

Apparatus and methods are provided for processing neural network models. The apparatus includes a plurality of identical artificial intelligence processing dies. Each artificial intelligence processing die of the plurality of identical artificial intelligence processing dies includes one or more inter-die input blocks and one or more inter-die output blocks. Each artificial intelligence processing die of the plurality of identical artificial intelligence processing dies is passed through at least one communication path from at least one inter-die output block of the artificial intelligence processing die to at least one inter-die input block of the artificial intelligence processing die. And, communicatively connected with another AI processing die of the plurality of same AI processing dies. Each artificial intelligence processing die of the plurality of identical artificial intelligence processing dies corresponds to at least one layer of the neural network.