矩阵行方向最大值确定方法、电子设备及存储介质

AITNT
正文
推荐专利
矩阵行方向最大值确定方法、电子设备及存储介质
申请号:CN202510637126
申请日期:2025-05-16
公开号:CN120508389A
公开日期:2025-08-19
类型:发明专利
摘要
本发明涉及人工智能技术领域,提供一种矩阵行方向最大值确定方法、电子设备及存储介质,其中方法包括:在人工智能芯片的通用矩阵主缓存中申请目标矩阵对应的矩阵布局空间,在高带宽内存中申请目标矩阵对应的矢量布局空间;将目标矩阵行方向最大值确定过程中产生的中间结果覆盖写入目标矩阵对应的矩阵布局空间;在确定完成目标矩阵任一行中元素的最大值的计算的情况下,将目标矩阵对应的矩阵布局空间中上述行对应的元素覆盖写入目标矩阵对应的矢量布局空间。本发明能在矩阵行方向最大值确定的过程中避免触发广播机制,并通过将临时数据存储至通用矩阵主缓存,提高矩阵行方向最大值确定的计算效率和降低人工智能芯片的硬件计算资源占用。
技术关键词
矩阵 人工智能芯片 布局 元素 高带宽 非暂态计算机可读存储介质 电子设备 处理器 内存 人工智能技术 计算机程序产品 数据存储 存储器 机制
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号