[發明專利]計算資源調度方法及裝置

在審

• 申請號: 201910654089.6
• 申請日: 2019-07-19
• 公開（公告）號: CN110362407A
• 公開（公告）日: 2019-10-22
• 發明（設計）人: 郭振宇;黃炳;姜璐;鐘華
• 申請（專利權）人: 中國工商銀行股份有限公司
• 主分類號: G06F9/50
• 分類號: G06F9/50
• 代理公司: 北京三友知識產權代理有限公司 11127
• 代理人: 王濤;任默聞

1.一種計算資源調度方法，其特征在于，所述方法包括：

根據GPU當前資源占用信息，確定與目標任務的資源需求對應的GPU當前可分配資源；

根據所述GPU當前可分配資源對所述目標任務進行GPU資源分配。

2.根據權利要求1所述的計算資源調度方法，其特征在于，所述根據GPU當前資源占用信息，確定與目標任務的資源需求對應的GPU當前可分配資源，包括：

根據所述GPU當前資源占用信息中的GPU當前剩余資源，確定GPU剩余可用資源；

判斷所述GPU剩余可用資源是否滿足所述目標任務的最大資源需求；

若是，則將所述GPU剩余可用資源設定為所述GPU當前可分配資源。

3.根據權利要求2所述的計算資源調度方法，其特征在于，在所述判斷所述GPU剩余可用資源是否滿足所述目標任務的最大資源需求之后，包括：

若否，則判斷所述GPU當前資源占用信息中是否存在比所述目標任務的任務優先級低的當前任務；

若不存在，則判斷所述GPU剩余可用資源是否滿足所述目標任務的最小資源需求；

若不滿足，則判定所述GPU當前可分配資源計算失敗，否則將所述GPU剩余可用資源設定為所述GPU當前可分配資源。

4.根據權利要求3所述的計算資源調度方法，其特征在于，在所述判斷所述GPU當前資源占用信息中是否存在當前任務優先級比所述目標任務的目標任務優先級低的當前任務之后，包括：

若存在，則根據所述目標任務的最大資源需求和所述比所述目標任務優先級低的當前任務的當前占用資源與最小資源需求，得到GPU任務搶奪資源，其中，所述GPU任務搶奪資源為在滿足所述當前任務的最小資源需求的前提下，當前任務的當前占用資源中的剩余資源；

將所述GPU任務搶奪資源和所述GPU剩余可用資源的總和設定為所述GPU當前可分配資源。

5.根據權利要求4所述的計算資源調度方法，其特征在于，所述根據所述目標任務的最大資源需求和所述當前任務優先級低于所述目標任務優先級的當前任務的當前占用資源與最小資源需求，得到GPU任務搶奪資源，包括：

根據任務優先級低于所述目標任務優先級的當前任務中一個當前任務的當前占用資源和最小資源需求，得到所述一個當前任務的可搶奪資源；

將所述一個當前任務的可搶奪資源增加至可搶奪資源總和中以更新所述可搶奪資源總和，判斷更新后的所述可搶奪資源總和以及所述GPU剩余可用資源的和是否滿足所述目標任務的最大資源需求；

若是，則將所述更新后的所述可搶奪資源總和設定為所述GPU任務搶奪資源；

若否，則重新獲取當前任務中下一個當前任務的當前占用資源和最小資源需求以更新所述可搶奪資源總和，直至更新后的所述可搶奪資源總和以及所述GPU剩余可用資源的和滿足所述目標任務的最大資源需求或者當前任務中所有當前任務已經獲取完畢。

6.根據權利要求5所述的計算資源調度方法，其特征在于，在所述重新獲取當前任務中下一個當前任務的當前占用資源和最小資源需求以更新所述可搶奪資源總和之后，還包括：

判斷更新后的所述可搶奪資源總和以及所述GPU剩余可用資源的和是否滿足所述目標任務的最小資源需求；

若不滿足，則判定所述GPU當前可分配資源計算失敗，否則將所述更新后的所述可搶奪資源總和設定為所述GPU任務搶奪資源。

7.根據權利要求4所述的計算資源調度方法，其特征在于，所述根據所述GPU當前可分配資源對所述目標任務進行GPU資源分配，包括：

將所述GPU當前可分配資源中的所述GPU剩余可用資源分配給所述目標任務；

判斷所述GPU當前可分配資源中是否包含所述GPU任務搶奪資源，若是，則從與所述GPU任務搶奪資源對應的所述當前任務中將該GPU任務搶奪資源分配給所述目標任務。