昇腾论坛

点赞

收藏

回复

举报

mindspore(Ascend 910)的矩阵乘法算子过慢，与pytorch(Titan XP)相差几十倍，有没有什么优化方式？

新人帖

发表于2022-11-11 22:22:16

0 查看

上图是mindspore的矩阵乘法耗时，对应算子为ops.BatchMatMul()与ops.MatMul()

而GPU仅需0.183ms

本帖最后由匿名用户于 2022/12/28 14:29:08 编辑

匿名回复

我要发帖子