技术栈
winograd
Briwisdom
2 小时前
gemm
·
fft
·
matmul
·
conv2d
·
implicit gemm
·
winograd
Conv2D 算子实现深入——im2col、Implicit GEMM、Winograd、FFT 代码级拆解
副标题: 一个卷积算子,四种实现路线。本文不再停留在概念层面,而是用可运行的代码逐一拆解:直接卷积怎么算、im2col 怎么展开成矩阵、Implicit GEMM 怎么在 kernel 内部在线计算访存地址、Winograd F(2,3) 的三组变换矩阵从哪来、FFT 怎么利用卷积定理做一次频域点乘。训练 vs 推理的差异分析见 博客 #34。
我是有底线的