[发明专利]基于DMA架构的SG指令实现方法、装置、设备及存储介质

说明书

本发明实施例提供一种基于DMA架构的SG指令实现方法、装置、设备及存储介质，涉及神经网络技术领域。所述方法基于SG指令，获取索引张量数据，并对索引张量数据进行解析，获得索引数据的索引数据读取地址；基于所述索引数据的数据读取地址，读取索引数据；基于所述索引数据，计算数据读取地址和所述数据写入地址；基于所述数据读取地址，从源数据存储单元中读取目标数据，并基于所述数据写入地址，将所述目标数据写入目标存储单元中，以完成数据迁移。通过对多维的索引张量数据的解析，计算数据读取地址和数据写入地址，从而在源数据存储单元中读取目标数据，并搬迁至目标存储单元中，避免了对复杂数据的解析，提高了数据搬迁效率。

技术领域

本发明涉及神经网络技术领域，尤其涉及一种基于DMA架构的SG指令实现方法、装置、设备及存储介质。

背景技术

数据迁移是一种将离线存储与在线存储融合的技术。它将高速、高容量的非在线存储设备作为磁盘设备的下一级设备，然后将磁盘中常用的数据按指定的策略自动迁移到磁带库(简称带库)等二级大容量存储设备上。DMA(直接存储器存取，Direct MemoryAccess)主要是实现数据的高速搬运，为CPU(中央处理器，Central Processing Unit)完成简单性重复性数据搬运工作。这个过程无需CPU干预。

但是目前通过DMA进行数据迁移时，如果需要接收大量的数据，则需要频繁触发中断，且需要CPU参与，从而使得数据搬移的效率降低。因此，如何解决数据搬迁效率低下成为目前亟待解决的技术问题。

发明内容

本发明实施例提供了一种基于DMA架构的SG指令实现方法、装置、设备及存储介质，旨在提高数据的搬迁效率。

第一方面，本发明实施例提供一种基于DMA架构的SG指令实现方法，包括：

基于SG指令，获取索引张量数据，并对索引张量数据进行解析，获得索引数据的索引数据读取地址；

基于所述索引数据的数据读取地址，读取索引数据；

基于所述索引数据，计算数据读取地址和所述数据写入地址；

基于所述数据读取地址，从源数据存储单元中读取目标数据，并基于所述数据写入地址，将所述目标数据写入目标存储单元中，以完成数据迁移。

第二方面，本发明实施例还提供一种基于DMA架构的SG指令实现装置，所述基于DMA架构的SG指令实现装置包括：

索引张量数据解析模块，用于基于SG指令，获取索引张量数据，并对索引张量数据进行解析，获得索引数据的索引数据读取地址；

索引数据读取模块，用于基于所述索引数据的数据读取地址，读取索引数据；

数据读写地址计算模块，用于基于所述索引数据，计算数据读取地址和所述数据写入地址；

数据迁移模块，用于基于所述数据读取地址，从源数据存储单元中读取目标数据，并基于所述数据写入地址，将所述目标数据写入目标存储单元中，以完成数据迁移。

第三方面，本发明实施例还提供一种计算机设备，所述计算机设备包括处理器、存储器、存储在所述存储器上并可被所述处理器执行的计算机程序以及用于实现所述处理器和所述存储器之间的连接通信的数据总线，其中所述计算机程序被所述处理器执行时，实现如本发明说明书提供的任一项基于DMA架构的SG指令实现方法。

第四方面，本发明实施例还提供一种存储介质，用于计算机可读存储，所述存储介质存储有一个或者多个程序，所述一个或者多个程序可被一个或者多个处理器执行，以实现如本发明说明书提供的任一项基于DMA架构的SG指令实现方法。