2024年1月31日
MindSpore导入CUDA算子
摘要： MindSpore导入CUDA算子 本文介绍了在MindSpore标准格式下进行CUDA算子开发的方法和流程，可以让开发者在现有的AI框架下仍然可以调用基于CUDA实现的高性能的算子。并且，除了常规的数值计算之外，在MindSpore框架下，我们还可以通过实现一个bprop函数，使得我们手写的这个CUDA算子也可以使用MindSpore框架本身自带的自动微分-端到端微分技术。 阅读全文
posted @ 2024-01-31 15:29 DECHIN 阅读(332) 评论(0) 推荐(0) 编辑