[发明专利]一种显存分配方法以及装置

权利要求书

1.一种显存分配方法，其特征在于，包括：

根据虚拟GPU所需显存的容量要求，为所述虚拟GPU分配至少两个显存段；其中，所述显存段是预先对物理GPU所对应的显存进行划分而获得的；所述为所述虚拟GPU分配至少两个显存段，包括：按所述虚拟GPU所需显存的容量要求和所述显存段的存储容量计算所述虚拟GPU所需显存段的数量；从所述物理GPU所对应的显存的剩余显存段中，按所述所需显存段的数量选取显存段；所述剩余显存段是指所述物理GPU所对应的显存空间中未分配给其它虚拟GPU的、或者其它虚拟GPU释放后所获得的可用于重新分配的显存段；

保存为所述虚拟GPU分配的至少两个显存段的物理地址。

2.根据权利要求1所述的显存分配方法，其特征在于，所述虚拟GPU对应设置有分段页表，所述为所述虚拟GPU分配的至少两个显存段的物理地址存储于所述分段页表中。

3.根据权利要求2所述的显存分配方法，其特征在于，所述分段页表可用于实现将所述虚拟GPU地址空间上的物理地址转换为物理GPU地址空间上的物理地址。

4.根据权利要求2所述的显存分配方法，其特征在于，所述分段页表可独立设置为打开/闭合模式。

5.根据权利要求2所述的显存分配方法，其特征在于，所述分段页表包括多个条目，每个所述条目对应一个为所述虚拟GPU分配的显存段。

6.根据权利要求2所述的显存分配方法，其特征在于，所述分段页表的存放地址存储于分段页表地址寄存器中，所述分段页表地址寄存器指向处于运行中的虚拟GPU所对应的分段页表。

7.根据权利要求1-6任一项所述的显存分配方法，其特征在于，所述预先对物理GPU所对应的显存进行划分，包括：

将所述物理GPU所对应的显存按存储容量划分成预定数目的具有相同存储容量的显存段。

8.根据权利要求7所述的显存分配方法，其特征在于，所述将所述物理GPU所对应的显存按存储容量划分成预定数目的具有相同存储容量的显存段，可通过显存段配置接口完成。

9.一种显存分配的装置，其特征在于，包括：

显存段分配单元，用于根据虚拟GPU所需显存的容量要求，为所述虚拟GPU分配至少两个显存段；其中，所述显存段是预先对物理GPU所对应的显存进行划分而获得的；所述为所述虚拟GPU分配至少两个显存段，包括：按所述虚拟GPU所需显存的容量要求和所述显存段的存储容量计算所述虚拟GPU所需显存段的数量；从所述物理GPU所对应的显存的剩余显存段中，按所述所需显存段的数量选取显存段；所述剩余显存段是指所述物理GPU所对应的显存空间中未分配给其它虚拟GPU的、或者其它虚拟GPU释放后所获得的可用于重新分配的显存段；

显存段物理地址保存单元，用于保存为所述虚拟GPU分配的至少两个显存段的物理地址。

10.一种电子设备，其特征在于，包括：

处理器；以及

存储器，用于存储一种显存分配的程序，该设备通电并通过所述处理器运行所述显存分配的程序后，执行下述步骤：

根据虚拟GPU所需显存的容量要求，为所述虚拟GPU分配至少两个显存段；其中，所述显存段是预先对物理GPU所对应的显存进行划分而获得的；所述为所述虚拟GPU分配至少两个显存段，包括：按所述虚拟GPU所需显存的容量要求和所述显存段的存储容量计算所述虚拟GPU所需显存段的数量；从所述物理GPU所对应的显存的剩余显存段中，按所述所需显存段的数量选取显存段；所述剩余显存段是指所述物理GPU所对应的显存空间中未分配给其它虚拟GPU的、或者其它虚拟GPU释放后所获得的可用于重新分配的显存段；

保存为所述虚拟GPU分配的至少两个显存段的物理地址。