[发明专利]数据流的时空映射方法、装置、电子设备及存储介质

说明书

本发明公开了一种数据流的时空映射方法、装置、设备及介质。该方法包括：获取待时空映射的目标数据流；根据目标数据流中定义的与该目标子任务匹配的各数据维度的执行顺序，并行维度和各数据维度的切片大小，对目标数据流进行时域映射，得到多个具有时域执行顺序的子数据流；根据目标子任务在目标架构中所映射的目标子系统和目标架构中定义的各子系统的连接关系，对各子数据流再进行空域映射，将在设定时间片下执行的每个子数据流分别映射至目标架构各层级的子系统中，作为该基准测试中的一项建模参考信息。本发明实施例的技术方案可以准确、高效的将数据流任务在时间和空间维度进行双重映射，为后续进行基准测试建模提供数据准备。

技术领域

本发明实施例涉及人工智能芯片架构的建模测试技术，尤其涉及一种数据流的时空映射方法、装置、电子设备及存储介质。

背景技术

在进行AI（Artificial Intelligence，人工智能）芯片的架构探索时，对于给定的多个数据流任务以及特定的某种架构，我们需要通过基准测试评估这些数据流任务在该架构上的执行情况。在进行基准测试建模时，通常需要对特定架构和数据流任务进行比较复杂的建模，然后进行仿真，整体的开发代价和周期都较大。

因此，如何能在进行基准测试之前构建得到有效的建模参考信息，为后续进行灵活性、可扩展性的基准测试建模提供高效可用的数据准备，是目前有待解决的重要问题。

发明内容

本发明实施例提供一种数据流的时空映射方法、装置、电子设备及存储介质，以准确、高效的将数据流任务在时间维度和空间维度进行双重映射，为后续进行基准测试建模提供高效可用的数据准备。

第一方面，本发明实施例提供了一种数据流的时空映射方法，该方法包括：

获取待时空映射的目标数据流，其中，目标数据流由负载任务拆分出的一个目标子任务生成，负载任务为待加载至目标架构执行基准测试的任务；

根据目标数据流中定义的与该目标子任务匹配的各数据维度的执行顺序，并行维度和各数据维度的切片大小，对目标数据流进行时域映射，得到多个具有时域执行顺序的子数据流；

根据目标子任务在目标架构中所映射的目标子系统和目标架构中定义的各子系统的连接关系，对各子数据流再进行空域映射，将在设定时间片下执行的每个子数据流分别映射至目标架构各层级的子系统中，作为该基准测试中的一项建模参考信息。

第二方面，本发明实施例还提供了一种数据流的时空映射装置，该装置包括：

目标数据流获取模块，用于获取待时空映射的目标数据流，其中，目标数据流由负载任务拆分出的一个目标子任务生成，负载任务为待加载至目标架构执行基准测试的任务；

时域映射模块，用于根据目标数据流中定义的与该目标子任务匹配的各数据维度的执行顺序，并行维度和各数据维度的切片大小，对目标数据流进行时域映射，得到多个具有时域执行顺序的子数据流；

空域映射模块，用于根据目标子任务在目标架构中所映射的目标子系统和目标架构中定义的各子系统的连接关系，对各子数据流再进行空域映射，将在设定时间片下执行的每个子数据流分别映射至目标架构各层级的子系统中，作为该基准测试中的一项建模参考信息。

第三方面，本发明实施例还提供了一种电子设备，所述电子设备包括：

至少一个处理器；以及

与所述至少一个处理器通信连接的存储器；其中，

所述存储器存储有可被所述至少一个处理器执行的计算机程序，所述计算机程序被所述至少一个处理器执行，以使所述至少一个处理器能够执行如本发明任一实施例所述的数据流的时空映射方法。