自适应检查数据的比特位宽的装置、方法及设备与流程

本技术涉及数据处理，特别涉及一种自适应检查数据的比特位宽的装置、方法及设备。

背景技术：

1、npu（natural processing unit，神经网络处理器）处理数据（即输入数据或输出数据）的维度包括多个。一般硬件处理或者软件处理时出于提升运行效率或者降低代码难度等目的，会对各个维度对应的比特位宽有一定要求，特别是硬件处理时，如果各个维度数据对应的各种比特位宽均可处理，会大大增加芯片的面积和功耗。

2、为了解决上述技术问题，相关技术中，软件硬件工作人员需要对各个维度数据的比特宽度要求进行频繁交流。

3、但是，由于上述方法中算子种类数量繁多，各个维度数据比特位宽情况繁多，进而会造成软硬件工作人员的沟通成本急剧增加。此外，上述方法由于没有固定的机制对各个维度数据的位宽进行约束，导致不同项目对设计要求不同，造成已有工作可继承至下个项目的较少，降低了工作效率。

技术实现思路

1、本技术实施例提供了一种自适应检查数据的比特位宽的装置、方法及设备，能够降低软件硬件工作人员的沟通成本，且提供了对各个维度数据的位宽进行约束的预设规则，从而增加了项目的继承性，进而提高了工作效率。该技术方案如下：

2、根据本技术实施例的第一方面，提供了一种自适应检查数据的比特位宽的装置，该装置包括：

3、顺次电连接的获取模块、填充模块、存储模块以及读取器；

4、该获取模块，用于获取待处理图像以及该待处理图像每个维度的初始比特位宽；

5、该填充模块，用于填充每个该维度的该初始比特位宽，得到每个该维度的目标比特位宽；根据每个该维度的该目标比特位宽以及该待处理图像，生成填充图像；该填充图像为填充了边缘区域的该待处理图像；该目标比特位宽大于或等于第一预设阈值，小于第二预设阈值，且是第三预设阈值的幂次方，或该目标比特位宽大于该第二预设阈值，且是该第二预设阈值的整数倍，或等于该第二预设阈值；

6、该存储模块，用于存储该填充图像以及每个该维度的该初始比特位宽；

7、该读取器，用于从该存储模块中读取该填充图像，根据每个该维度的该初始比特位宽以及该填充图像，确定该待处理图像，处理该待处理图像，得到该目标图像；

8、该存储模块，还用于存储该目标图像。

9、在一些可能的实现方式中，该维度包括该待处理图像的通道数量、该待处理图像的宽度、该待处理图像的高度以及该待处理图像的数量。

10、在一些可能的实现方式中，该填充模块，用于获取每个通道的通道比特位宽，基于该通道比特位宽以及该通道数量之积，得到该通道数量的该初始比特位宽；

11、将该通道数量的该初始比特位宽填充至该通道数量的该目标比特位宽；

12、将该通道数量的该目标比特位宽以及该宽度之积，作为该宽度的该初始比特位宽；

13、将该宽度的该初始比特位宽填充至该宽度的该目标比特位宽；

14、将该宽度的该目标比特位宽以及该高度之积，作为该高度的该初始比特位宽；

15、将该高度的该初始比特位宽填充至该高度的该目标比特位宽；

16、将该高度的该目标比特位宽以及该数量之积，作为该数量的该初始比特位宽；

17、将该数量的该初始比特位宽填充至该数量的该目标比特位宽。

18、在一些可能的实现方式中，该填充模块，还用于：若该通道数量的该初始比特位宽小于该第一预设阈值，则将该通道数量的该初始比特位宽填充至该通道数量的该目标比特位宽，该通道数量的该目标比特位宽等于该第一预设阈值；或

19、若该通道数量的该初始比特位宽大于该第一预设阈值，小于该第二预设阈值且不是该第三预设阈值的幂次方，则将该通道数量的该初始比特位宽填充至该通道数量的该目标比特位宽，该通道数量的该目标比特位宽为该第三预设阈值的幂次方；或

20、若该通道数量的该初始比特位宽大于该第二预设阈值，且不是该第二预设阈值的整数倍，则将该通道数量的该初始比特位宽填充至该通道数量的该目标比特位宽，该通道数量的该目标比特位宽为该第二预设阈值的整数倍；

21、若该宽度的该初始比特位宽小于该第一预设阈值，则将该宽度的该初始比特位宽填充至该宽度的该目标比特位宽，该宽度的该目标比特位宽等于该第一预设阈值；或

22、若该宽度的该初始比特位宽大于该第一预设阈值，小于该第二预设阈值且不是该第三预设阈值的幂次方，则将该宽度的该初始比特位宽填充至该宽度的该目标比特位宽，该宽度的该目标比特位宽为该第三预设阈值的幂次方；或

23、若该宽度的该初始比特位宽大于该第二预设阈值，且不是该第二预设阈值的整数倍，则将该宽度的该初始比特位宽填充至该宽度的该目标比特位宽，该宽度的该目标比特位宽为该第二预设阈值的整数倍；

24、若该高度的该初始比特位宽小于该第一预设阈值，则将该高度的该初始比特位宽填充至该高度的该目标比特位宽，该高度的该目标比特位宽等于该第一预设阈值；或

25、若该高度的该初始比特位宽大于该第一预设阈值，小于该第二预设阈值，且不是该第三预设阈值的幂次方，则将该高度的该初始比特位宽填充至该高度的该目标比特位宽，该高度的该目标比特位宽为该第三预设阈值的幂次方；或

26、若该高度的该初始比特位宽大于该第二预设阈值，且不是该第二预设阈值的整数倍时，则将该高度的该初始比特位宽填充至该高度的该目标比特位宽，该高度的该目标比特位宽为该第二预设阈值的整数倍；

27、若该数量的该初始比特位宽小于该第一预设阈值，则将该数量的该初始比特位宽填充至该数量的该目标比特位宽，该数量的该目标比特位宽等于该第一预设阈值；或

28、若该数量的该初始比特位宽大于该第一预设阈值，小于该第二预设阈值且不是该第三预设阈值的幂次方，则将该数量的该初始比特位宽填充至该数量的该目标比特位宽，该数量的该目标比特位宽为该第三预设阈值的幂次方；或

29、若该数量的该初始比特位宽大于该第二预设阈值，且不是该第二预设阈值的整数倍时，则将该数量的该初始比特位宽填充至该数量的该目标比特位宽，该数量的该目标比特位宽为该第二预设阈值的整数倍。

30、在一些可能的实现方式中，该装置，还包括：

31、若该通道数量的该初始比特位宽等于该第一预设阈值，则将该通道数量的该初始比特位宽，作为该通道数量的该目标比特位宽。

32、在一些可能的实现方式中，该装置，还包括：

33、若该通道数量的该初始比特位宽大于该第一预设阈值，小于该第二预设阈值，且是该第三预设阈值的幂次方，则将该通道数量的该初始比特位宽，作为该通道数量的该目标比特位宽。

34、在一些可能的实现方式中，该装置，还包括：

35、若该通道数量的该初始比特位宽大于该第二预设阈值，且是该第二预设阈值的整数倍，或等于该第二预设阈值，则将该通道数量的该初始比特位宽，作为该通道数量的该目标比特位宽。

36、在一些可能的实现方式中，该装置，还包括：

37、若该宽度的该初始比特位宽等于该第一预设阈值，则将该宽度的该初始比特位宽，作为该宽度的该目标比特位宽。

38、在一些可能的实现方式中，该装置，还包括：

39、若该宽度的该初始比特位宽大于该第一预设阈值，小于该第二预设阈值，且是该第三预设阈值的幂次方，则将该宽度的该初始比特位宽，作为该宽度的该目标比特位宽。

40、在一些可能的实现方式中，该装置，还包括：

41、若该宽度的该初始比特位宽大于该第二预设阈值，且是该第二预设阈值的整数倍，或等于该第二预设阈值，则将该宽度的该初始比特位宽，作为该宽度的该目标比特位宽。

42、在一些可能的实现方式中，该装置，还包括：

43、若该高度的该初始比特位宽等于该第一预设阈值，则将该高度的该初始比特位宽，作为该高度的该目标比特位宽。

44、在一些可能的实现方式中，该装置，还包括：

45、若该高度的该初始比特位宽大于该第一预设阈值，小于该第二预设阈值，且是该第三预设阈值的幂次方，则将该高度的该初始比特位宽，作为该高度的该目标比特位宽。

46、在一些可能的实现方式中，该装置，还包括：

47、若该高度的该初始比特位宽大于该第二预设阈值，且是该第二预设阈值的整数倍，或等于该第二预设阈值，则将该高度的该初始比特位宽，作为该高度的该目标比特位宽。

48、在一些可能的实现方式中，该装置，还包括：

49、若该数量的该初始比特位宽等于该第一预设阈值，则将该数量的该初始比特位宽，作为该数量的该目标比特位宽。

50、在一些可能的实现方式中，该装置，还包括：

51、若该数量的该初始比特位宽大于该第一预设阈值，小于该第二预设阈值，且是该第三预设阈值的幂次方，则将该数量的该初始比特位宽，作为该数量的该目标比特位宽。

52、在一些可能的实现方式中，该装置，还包括：

53、若该数量的该初始比特位宽大于该第二预设阈值，且是该第二预设阈值的整数倍，或等于该第二预设阈值，则将该数量的该初始比特位宽，作为该数量的该目标比特位宽。

54、在一些可能的实现方式中，该读取器，还用于在得到该目标图像之后，填充该目标图像，得到待存储图像，该待存储图像的每个该维度的比特位宽为各个该维度的该目标比特位宽。

55、在一些可能的实现方式中，该第一预设阈值为8比特，该第二预设阈值为256比特，该第三预设阈值为2。

56、根据本技术实施例的第二方面，提供了一种自适应检查数据的比特位宽的方法，应用于上述的自适应检查数据的比特位宽的装置，该方法包括：

57、获取待处理图像以及该待处理图像每个维度的初始比特位宽；

58、获取每个该维度的初始比特位宽；

59、填充每个该维度的该初始比特位宽，得到每个该维度的目标比特位宽；该目标比特位宽大于或等于第一预设阈值，小于第二预设阈值，且是第三预设阈值的幂次方，或该目标比特位宽大于该第二预设阈值，且是该第二预设阈值的整数倍，或等于该第二预设阈值；

60、根据每个该维度的该目标比特位宽以及该待处理图像，生成填充图像，该填充图像为填充了边缘区域的该待处理图像；

61、存储该填充图像以及每个该维度的该初始比特位宽；

62、读取该填充图像；

63、根据每个该维度的该初始比特位宽以及该填充图像，确定该待处理图像，处理该待处理图像，得到该目标图像；

64、存储该目标图像。

65、根据本技术实施例的第三方面，提供了一种计算机设备，该计算机设备包括处理器和存储器，该存储器用于存储至少一段程序，该至少一段程序由该处理器加载并执行上述的自适应检查数据的比特位宽的方法。

66、根据本技术实施例的第四方面，提供了一种计算机可读存储介质，该计算机可读存储介质中存储有至少一段程序，该至少一段程序由处理器加载并执行以实现上述的自适应检查数据的比特位宽的方法。

67、在本技术实施例中，本技术实施例提供了一种自适应检查数据的比特位宽的装置，设置了预设规则，预设规则用于对待处理图像的各个维度数据的比特位宽进行约束，基于预设规则自适应检查待处理图像的各个维度数据的比特位宽，若维度数据的比特位宽不满足预设规则，则将维度数据的比特位宽调整至满足预设规则，存储满足预设规则的维度数据，读取存储的满足预设规则的维度数据。通过上述分析，本技术的实施例能够降低软硬件工作人员的沟通成本，实现了软硬件之间的对齐要求。此外，基于预设规则对各个维度数据的位宽进行约束，从而增加了项目的继承性，进而提高了工作效率。