HIGH-SPEED IN-MEMORY COMPUTING USING DYNAMICAL MEMORY
A hardware accelerator tile for performing vector matric multiplications (VMMs) using a set of parameters, and a method for loading the parameters to compute engines of the hardware accelerator tile for use in the VMMs. The hardware accelerator tile includes (i) a plurality of compute engines respectively including compute-in-memory (CIM) modules configured to perform, in parallel, VMMs on stored parameters, (ii) one or more stationary memory units coupled with the plurality of compute engines, and (iii) local memory coupled with the plurality of compute engines.
1 . A hardware accelerator tile, comprising:
a plurality of compute engines respectively including compute-in-memory (CIM) modules configured to perform, in parallel, vector matrix multiplications (VMMs) on stored parameters;
one or more stationary memory units coupled with the plurality of compute engines; and
local memory coupled with the plurality of compute engines.
2 . The hardware accelerator tile of claim 1 , wherein data is loadable to the plurality of compute engines from the local memory or from the stationary memory units.
3 . The hardware accelerator tile of claim 2 , wherein the data loadable from the local memory or the stationary memory units comprises the parameters used in connection with the VMMs.
4 . The hardware accelerator tile of claim 1 , wherein the local memory is a static random-access memory (SRAM).
5 . The hardware accelerator tile of claim 1 , wherein a particular stationary memory unit of the one or more stationary memory units comprises a static random-access memory (SRAM).
6 . The hardware accelerator tile of claim 1 , wherein:
a set of parameters is loaded to a particular compute engine directly from a corresponding stationary memory unit;
the particular compute engine is comprised in the plurality of compute engines; and
the corresponding stationary memory unit is comprised in the one or more stationary memory units.
7 . The hardware accelerator tile of claim 6 , wherein the corresponding stationary memory unit is connected directly to the particular compute engine.
8 . The hardware accelerator tile of claim 1 , wherein data is provided from a memory to at least one stationary memory unit of the one or more stationary memory units.
9 . The hardware accelerator tile of claim 8 , wherein a time for loading the data from the memory to the at least one stationary memory unit is longer than a time for loading the data from the at least one stationary memory unit to the corresponding at least one compute engine.
10 . The hardware accelerator tile of claim 8 , wherein the memory is a dynamic random-access memory (DRAM), a double data rate (DDR), or a high bandwidth memory (HBM).
11 . The hardware accelerator tile of claim 1 , wherein the one or more stationary memory units are used as a cache for a set of parameters that is to be loaded to the plurality of compute engines.
12 . The hardware accelerator tile of claim 1 , wherein the one or more stationary memory units are vertically integrated with the plurality of compute engines.
13 . The hardware accelerator tile of claim 1 , further comprising a memory that stores a set of parameters to be loaded to the plurality of compute engines.
14 . The hardware accelerator tile of claim 15 , wherein at least a subset of the set of parameters are cached in the one or more stationary memory units.
15 . The hardware accelerator tile of claim 14 , wherein two of the plurality of layers communicate wirelessly.
16 . The hardware accelerator tile of claim 14 , wherein the hardware accelerator is one tile among a plurality of tiles in a chip architecture.
17 . A machine learning system, comprising:
at least one processor; and
a plurality of tiles coupled with the at least one processor, each of the plurality of tiles including: (i) a plurality of compute engines respectively including compute-in-memory (CIM) modules configured to perform, in parallel, vector matrix multiplications (VMMs) on stored parameters, (ii) one or more stationary memory units coupled with the plurality of compute engines, and (iii) local memory coupled with the plurality of compute engines.
18 . The machine learning system of claim 17 , wherein:
a set of parameters is loaded to a particular compute engine directly from a corresponding stationary memory unit;
the particular compute engine is comprised in the plurality of compute engines; and
the corresponding stationary memory unit is comprised in the one or more stationary memory units.
19 . A method, comprising:
storing a set of parameters to be used in connection with vector matrix multiplications (VMMs);
loading a subset of the parameters to a particular stationary memory unit; and
loading the subset of parameters from the particular stationary memory unit to a particular compute engine,
wherein the particular compute engine and the particular stationary memory unit are comprised in a hardware accelerator tile comprising (i) a plurality of compute engines, (ii) one or more stationary memory units coupled with the plurality of compute engines, and (iii) local memory coupled with the plurality of compute engines.
20 . The method of claim 19 , wherein the subset of parameters are cached at the particular stationary memory unit before the particular compute engine is to use the subset of parameters in connection with the VMMs.