【MindSpore产品】【算子种类】AICPU、GPU算子、TBE算子、计算算子的区别和联系？

最新推荐文章于 2025-03-11 00:42:11 发布

小乐快乐

最新推荐文章于 2025-03-11 00:42:11 发布

阅读量921

点赞数

文章标签： python 人工智能 big data

本文链接：https://blog.csdn.net/weixin_45666880/article/details/125379730

版权

MindSpore的GPU、TBE和AICPU算子主要区别在于执行设备，GPU算子运行于GPU，TBE（AICORE）和AICPU算子则在华为昇腾处理器上运行。了解算子开发，可参考官方提供的AICPU算子开发和TBE算子开发文档。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述：

我了解到MindSpore算子种类中还有诸多其它类型的算子如AICPU、GPU、TBE算子，那么这些算子有哪些区别和联系呢，我应该去哪里获取文件资料？

解答：

你好，计算算子与Mindspore中的GPU、TBE和AICPU算子并不是平行的概念，GPU、TBE和AICPU是以算子的执行设备为区分的，GPU算子是在GPU设备上运行的算子，而TBE算子和AICPU算子是在Ascend设备运行的算子，其中TBE算子也叫AICORE算子。AICPU算子和AICORE算子是以在昇腾处理器上运行计算单元为区分的。详细的资料可以参考： Mindspore算子众智开发指导：https://gitee.com/david-he91/mindspore/wikis/%E4%B8%80%E7%AB%99%E5%BC%8F%E5%AF%BC%E8%88%AA?sort_id=4324745 AICPU算子开发：https://support.huaweicloud.com/aicpu_devg_community_beta/atlasaicpu_10_0007.html TBE算子开发：https://support.huaweicloud.com/odevg-training-cann/atlaste_10_0014.html