MegEngine
Known Issue
训练和推理的GPU显存占用(MiB)各模型有不同程度的增加。
New Features
CUDA
- 添加大卷积核的 direct conv 实现。
- 添加 implicit bmm 大卷积核 depthwise conv 的实现。
MegEngine
New Features
CUDA
- Add implementation of large kernel's direct conv algo.
- Add implementation of large kernel's depthwise conv by implicit bmm.