[发明专利]数据处理装置、集成电路和AI加速器

说明书

本公开实施例提供一种数据处理装置、集成电路和AI加速器，包括多个堆存储单元，每个堆存储单元用于存储堆的一组节点的数据，所述一组节点中包括所述堆的同一层节点中的至少部分节点；以及多个堆调整单元，每个堆调整单元用于访问至少两个堆存储单元，以对输入所述每个堆调整单元的原始数据与所述至少两个堆存储单元中存储的数据进行排序。

技术领域

本公开涉及数据处理技术领域，尤其涉及数据处理装置、集成电路和AI加速器。

背景技术

在许多算法或者模型中，经常需要处理排序问题，堆排序被广泛用于处理排序问题。堆排序是指利用堆这种数据结构所设计的一种排序方式。在传统的堆排序方式中，排序效率较低。

发明内容

本公开提供一种数据处理装置、集成电路和AI加速器。

根据本公开实施例的第一方面，提供一种数据处理装置，所述装置包括：多个堆存储单元，每个堆存储单元用于存储堆的一组节点的数据，所述一组节点中包括所述堆的同一层节点中的至少部分节点；以及多个堆调整单元，每个堆调整单元用于访问至少两个堆存储单元，以对输入所述每个堆调整单元的原始数据与所述至少两个堆存储单元中存储的数据进行排序。

在一些实施例中，一个堆存储单元用于存储同一层的所有节点的数据。

在一些实施例中，每个堆调整单元访问的所述至少两个堆存储单元用于存储所述堆的相邻层节点的数据。

在一些实施例中，一个节点的所有子节点的数据可以同时或一次读取。

在一些实施例中，所述多个堆调整单元中每个堆调整单用于获取数据，并将获取到的数据和访问的所述至少两个堆存储单元中的至少一个堆存储单元中的数据进行排序。

在一些实施例中，所述多个堆调整单元中两个相邻堆调整单元分别访问的两个堆存储单元中包括一个相同的堆存储单元。

在一些实施例中，所述至少两个第一堆调整单元中的任意两个第一堆调整单元之间间隔至少一个堆调整单元。

在一些实施例中，所述多个堆调整单元中的至少两个第一堆调整单元并行地进行排序，所述至少两个第一堆调整单元访问的堆存储单元互不相同。

在一些实施例中，相邻两个数据的进堆时序之间间隔两个时钟周期。

在一些实施例中，在某个原始数据输入到顺序连接的所述多个堆调整单元中的第二个堆调整单元时，所述原始数据的下一数据输入到所述多个堆调整单元的第一个堆调整单元。

在一些实施例中，所述堆的同一节点的各个子节点的数据存储在同一堆存储单元的同一地址中。

在一些实施例中，所述装置还包括：预处理单元，用于对从数据存储装置获取的原始数据进行预筛选处理。

在一些实施例中，所述预处理单元用于在所述堆存储单元中存储的数据达到预设数量的情况下，对新获取到的所述原始数据进行预筛选处理。

在一些实施例中，所述预处理单元用于通过比较所述原始数据与所述堆的根节点的数据，对所述原始数据进行预筛选处理，以预先判定所述原始数据是否需要进堆。

在一些实施例中，所述预处理单元的数量为多个，多个所述预处理单元用于并行地对获取到的原始数据进行预筛选处理。

在一些实施例中，所述预处理单元用于：在判定所述原始数据需要进堆的情况下，将所述原始数据传输至缓存单元或所述多个堆调整单元。

在判定所述原始数据不需要进堆的情况下，将所述原始数据删除或返回所述数据存储装置。

在一些实施例中，所述多个堆调整单元还用于：在所述多个堆存储单元中的数据均排序完成的情况下，对返回所述数据存储装置的原始数据进行再次排序。