oneAPI Community

Accelerate PyTorch with IPEX and oneDNN

Enjoy Ease of Use and More Efficient Hardware Performance

Accelerate PyTorch Figure

Figure 1. AVX512_BF16 instruction set introduced in 3rd gen Intel Xeon Processor adds new instructions for converting and operating on bfloat16

Intel and Facebook collaborate closely on optimizations to the PyTorch community that will one day drive Intel product advancements.  Intel Extension for PyTorch (IPEX) relies on the open-source, cross-platform oneAPI Deep Neural Network (oneDNN) performance library of basic building blocks for deep learning applications optimized for Intel architecture.  This post details two new software advancements added to IPEX on top of PyTorch and oneDNN for PyTorch BF16 CPU optimization: one that frees users from model changes by automating certain conversion processes, and another that optimizes graph fusion to boost BF16 inference performance with oneDNN fusion kernels.

Read Full Article 


Learn DPC++ for Free

Learn More