摘要
本发明涉及人工智能技术领域,提供一种矩阵行方向最大值确定方法、电子设备及存储介质,其中方法包括:在人工智能芯片的通用矩阵主缓存中申请目标矩阵对应的矩阵布局空间,在高带宽内存中申请目标矩阵对应的矢量布局空间;将目标矩阵行方向最大值确定过程中产生的中间结果覆盖写入目标矩阵对应的矩阵布局空间;在确定完成目标矩阵任一行中元素的最大值的计算的情况下,将目标矩阵对应的矩阵布局空间中上述行对应的元素覆盖写入目标矩阵对应的矢量布局空间。本发明能在矩阵行方向最大值确定的过程中避免触发广播机制,并通过将临时数据存储至通用矩阵主缓存,提高矩阵行方向最大值确定的计算效率和降低人工智能芯片的硬件计算资源占用。
技术关键词
矩阵
人工智能芯片
布局
元素
高带宽
非暂态计算机可读存储介质
电子设备
处理器
内存
人工智能技术
计算机程序产品
数据存储
存储器
机制