🖼️卷积优化

type
status
date
slug
summary
tags
category
icon
password

gemm优化(CPU)

基于算法分析的方法

  • Strassen 算法
  • Coppersmith–Winograd 算法
为什么不用呢?

通过展开循环,尽量减少访存指令占比

naive version:
访存数: 4MNK, 但编译器可能优化到2MNK?
 
version1:
访存数:(8+1+4)/4 MNK = 13/4 MNK(寄存器数量够的话,编译器可能优化到5/4 MNK?)
version2:
访存数:(16*2 + 4 + 4) / 16MNK = 5/2 MMN, 编译器优化后可达:1/2MNK
version3:
访存数:(16*2 + 16 + 16) / 64MNK = 1 MNK, 编译器优化后可达:1/2MNK
如果存在编译器优化的情况下,这一维度的展开还会带来perf 提升吗?

对于支持SIMD的平台,进行向量化的支持

notion image

优化内存布局,减少cache miss

使用图中右边的内存布局,减少一个块内的cache miss以及块间的cache miss
notion image

QNNPACK

卷积优化(CPU)

im2col

notion image

内存布局的影响

NCHW:
 
notion image
NHWC:
notion image
  • 输出:两种布局都一样,因为输出没有访存复用
  • 输入:NHWC优于NCHW,因为前者在小块间的局部性更高
  • filter:NCHW优于NHWC,因为前者在小块间的局部性更高。但是通常filter可以在模型准备阶段转换内存布局,因为一般filter的内容是固定的

空间组合优化算法

notion image
将大输入的tensor划分成小的tensor,卷积后的得到部分输出tensor,再组合起来。划分的目的是为了增加局部性。同时由于卷积计算的特点,以及padding的存在,划分粒度越细,额外的内存开销就越大。
当划分粒度最细时,退化到成im2col
最优的划分粒度对于不同规模的卷积,不同的架构平台可能都是不同的 ,该粒度的寻找可以通过自动化的方式完成:autoTvm.

Winograd算法

1D

notion image

2D

notion image

计算量的讨论

工程实现

误差来源

计算机领域的浮点算术本来就存在精度误差,导致即使在数学上等价的计算,通过计算机实际得到的结果可能不一致。

间接卷积优化算法

im2col的过程需要额外的内存空间开销,以及数据拷贝的时间开销。而间接卷积优化算法则使用一种间接寻址的方法避免了这两个问题。
notion image
并且也可以巧妙的解决pading带来的拷贝问题。
problem:
  • 这种方法的cache友好度如何?
  • 通过使一次计算M*N所需要的输入数据重排为连续的来提升cache友好度?

SGEMM优化(GPU

 
notion image

global memory → shared memory

因为shared memory在一个block内共享,让一个block计算bm*bn大小的输出
notion image
global memory访存量从m*n*k变为 m*n*k(1/bm + 1/bn)

shared memory → register

一个thread内,充分利用register, 减少访问shared memory。
一个thread计算rm*rn大小的输出。
notion image

register分块

数据的prefetch

实际就是让不同模块并行起来

针对cache的优化?

TVM FuseMLIR Operation内存布局
Loading...
Catalog