[發明專利]計算任務的分配方法、裝置、計算機設備和存儲介質

在審

• 申請號: 202210426012.5
• 申請日: 2022-04-22
• 公開（公告）號: CN114706687A
• 公開（公告）日: 2022-07-05
• 發明（設計）人: 李肯立;翁萬東;肖正;唐卓;肖國慶;段明星;周旭;廖清
• 申請（專利權）人: 湖南大學
• 主分類號: G06F9/50
• 分類號: G06F9/50;G06F16/901
• 代理公司: 華進聯合專利商標代理有限公司 44224
• 代理人: 張思佳

1.一種計算任務的分配方法，其特征在于，所述方法包括：

獲取所述計算任務對應的第一數量個計算圖以及所述計算任務中各計算子任務之間的流轉關系，每個所述計算圖包括至少一計算子任務；

計算每個邊界計算子任務的關鍵關系增益，所述關鍵關系增益包括第一增益和第二增益，其中，所述第一增益根據所述邊界計算子任務所屬計算圖中與所述邊界計算子任務存在的關鍵關系獲得，所述第二增益根據所述邊界計算子任務分別與其他計算圖之間的關鍵關系獲得；

根據各邊界計算子任務的所述第一增益和第二增益調整所述邊界計算子任務至其他計算圖，獲得新的第一數量個計算圖；

若新的第一數量個計算圖存在新的邊界計算子任務，返回確定每個邊界計算子任務的關鍵關系增益的步驟，直至獲得的新的第一數量個計算圖不存在新的邊界計算子任務，并將每次獲得第一數量個計算圖之間的關鍵關系數量最小的第一數量個計算圖，分配至對應的第一數量個計算設備。

2.根據權利要求1所述的方法，其特征在于，所述計算每個邊界計算子任務的關鍵關系增益，所述關鍵關系增益包括第一增益和第二增益，包括：

統計所述邊界計算子任務所屬計算圖中與所述邊界計算子任務存在關鍵關系的計算子任務的數量，獲得所述第一增益；并統計所述邊界計算子任務分別與其他計算圖之間存在關鍵關系的計算子任務的數量，獲得所述邊界計算子任務分別相對其他計算圖的第二增益。

3.根據權利要求1所述的方法，其特征在于，所述計算每個邊界計算子任務的關鍵關系增益，所述關鍵關系增益包括第一增益和第二增益，包括：

將所述邊界計算子任務的流轉關系中關鍵關系賦值為第一值；

根據所述邊界計算子任務所屬計算圖中與所述邊界計算子任務存在的關鍵關系，以及所述第一值進行計算，獲得所述第一增益；并根據所述邊界計算子任務分別與其他計算圖之間的關鍵關系，以及所述第一值進行計算，獲得所述第二增益。

4.根據權利要求1至3任一項所述的方法，其特征在于，所述根據各邊界計算子任務的所述第一增益和第二增益調整所述邊界計算子任務至其他計算圖，獲得新的第一數量個計算圖，包括：

計算每個所述邊界計算子任務的所述第二增益和所述第一增益的差值；

將所述差值大于預設閾值的邊界計算子任務分配至所述第二增益對應的計算圖，獲得重新劃分后的第一數量個計算圖。

5.根據權利要求4所述的方法，其特征在于，所述將所述差值大于預設閾值的邊界計算子任務分配至所述第二增益對應的計算圖，獲得重新劃分后的第一預設數量個計算圖，包括：

根據所述流轉關系確定所述差值大于預設閾值的邊界計算子任務中的目標邊界計算子任務；

將所述目標邊界計算子任務分配至所述第二增益對應的計算圖，獲得重新劃分后的第一預設數量個計算圖。

6.根據權利要求1所述的方法，其特征在于，所述獲取所述計算任務對應的第一預設數量個計算圖以及所述計算任務中各計算子任務之間的流轉關系，包括：

獲取所述計算任務對應的有向圖，以及執行所述計算任務的計算設備的數量，所述有向圖包括第二數量個計算子任務，以及各計算子任務之間的流轉關系，所述流轉關系包括關鍵關系和非關鍵關系，所述第二數量大于所述第一數量；

按照預設規則將所述有向圖劃分為第一數量個計算圖。

7.根據權利要求6所述的方法，其特征在于，所述按照預設規則將所述有向圖劃分為第一數量個計算圖，包括：

根據所述有向圖中各計算子任務的流轉關系進行聚類處理，獲得多個集合；

將所述多個集合劃分為第一數量個計算圖。

8.根據權利要求7所述的方法，其特征在于，所述將所述多個集合劃分為第一數量個計算圖，包括：

將所述多個集合隨機劃分為第一數量個分區；

隨機從任一分區中選擇任一集合劃分到其他分區，并計算每次隨機從任一分區中選擇任一集合劃分到其他分區后，獲得所有新的分區之間的切邊數；

將切邊數最小對應的各分區作為所述計算圖。