Home avatar

Wentao's Blog

TVM: 2D Depth Conv GPU Optimization

This blog demonstrates optimization techniques for 2D depth Convolution in GPU using TVM, including block and thread organization, memory hierarchy exploitation and dimension fuse, etc.