Remote artificial intelligence (AI) acceleration system
A remote artificial intelligence (AI) acceleration system is provided. The system includes a plurality of application servers, wherein each of the plurality of application server is configured to execute AI applications over an AI software framework; at least one artificial intelligence accelerator (AIA) appliance server configured to execute AI processing tasks in response to requests from the AI applications; and at least one switch configured to allow connectivity between the plurality of application servers and the at least one AIA appliance server, wherein the plurality of network attached artificial intelligence accelerator (NA-AIA) engines connected to the AIA switch, wherein each of the plurality of NA-AIA engines connected is configured to execute at least one processing AI task.
Latest NeuReality LTD. Patents:
- Communication protocol, and a method thereof for accelerating artificial intelligence processing tasks
- COMMUNICATION PROTOCOL, AND A METHOD THEREOF FOR ACCELERATING ARTIFICIAL INTELLIGENCE PROCESSING TASKS
- METHOD AND SYSTEM FOR SEQUENCING ARTIFICIAL INTELLIGENCE (AI) JOBS FOR EXECUTION AT AI ACCELERATORS
- REMOTE ARTIFICIAL INTELLIGENCE (AI) ACCELERATION SYSTEM
- Method and system for sequencing artificial intelligence (AI) jobs for execution at AI accelerators
This application is a continuation of International Patent Application No. PCT/US2020/019157 filed on Feb. 21, 2020, which claims the benefit of U.S. Provisional Application No. 62/808,993 filed on Feb. 22, 2019, the contents of which are hereby incorporated by reference.
TECHNICAL FIELDThe disclosure generally relates to system architectures, and more specifically to embedded computing architectures and reconfigurable computing architectures.
BACKGROUNDThe demand and need for stronger, more scalable, higher capacity compute systems increases annually. This need and demand is due in a large part to the increased popularity of AI applications and its usage of machine learning and deep learning workloads. The execution of such applications is performed by servers configured as a dedicated AI infrastructure including hardware, software and AI-framework. The AI framework may be, for example, TensorFlow, Caffe2, Pytorch, or Microsoft® Cognitive Toolkit (CNTK), usually implementing the framework's APIs that are used by the application and implemented in wither software, hardware or combined. The hardware typically includes a dedicated hardware accelerator device, also known as a deep learning accelerator (DLA) that is offloading deep learning tasks from the software running on the CPU.
In the local direct-attached DLA architecture shown in
The DLA 114 is typically an AI Accelerator designed to accelerate execution of specific AI functions used by AI applications, such as deep learning tasks or neural networks. The DLA 114 is typically realized as a graphics processing unit (GPU), a Field Programmable Gate Arrays (FPGA), or a Application-specific integrated circuit (ASIC). The DLA 114 is connected to the host-CPU 112 through a PCIe bus (not labeled). In the DLA system 100, an application is executed by the host-CPU 112, where deep-learning tasks are off-loaded to the DLA 114. The DLA 114 executes dedicated software and utilizes dedicated hardware in order to accelerate the execution of such tasks.
While the direct-attached architecture shown in
Another type of conventional architecture of a system for executing AI applications is a remote direct-attached DLA system. An example for such a system 200 is shown in
The remote direct-attached DLA system 200 includes a plurality of application servers 210 and a DLA server 220 connected over a network switch 230. Each application server 210 executes an AI application over an AI software framework. Further, each application server 210 communicates with a DLA server 220 that executes specific deep-learning tasks. The DLA server 220 includes a host-CPU 221 executing the AI server application with its multi-client support, the network driver, the DLA driver and other AI server related tasks. Such tasks include, but are not limited to, an AI framework and runtime API's. The DLA server also includes a direct-attached DLA 222. The DLA 222 is the same as the DLA discussed above.
In an embodiment, the remote direct-attached DLA system 200 is configured to allocate the various workloads associated with the AI server to run on the host-CPU 221 and the deep-learning workload to run on the DLA 222. One of the disadvantages of the system 200 is scalability cost and power. Specifically, when an additional DLA resource is needed, an additional DLA server 220 should be added to the system including the associated computing resources, such as CPU, NIC and memory. Another disadvantage of the system 200, is that the utilization of such a system should require an optimization and balance of the workloads running on the host-CPU and the DLA. However, the optimization may not be optimal if the workloads to be executed by the DLA server 220 are varied in their CPU-DLA balance of workload overhead.
As an example, if the data of which needs to be processed in the DLA requires pre-processing (resizing, restructuring, or other functions), such tasks need to be allocated to the CPU and can cause a computation bottleneck in the CPU, lowering the effective utilization of the DLA.
In some configurations, the DLA server 220 is replaced with a DLA appliance server. An example architecture for a remote direct-attached DLA system 300 with a DLA appliance server 320 is shown in
The DLA appliance server 320 includes a plurality of DLAs 321 coupled to a host-CPU 322 through a PCIe switch 323. The DLA appliance server 320 balances the workloads discussed above between the attached DLAs 321 and within each DLA 321.
However, the DLA appliance server 320, and thus the entire direct-attached DLA system 300 suffers from high latency due to data path and computation bottleneck between the host-CPUs, its memory and the DLAs 321. The operation of the system 300 also requires multiple data copies between a memory of an application server 310 requesting, for example, a deep-learning task. The data copies are transferred from a memory of the host CPU 322 in the DLA appliance server 320 to the memory of a DLA 321 allocated to execute the requested deep-learning task. Further, data copies are transferred in the opposite direction while sending the result back to the AI application. The multiple data copies increase the latency and power of the entire system.
It would be therefore an advantageous to provide an AI acceleration solution that would overcome the deficiencies mentioned above.
SUMMARYA summary of several example embodiments of the disclosure follows. This summary is provided for the convenience of the reader to provide a basic understanding of such embodiments and does not wholly define the breadth of the disclosure. This summary is not an extensive overview of all contemplated embodiments, and is intended to neither identify key or critical elements of all embodiments nor to delineate the scope of any or all aspects. Its sole purpose is to present some concepts of one or more embodiments in a simplified form as a prelude to the more detailed description that is presented later. For convenience, the term “certain embodiments” may be used herein to refer to a single embodiment or multiple embodiments of the disclosure.
Certain embodiments disclosed herein include a remote artificial intelligence (AI) acceleration system. The system comprises a plurality of application servers, wherein each of the plurality of application server is configured to execute AI applications over an AI software framework; at least one artificial intelligence accelerator (AIA) appliance server configured to execute AI processing tasks in response to requests from the AI applications; and at least one switch configured to allow connectivity between the plurality of application servers and the at least one AIA appliance server, wherein the plurality of network attached artificial intelligence accelerator (NA-AIA) engines connected to the AIA switch, wherein each of the plurality of NA-AIA engines connected is configured to execute at least one processing AI task.
The subject matter disclosed herein is particularly pointed out and distinctly claimed in the claims at the conclusion of the specification. The foregoing and other objects, features, and advantages of the disclosed embodiments will be apparent from the following detailed description taken in conjunction with the accompanying drawings.
The embodiments disclosed by the invention are only examples of the many possible advantageous uses and implementations of the innovative teachings presented herein. In general, statements made in the specification of the present application do not necessarily limit any of the various claimed embodiments. Moreover, some statements may apply to some inventive features but not to others. In general, unless otherwise indicated, singular elements may be in plural and vice versa with no loss of generality. In the drawings, like numerals refer to like parts through several views.
The various disclosed embodiments allow for the acceleration and execution of AI workloads, used in AI applications in, for example, datacenters while reducing the TCO for datacenters' owners, that are also translated to AI-as-a-Service cost model. Specifically, the disclosed embodiments eliminates the need to balance between workloads. The workloads are allocated to a host-CPU (e.g., the host-CPU in systems 210 and 310,
In an embodiment, an AIA appliance server based on a network-attached AI accelerator (hereinafter “NA-AIA”) engine is disclosed. The NA-AIA engine is a system on chip (SoC) or integrated circuity (IC) providing AI computation and network capabilities that are required to transfer and compute the AI models and data (to be executed), in addition to the management, data path and control tasks associated with these offload requests of AI specific tasks, previously done partially by the host-CPU in system 200 and 300. As will be demonstrated below, the NA-AIA engines remove the compute and communication bottlenecks, thus provide a scalable solution.
The system 400 includes a plurality of application servers 410-1 through 410-n (collectively referred to as application server 410 or application servers 410), at least one AIA appliance server 420, and a number of switches 430 connecting the various components of the system 400. In an embodiment, each of the switches 430 is an Ethernet switch. In some configurations, the system 400 may include only one application server 410.
Each application server 410 is configured to execute an AI application over an AI software framework. Such a framework may be, for example, TensorFlow, Caffe2, Pytorch, or CNTK, and the like.
Each application server 410 further communicates with the AIA appliance server 420 that executes specific AI processing tasks. The AIA appliance server 420 is designed according to the disclosed embodiments to accelerate the execution of the AI tasks. The AIA appliance server 420 may include, in one configuration, a switch 421 (hereinafter a “AIA switch” 421) with a plurality of NA-AIA engines 422. Each NA-AIA engine 422 can potentially be coupled with its own memory 422-M. The AIA switch 421 may be an Ethernet switch.
The system 400 provides a scalable solution as there is no compute and communication bottlenecks. Specifically, in an embodiment, additional NA-AIA engines can be connected to the AIA switch 421, thereby expanding the AIA switch 421 to the AIA appliance server 420. The AIA switch 421 is configured to have enough ports and bandwidth to allow data traffic to be transferred flawlessly between the application servers 410 and the AIA appliance server 420 and the NA-AIA engines 422.
In an embodiment, further acceleration may be achieved by utilizing a proprietary protocol (AI over fabric, or AIoF). The AIoF protocol is a communication protocol designed to support installation and execution of AI models and AI operations, including but not limited, to deep-learning and Neural Networks. The AI models further include, but not limited to, transferring the acceleration and offload requests, data and responses. An AI model is a database of an AI computation graph including one or more functions. The function may include a Neural Network. The computation graph describes all the computation steps and connection between computation steps including the order of steps and interface between steps.
The AIoF protocol is designed to remove the network overhead, latency, and the multiple data copies required to transfer data from the application server 410 to the AIA appliance server 420 associated with the network protocol. Examples for such protocols include, but are not limited to, TCP, UDP, RDMA, and the like.
In an embodiment, the AIoF protocol can be configured using a shared memory over network, in which the application can use its memory while the hardware transparently copies the AI model or the data from the application memory to an NA-AIA engine's memory via the network and also copies the results and responses from the NA-AIA engine's memory to the application memory.
Specifically, in the arrangement disclosed in
In an embodiment, the data provided by the application server 410 may include a higher-level description of a compute pipeline of tasks or a sequence of multiple compute pipelines of the AI application. The AIA appliance server 420, and specifically the NA-AIA engine 422 can potentially perform the entire compute pipeline of tasks and also the complete AI sequence of compute pipelines and their associated data and also the responses sent back to the application server 410 upon completion.
This demonstrates further optimization as the processing of specific compute pipeline tasks are not performed by the application server's or DLA appliance server's host-CPU. Further, the data is not copied between the application server and the DLA appliance server before and after any step of the AI pipeline or the AI sequence requests being processed. The flow of data and control between an application server 410 and the AIA appliance server 420 is discussed in
For example, a virtual assistant application, such as Siri® by Apple®, AIexa® by Amazon®, and the like includes the following AI sequence: 1) voice recognition; 2) a natural language processing (NLP) processed to understand the context of the speech query; 3) a recommendation process that provide the result or answer; 4) a NLP process to phrase the recommendation (e.g., search results), and 5) a speech synthesis process to generate the speech. Reducing the latency and processing time of such sequences would dramatically improve the performance of such virtual assistance applications. Moreover, such applications that are projected to reach ultra-scale, serving millions of users. Thus, the AI applications would incur very high cost of computation and are dependent on technologies that expected to provide more affordable solutions such as described above.
It should be noted that the remote AI acceleration system 400 discussed herein can be implemented or otherwise integrated in a cloud computing platform an on-premises data center or residential gateway or home access point, a decentralized computing infrastructure (fog computing), and the like. The cloud computing platform may include a public cloud, a private cloud, or a hybrid cloud.
According to the disclosed embodiments, the system 400 can support the execution of any AI application. Examples for such application include cyber-security applications, speech recognition, object detections, digital payments, image recognition, loT applications, educational applications, medical applications, automotive applications, ecommerce applications, chatbots and other bots applications and the like. The nature of the application would not change the arrangement of the system 400.
The memory 545 may be a volatile memory such as, but not limited to random access memory (RAM), DRAM, or non-volatile memory (NVM), such as, but not limited to: flash memory.
In an embodiment, the processing circuitry 520 may be, or be a component of, a larger processing unit implemented with one or more processors. The processing circuitry 520 may be realized as one or more hardware logic components and circuits. For example, and without limitation, illustrative types of hardware logic components that can be used include external, non-embedded solutions such as: a CPU, field programmable gate arrays (FPGAs), application-specific integrated circuits (ASICs), application-specific standard products (ASSPs), system-on-a-chip systems (SOCs), general-purpose microprocessors, microcontrollers, digital signal processors (DSPs), and the like, or any other hardware logic components that can perform calculations or other manipulations of information.
According to the disclosed embodiments, the pre-processing engine 550 and the post-processing engine 570 and at least one AI engine 560. The pre-processing engine 550 and the post-processing engine 570 may be realized as DSP. The AI engine 560 can support, in an example embodiment, processing of specific AI and deep-learning tasks.
For example, the AI engine 560 may be configured to train and create neural networks, such as DNN, CNN, RNN or to infer based on existing pre-trained neural networks. The pre-processing engine 550 can be utilized to support processing of tasks that are not related to neural networks, neuromorphic computing, or other AI functions. Such processing may be performed prior to, after or between and other neural network functions to be executed on the AI engine 560. Examples for non-NN AI tasks and functions are, but are not limited to, JPEG image decoding, MPEG video decoding, Resizing, Rendering, Input data modifications, Buffer restructuring, Image processing, Voice processing, Text manipulation, Rule based decision making or data modifications.
In an embodiment, the post-processing engine 570 can perform functions that are not neural network related. Examples for such function include, but are not limited to: image processing, mathematical equations processing, rule based decision making or data modifications, resizing, rendering, output data modifications, buffer restructuring, image processing, voice processing, and text manipulation.
In some embodiments, the NIC 510 may have remote direct memory access (RDMA) capabilities. The eNIC 510 is operative for allowing the NA-AIA engine 422 to connect to the AIA switch (e.g., switch 421,
The AI sequencer 530 is configured to control and manage the pipeline of tasks of jobs between an input direct memory access (DMA) of a packet processor, an external memory interface, and the various processing components of the NA-AIA engine 422. The packet processor is part of the eNIC 510.
In an embodiment, the AI sequencer 530 performs the tasks related to queue management between the various requests from various clients, scheduling of jobs and tasks within jobs, chaining between the processing elements. The sequencer 530 may further perform operations related to stitching the results between processing elements before feeding to another processing elements. In general, the sequencer 530 is configured to manage and control the data path between the processing elements and other components in the system without the intervention of software.
According to some embodiments, the AI sequencer 530 supports execution of deep-neural-network tasks, such as CNN, RNN, LSTM, and the like; Pre and post and intermediate processing; a multi computation graph support, and intra-chip/inter-chip computation graph partitioning (multi-dimensional); intra-chip/inter-chip computation graph chaining (pipelining and sequencing); queue management; and a scheduler.
In an example, the NA-AIA engine 422 can be realized as an FPGA, an ASIC, a system-on-a-chip system (SOC), and the like.
The request sent at S610, may include the data require to execute the AI task or AI sequence. The data copied to a memory (422-M) allocated to a engine 422 assigned to execute the task. The data is copied over a fast network connection, examples of which are provided above.
At S620, the requested AI task is being processed by the NA-AIA assigned for this task. The processing as performed by the NA-AIA engine is discussed with reference to
At S630, the processing results are returned to the application results also over the fast connection. In an embodiment, the results are copied directly from the AIA appliance server 420 (from the NA-AIA engine) to the application server 410. It should be appreciated that the data is copied only once from the server 410 to the appliance 420 over a fast connection and only one on the way back.
The embodiments disclosed herein can be implemented as hardware, firmware, software, or any combination thereof. The application program may be uploaded to, and executed by, a machine comprising any suitable architecture. Examples for such architectures are discussed above with reference to at least
The computer platform may also include an operating system and microinstruction code. The various processes and functions described herein may be either part of the microinstruction code or part of the application program, or any combination thereof, which may be executed by a CPU, whether or not such computer or processor is explicitly shown.
In addition, various other peripheral units may be connected to the computer platform such as an additional network fabric, network interface card or device, storage unit and a printing unit. Furthermore, a non-transitory computer readable medium is any computer readable medium except for a transitory propagating signal.
It should be understood that any reference to an element herein using a designation such as “first,” “second,” and so forth does not generally limit the quantity or order of those elements. Rather, these designations are generally used herein as a convenient method of distinguishing between two or more elements or instances of an element. Thus, a reference to first and second elements does not mean that only two elements may be employed there or that the first element must precede the second element in some manner. Also, unless stated otherwise, a set of elements comprises one or more elements.
As used herein, the phrase “at least one of” followed by a listing of items means that any of the listed items can be utilized individually, or any combination of two or more of the listed items can be utilized. For example, if a system is described as including “at least one of A, B, and C,” the system can include A alone; B alone; C alone; A and B in combination; B and C in combination; A and C in combination; or A, B, and C in combination.
All examples and conditional language recited herein are intended for pedagogical purposes to aid the reader in understanding the principles of the disclosure and the concepts contributed by the inventor to furthering the art, and are to be construed as being without limitation to such specifically recited examples and conditions.
Claims
1. A remote artificial intelligence (AI) acceleration system, comprising:
- a plurality of application servers, wherein each of the plurality of application server is configured to execute AI applications over an AI software framework;
- at least one artificial intelligence accelerator (AIA) appliance server configured to execute AI processing tasks in response to requests from the AI applications; and
- at least one switch configured to allow connectivity between the plurality of application servers and the at least one AIA appliance server, wherein each of the at least one AIA appliance server includes:
- a plurality of network-attached artificial intelligence accelerator (NA-AIA) engines; and
- at least one least one AIA switch directly coupled to the plurality of NA-AIA engines and a network, wherein the at least one AIA switch allows direct communication between the plurality of application servers and the plurality of NA-AIA engines through the network,
- wherein each of the plurality of NA-AIA engines is configured to execute at least one processing AI task upon a request from an AI application executed by an application server, wherein data required for execution of an AI task is directly copied, over the network, from the application server to a NA-AIA engine executing the AI task,
- and wherein each of the plurality of NA-AIA engines includes an embedded network interface controller (eNIC), the eNIC is configured to connect the NA-AIA engine to the AIA switch and to implement a remote direct memory access to allow the application server to copy data directly to the NA-AIA engine over the network.
2. The system of claim 1, wherein the at least one switch is an Ethernet switch.
3. The system of claim 1, wherein the AI software platform is any one of: TensorFlow, Caffe2, Pytorch, and Microsoft Cognitive Toolkit (CNTK).
4. The system of claim 1, wherein each of the plurality of application servers comprises:
- a network interface card (NIC);
- a processor; and
- a memory coupled to the processor.
5. The system of claim 4, wherein the at least one AIA appliance server further comprises:
- a plurality of memory units, each of the plurality of memory units is coupled to a respective NA-AIA engine of the plurality of NA-AIA engines.
6. The system of claim 5, wherein application data of an AI application is directly copied from a memory of an application server executing the AI Application to a memory associated with a respective NA-AIA engine in the AIA appliance server.
7. The system of claim 6, wherein the application data is copied to the NA-AIA engine in response to a requested AI processing task by the AIA appliance server.
8. The system of claim 7, wherein data of processing results is directly copied from the NA-AIA engine to the memory of the AIA appliance server executing the AI Application.
9. The system of claim 1, wherein an AI processing task of the AI processing tasks includes an AI sequence.
10. The system of claim 5, wherein each of the plurality of NA-AIA engines are connected to the at least one AIA switch on-demand.
11. The system of claim 1, wherein each of the plurality of the NA-AIA engines is realized as a system on chip (SoC).
12. The system of claim 11, wherein each of the plurality of the NA-AIA engines comprises:
- a processing circuitry;
- an AI sequencer; and
- a memory.
13. The system of claim 12, wherein each of the plurality of the NA-AIA engines further comprises:
- an AI engine configured to control and manage a pipeline of processing AI task;
- a pre-processing engine configured to preform pre-processing of functions that are not neural network related; and
- a post-processing engine configured to preform post-processing of functions that are not neural network related.
14. The system of claim 1, wherein the communication between the at least one AIA appliance server and an application server is over a network connection, wherein the network connection includes any one of: an Ethernet connection, a Transmission Control Protocol (TCP) connection, a remote direct memory access (RDMA) connection, and a RDMA over Converged Ethernet (RoCE) connection.
15. The system of claim 14, wherein a protocol utilized to copy data over the network connection is an AI over Fabric (AIoF) protocol, wherein the AIoF protocol is a communication protocol designed to support installation and execution of AI models and AI operations.
16. A method for accelerating the operations of artificial intelligence (AI) processing tasks, comprising:
- receiving, from an application server, a request to execute an AI task, the request is received at an artificial intelligence accelerator (AIA) appliance server, wherein the request is copied directly from the application server to the AIA appliance server over a network connection by means of at least one AIA switch connected in the AIA appliance server, wherein the at least one AIA switch allows direct communication between the application server and the AIA appliance server;
- processing the received request by the AIA appliance server; and
- copying the processing results directly to the application server over the network connection.
17. The method of claim 16, wherein the AI task includes an AI sequence.
18. The method of claim 16, wherein the network connection includes any one of: an Ethernet connection, a Transmission Control Protocol (TCP) connection, a remote direct memory access (RDMA) connection, and a RDMA over Converged Ethernet (RoCE) connection.
19. The method of claim 16, wherein a protocol utilized to copy data over the network connection is an AI over Fabric (AIoF) protocol, wherein the AoF protocol is a communication protocol designed to support installation and execution of AI models and AI operations.
20120303932 | November 29, 2012 | Farabet et al. |
20150324690 | November 12, 2015 | Chilimbi et al. |
20170357891 | December 14, 2017 | Judd et al. |
20170357910 | December 14, 2017 | Sommer et al. |
20180089562 | March 29, 2018 | Jin et al. |
20180218518 | August 2, 2018 | Yan et al. |
20180314941 | November 1, 2018 | Lie et al. |
20190042955 | February 7, 2019 | Cahill et al. |
20190325302 | October 24, 2019 | Savic |
20200174840 | June 4, 2020 | Zhao |
20210004658 | January 7, 2021 | Ziv |
108718249 | October 2018 | CN |
- Parashar, et al., “SCNN: An Accelerator for Compressed-sparse Convolutional Neural Networks,” Massachusetts, 2017, pp. 1-12.
- The International Search Report and The Written Opinion of the International Searching Authority for PCT/US2020/019157, dated Jun. 25, 2020, ISA/RU, Moscow, Russia.
Type: Grant
Filed: Jul 1, 2020
Date of Patent: Mar 5, 2024
Patent Publication Number: 20200380361
Assignee: NeuReality LTD. (Bet Herut)
Inventor: Moshe Tanach (Bet Herut)
Primary Examiner: Shirley X Zhang
Application Number: 16/918,491