Intel® MKL Packed on PaddlePaddle: Design Doc

Overview

我们计划将 Intel® MKL 中引入的 GEMM Packed APIs[1] 集成到 PaddlePaddle 中，充分发挥英特尔平台的优势，有效提升PaddlePaddle在英特尔架构上的性能。 现阶段的优化主要针对 Recurrent Neural Network（以下简称RNN）相关层（包括RecurrentLayer, GatedRecurrentLayer和LstmLayer）， 以及 PaddlePaddle V1 API。

Key Points

Background

目前PaddlePaddle采用了 Intel® MKL库的cblas_?gemm函数，这个函数本身会在计算前将原数据转换为更适合英特尔平台的内部格式。

  1. 转换耗时 这一数据格式的转换操作（Packing），在问题本身的计算量比较小的时候，显得相对来说较为耗时。例如在DeepSpeech2 [2] 的Vanilla RNN部分中，矩阵大小是batch_size * 2048。
  2. 转换冗余 由于在现有的某些情况下（例如RNN），多次调用 cblas_?gemm 会使用相同的原数据，因此，每次调用时对原数据的重复Packing便成为了冗余。

为了最大程度减少多次调用 cblas_?gemm 在Packing上的耗时，Intel® MKL 引入了以下四个API:

  • cblas_?gemm_alloc
  • cblas_?gemm_pack
  • cblas_?gemm_compute
  • cblas_?gemm_free

通过使用这些API，我们可以先完成对原数据的Packing操作，再把已转换为Packed格式的数据传递给那些复用同一数据的gemm_compute函数，从而避免了Packing冗余。

Solution

在RNN的情况下，同一次前向、后向（forward/backward）过程中所有时间步（time step）共享同一个权重（weight）。当只做推断（inference）时，各次前向之间也都使用了相同的权重，没有必要在每次前向中每个时间步的计算时对权重进行重复的Packing操作。

我们通过使用新引入的GEMM Packed APIs，在层初始化的时候，先完成对权重的Packing操作，然后在前向，后向时复用已经转换过的权重，并在每次权重更新后，对新的权重进行转换用于下次迭代。

  • 优化前，对于序列长度（sequence length）为T的网络模型（model）, N次迭代执行的转换次数为：
    • inference： N * T
    • training： 2 * N * T
  • 优化后，对于同样设置的网络模型，其转换次数减少至：
    • inference： 1
    • training： 2 * N

Actions

添加的相关文件和目录结构如下：

PaddlePaddle/Paddle
├── ...
└── paddle/
    ├── ...
    └── gserver/
        ├── ...
        ├── layers/
        │   ├── ...
        │   ├── MKLPackedRecurrentLayer.*
        |   ├── MKLPackedGatedRecurrentLayer.*
        |   ├── MKLPackedLstmLayer.*
        |   └── MKLPackedGemm.h
        └── tests/
            ├── ...
            └── test_MKLPacked.cpp

CMake

在对应的CMakeLists.txt中根据WITH_MKL是否打开，来决定是否开启MKL Packed相关功能。

Layers

所有的MKLPacked*Layer都继承于PaddlePaddle的基类Layer, 并添加头文件 MKLPackedGemm.h，该文件对相关GEMM Packed APIs做了封装。

Unit Tests

我们会添加test_MKLPacked.cpp用于MKL Packed优化后layer的测试。 对于每一个新加的RNN layer，我们会对比如下2个方面：

  1. 对比优化后layer自身，sequence mode（rnn_use_batch=false）与batch mode(rnn_use_batch=true)的结果。
  2. 对比优化后layer与相对应的PaddlePaddle原有layer, 在batch mode下的结果。

Python API

TBD

Benchmarking

会添加相应的脚本用于测试和对比在使用MKL Packed recurrent layers 前后的网络性能。