6月 30, 2022
MindSpore算子开发——CUDA代码优化
CUDA编程入门：谭升的博客 对于cuda编程，核心是如何高效率地利用多线程，每个线程完成一个小任务，最终实现完成一个算子任务。 一个核函数只能有一个grid，...
345 0
4月 22, 2022
MindSpore算子开发——GPU算子设计文档
首先下载算子设计文档模板：GPU算子设计文档模板（点击下载） （以下内容以算子FractionalAvgPool为例） 准备部分 首先通过搜索引擎查一下相应的算...
829 2