[發明專利]一種容器的GPU進程統計方法和裝置

有效

• 申請號: 202110943920.7
• 申請日: 2021-08-17
• 公開（公告）號: CN113742171B
• 公開（公告）日: 2023-07-14
• 發明（設計）人: 劉慧興
• 申請（專利權）人: 蘇州浪潮智能科技有限公司
• 主分類號: G06F11/30
• 分類號: G06F11/30;G06F11/34
• 代理公司: 北京連和連知識產權代理有限公司 11278
• 代理人: 陳黎明;張元

1.一種容器的GPU進程統計方法，其特征在于，包括執行以下步驟：

重定義原生動態庫的原生動態鏈接符號函數和GPU狀態顯示命令中的原生進程統計函數，并基于重定義的進程統計函數構建劫持函數表；

運行包括多個所述原生進程統計函數的函數符號的所述GPU狀態顯示命令；

使用打開函數調用所述原生動態庫以獲得操作句柄，并使用所述原生動態鏈接符號函數基于所述操作句柄和多個所述函數符號分別獲得多個原生函數地址；

使用重定義的動態鏈接符號函數分別判斷多個所述函數符號是否記載于所述劫持函數表內，并將記載于所述劫持函數表內的多個所述函數符號重定向到所述劫持函數表中對應的劫持函數地址以獲取GPU進程標識和進程名稱；

將所述GPU進程標識替換為對應的容器進程標識以顯示容器的GPU進程狀態；

其中，將所述GPU進程標識替換為對應的容器進程標識以顯示容器的GPU進程狀態包括：使用讀取進程表獲取每個所述容器進程標識，利用匹配規則基于每個所述容器進程標識來搜索并分別對應每個所述容器進程標識的調度文件；從每個所述調度文件中分別讀取每個所述容器進程標識對應的GPU進程標識和進程名稱，并響應于從每個所述調度文件中分別讀取的所述GPU進程標識和所述進程名稱、與根據多個所述函數符號分別獲取的所述GPU進程標識和所述進程名稱相匹配，而將根據多個所述函數符號分別獲取的所述GPU進程標識替換為對應的所述容器進程標識；輸出所述進程名稱、對應的所述容器進程標識、以及容器的GPU進程狀態。

2.根據權利要求1所述的方法，其特征在于，基于重定義的進程統計函數構建劫持函數表包括：向所述劫持函數表中寫入所述重定義的進程統計函數的函數符號、劫持函數地址、以及指針類型。

3.根據權利要求1所述的方法，其特征在于，在將記載于所述劫持函數表內的多個所述函數符號重定向到所述劫持函數表中對應的劫持函數地址以獲取GPU進程標識和進程名稱的同時，還使用未記載于所述劫持函數表內的多個所述函數符號各自的所述原生函數地址來獲取GPU進程標識和進程名稱。

4.根據權利要求1所述的方法，其特征在于，在使用讀取進程表獲取每個所述容器進程標識之后，先剔除所述容器進程標識0或1的進程以及GPU狀態顯示命令其本身所使用的進程，再利用匹配規則基于其余的所述容器進程標識來搜索并分別對應每個所述容器進程標識的調度文件。

5.一種容器的GPU進程統計裝置，其特征在于，包括：

處理器；

控制器，存儲有所述處理器可運行的程序代碼，所述處理器在運行所述程序代碼時執行以下步驟：

重定義原生動態庫的原生動態鏈接符號函數和GPU狀態顯示命令中的原生進程統計函數，并基于重定義的進程統計函數構建劫持函數表；

運行包括多個所述原生進程統計函數的函數符號的所述GPU狀態顯示命令；

使用打開函數調用所述原生動態庫以獲得操作句柄，并使用所述原生動態鏈接符號函數基于所述操作句柄和多個所述函數符號分別獲得多個原生函數地址；

使用重定義的動態鏈接符號函數分別判斷多個所述函數符號是否記載于所述劫持函數表內，并將記載于所述劫持函數表內的多個所述函數符號重定向到所述劫持函數表中對應的劫持函數地址以獲取GPU進程標識和進程名稱；

將所述GPU進程標識替換為對應的容器進程標識以顯示容器的GPU進程狀態；

其中，將所述GPU進程標識替換為對應的容器進程標識以顯示容器的GPU進程狀態包括：

使用讀取進程表獲取每個所述容器進程標識，利用匹配規則基于每個所述容器進程標識來搜索并分別對應每個所述容器進程標識的調度文件；

從每個所述調度文件中分別讀取每個所述容器進程標識對應的GPU進程標識和進程名稱，并響應于從每個所述調度文件中分別讀取的所述GPU進程標識和所述進程名稱、與根據多個所述函數符號分別獲取的所述GPU進程標識和所述進程名稱相匹配，而將根據多個所述函數符號分別獲取的所述GPU進程標識替換為對應的所述容器進程標識；

輸出所述進程名稱、對應的所述容器進程標識、以及容器的GPU進程狀態。