oneDNN Graph Compiler: A Hybrid Approach for High-Performance Deep Learning Compilation
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2023
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866911791942991872 |
|---|---|
| author | Li, Jianhui Qin, Zhennan Mei, Yijie Cui, Jingze Song, Yunfei Chen, Ciyong Zhang, Yifei Du, Longsheng Cheng, Xianhang Jin, Baihui Zhang, Yan Ye, Jason Lin, Eric Lavery, Dan |
| author_facet | Li, Jianhui Qin, Zhennan Mei, Yijie Cui, Jingze Song, Yunfei Chen, Ciyong Zhang, Yifei Du, Longsheng Cheng, Xianhang Jin, Baihui Zhang, Yan Ye, Jason Lin, Eric Lavery, Dan |
| contents | With the rapid development of deep learning models and hardware support for dense computing, the deep learning workload characteristics changed significantly from a few hot spots on compute-intensive operations to a broad range of operations scattered across the models. Accelerating a few compute-intensive operations using the expert-tuned implementation of primitives does not fully exploit the performance potential of AI hardware. Various efforts have been made to compile a full deep neural network (DNN) graph. One of the biggest challenges is to achieve high-performance tensor compilation by generating expert level performance code for the dense compute-intensive operations and applying compilation optimization at the scope of DNN computation graph across multiple compute-intensive operations.
We present oneDNN Graph Compiler, a tensor compiler that employs a hybrid approach of using techniques from both compiler optimization and expert-tuned kernels for high performance code generation of the deep neural network graph. oneDNN Graph Compiler addresses unique optimization challenges in the deep learning domain, such as low-precision computation, aggressive fusion of graph operations, optimization for static tensor shapes and memory layout, constant weight optimization, and memory buffer reuse. Experimental results demonstrate significant performance gains over existing tensor compiler and primitives library for performance-critical DNN computation graphs and end-to-end models on Intel Xeon Scalable Processors. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2301_01333 |
| institution | arXiv |
| publishDate | 2023 |
| record_format | arxiv |
| spellingShingle | oneDNN Graph Compiler: A Hybrid Approach for High-Performance Deep Learning Compilation Li, Jianhui Qin, Zhennan Mei, Yijie Cui, Jingze Song, Yunfei Chen, Ciyong Zhang, Yifei Du, Longsheng Cheng, Xianhang Jin, Baihui Zhang, Yan Ye, Jason Lin, Eric Lavery, Dan Machine Learning Performance With the rapid development of deep learning models and hardware support for dense computing, the deep learning workload characteristics changed significantly from a few hot spots on compute-intensive operations to a broad range of operations scattered across the models. Accelerating a few compute-intensive operations using the expert-tuned implementation of primitives does not fully exploit the performance potential of AI hardware. Various efforts have been made to compile a full deep neural network (DNN) graph. One of the biggest challenges is to achieve high-performance tensor compilation by generating expert level performance code for the dense compute-intensive operations and applying compilation optimization at the scope of DNN computation graph across multiple compute-intensive operations. We present oneDNN Graph Compiler, a tensor compiler that employs a hybrid approach of using techniques from both compiler optimization and expert-tuned kernels for high performance code generation of the deep neural network graph. oneDNN Graph Compiler addresses unique optimization challenges in the deep learning domain, such as low-precision computation, aggressive fusion of graph operations, optimization for static tensor shapes and memory layout, constant weight optimization, and memory buffer reuse. Experimental results demonstrate significant performance gains over existing tensor compiler and primitives library for performance-critical DNN computation graphs and end-to-end models on Intel Xeon Scalable Processors. |
| title | oneDNN Graph Compiler: A Hybrid Approach for High-Performance Deep Learning Compilation |
| topic | Machine Learning Performance |
| url | https://arxiv.org/abs/2301.01333 |