高效、易用、可拓展我全都要:OneFlow CUDA Elementwise模板库的设计优化思路 OneFlow 针对这种 Elementwise 操作抽象出一套 CUDA 模板,开发者只需把计算逻辑封装到一个结构体内,即可获得一个 CUDA Elementwise 算子 前沿技术 2024-01-26 41 点赞 0 评论 62 浏览