华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
pipe.InitBuffer(inQueueA1, 1, aSize * sizeof(half)); pipe.InitBuffer(inQueueA2, 1, aSize * sizeof(half)); pipe.InitBuffer(inQueueB1, 1, bSize * sizeof(half)); pipe.InitBuffer(inQueueB2, 2, bSize * sizeof(half) / 2); pipe.InitBuffer(outQueueCO1, 2, cSize * sizeof(float) / 2); pipe.InitBuffer(outQueueCO2, 1, cSize * sizeof(float));
Aicore中这几个位置的大小可以开多大呢？为啥用Mmad无法一次性做M=256,N=256,K=256的float矩阵乘法呢？
我要发帖子
点赞
收藏
回复
分享
举报
已解决 已解决
pipe.InitBuffer(inQueueA1, 1, aSize * sizeof(half)); pipe.InitBuffer(inQueueA2, 1, aSize * sizeof(half)); pipe.InitBuffer(inQueueB1, 1, bSize * sizeof(half)); pipe.InitBuffer(inQueueB2, 2, bSize * sizeof(half) / 2); pipe.InitBuffer(outQueueCO1, 2, cSize * sizeof(float) / 2); pipe.InitBuffer(outQueueCO2, 1, cSize * sizeof(float));Aicore中这几个位置的大小可以开多大呢？为啥用Mmad无法一次性做M=256,N=256,K=256的float矩阵乘法呢？