Intel Arc GPUs Now Supported In The Intel Extension For PyTorch, Boosting AI, Deep-Learning & LLM Capabilities

Hassan Mujtaba
Intel Arc GPUs Now Supported In The Intel Extension For PyTorch, Boosting AI, Deep-Learning & LLM Capabilities 1

Arc A-Series GPUs are now supported within the Intel Extension for PyTorch (IPEX), offering faster AI capabilities in deep learning & LLMs.

Intel Extension For PyTorch Now Takes Full Advantage of XMX AI Hardware Onboard The Arc A-Series GPUs

Intel's Arc A-Series GPUs based on the Alchemist architecture have loads of on-hardware potential that are getting realized day after day. The software division has done a great job in optimizing the game performance of these GPUs and now the focus is on the emerging AI market where the XMX units featured onboard these chips will get fully utilized.

Related Story Valve Confirms It’s Working With NVIDIA to Bring SteamOS To Even More PC Gamers

In an update, Intel has confirmed that IPEX (Intel Extension for PyTorch) now offers support for Arc A-Series GPUs across Windows, Linux, and WSL2 platforms. With this newly added support, Intel's A Series GPUs would not only boost the acceleration of deep learning models such as LLMs but also bring new optimizations for Intel hardware, enabling faster performance & giving users the ability to run PyTorch models on the Alchemist line of GPUs.

2022-03-30_15-32-19
2022-03-30_15-32-20
2022-03-30_15-32-21
2022-03-30_15-32-23

For those of you who have an Intel Arc A-Series GPU and want to take advantage of their XMX AI engine within PyTorch, the company has posted a full blog on how to run the Llama 2 interface on Windows and WSL2 here. According to Intel's AI software team:

  • GPUs are well suited for large language model (LLM) workloads as GPUs excel at massive data parallelism and high memory bandwidth. The Arc A-Series Graphics, including Arc A770 Graphics, are high-performance graphics, with up to 512 Xe Matrix Extensions (XMX) Engines, providing capability for accelerated execution of deep learning models, including LLMs.
  • IPEX extends PyTorch with the latest performance optimizations for Intel hardware, taking advantage of XMX Engines on Intel discrete GPUs.
  • IPEX Extension for PyTorch enables PyTorch XPU device, which allows it to easily move PyTorch model and input data to the device to execute on Intel discrete GPU with GPU acceleration.
  • The latest release of IPEX (v2.1.10+xpu) officially supports Arc A-Series Graphics on WSL2, native Windows and native Linux.
  • Demonstrated running Llama 2 7B and Llama 2-Chat 7B inference on Arc A770 Graphics on Windows and WSL2 via Ipex.
  • To run Llama 2, or any other PyTorch models, on Arc A-Series GPUs, simply add a few additional lines of code to import intel_extension_for_pytorch and .to("xpu") to move model and data to device to execute on Arc A-Series GPU.

We recently posted an article quoting Intel's ex-GPU boss, Raja Koduri, who mentioned how client and gaming PC GPUs played a crucial role in enabling AI success. Intel expanding support to its Arc A-Series GPUs is a great thing for the market segment where NVIDIA and AMD have been leading the charge by offering swift AI support on consumer-centric hardware.

Hassan Mujtaba Photo

About the author: A Software Engineer by training and a PC enthusiast by passion, Hassan Mujtaba serves as Wccftech's Senior Editor for hardware section. With years of experience in the industry, he specializes in deep-dive technical analysis of next-generation CPU and GPU architectures, motherboards, and cooling solutions. His work involves not only breaking news on upcoming technologies but also extensive hands-on reviews and benchmarking.

Follow Wccftech on Google to get more of our news coverage in your feeds.

Button