[發(fā)明專利]一種圖形處理器GPU卡的自動化測試方法和裝置

在審

• 申請?zhí)枺?/td>: 202211214619.3
• 申請日: 2022-09-30
• 公開（公告）號: CN115525498A
• 公開（公告）日: 2022-12-27
• 發(fā)明（設計）人: 谷海洋
• 申請（專利權）人: 蘇州浪潮智能科技有限公司
• 主分類號: G06F11/26
• 分類號: G06F11/26;G06F11/22
• 代理公司: 北京潤澤恒知識產(chǎn)權代理有限公司 11319
• 代理人: 蘇培華

本發(fā)明實施例提供了一種圖形處理器GPU卡的自動化測試方法和裝置，應用于測試服務器，多張GPU卡通過外部設備互連標準PCI總線與所述測試服務器連接，通過獲取所述多張GPU卡的外部設備互連總線接口PCIE地址；從所述多張GPU卡中確定當前測試的目標GPU卡；保留所述目標GPU卡的PCIE地址，并卸載除所述目標GPU卡之外的GPU卡的PCIE地址，以使所述除所述目標GPU卡之外的GPU卡掉電；對所述目標GPU卡執(zhí)行計算性能測試，從而通過控制GPU卡卸載來模擬GPU卡物理移除的測試環(huán)境以滿足測試用例要求，每次對單卡進行測試時都無需手動更換GPU位置，避免了每次更換GPU位置時都需要進行開關機的問題，從而縮減了人力時間及成本，優(yōu)化了GPU卡的測試步驟，提高了測試效率以及測試可靠性。

技術領域

本發(fā)明涉及測試技術領域，特別是涉及一種圖形處理器GPU卡的自動化測試方法和一種圖形處理器GPU卡的自動化測試裝置。

背景技術

隨著信息技術的發(fā)展，AI(Artificial Intelligence，人工智能)技術應用越來越廣泛，對服務器AI計算性能的要求也越來越高。

服務器中通?？梢耘渲肎PU(Graphics Processing Unit，圖形處理器)，在前期研發(fā)階段，通過對GPU的計算性能進行全面測試，有利于提高服務器的性能和穩(wěn)定性，增強產(chǎn)品競爭力。

在對GPU的計算性能進行測試時，由于需要測試GPU在理想狀態(tài)下的計算性能，因此通常需要對單卡執(zhí)行測試。然而，在對GPU單卡執(zhí)行測試時，需要手動更換GPU位置，每次更換GPU位置時都需要進行開關機，導致占用大量人力物力，時間成本較高。

發(fā)明內(nèi)容

鑒于上述問題，提出了本發(fā)明實施例以便提供一種克服上述問題或者至少部分地解決上述問題的一種圖形處理器GPU卡的自動化測試方法和相應的一種圖形處理器GPU卡的自動化測試裝置。

為了解決上述問題，本發(fā)明實施例公開了一種圖形處理器GPU卡的自動化測試方法，應用于測試服務器，多張GPU卡通過外部設備互連標準PCI總線與所述測試服務器連接，所述方法包括：

獲取所述多張GPU卡的外部設備互連總線接口PCIE地址；

從所述多張GPU卡中確定當前測試的目標GPU卡；

保留所述目標GPU卡的PCIE地址，并卸載除所述目標GPU卡之外的GPU卡的PCIE地址，以使所述除所述目標GPU卡之外的GPU卡掉電；

對所述目標GPU卡執(zhí)行計算性能測試。

可選地，所述從所述多張GPU卡中確定當前測試的目標GPU卡，包括：

確定所述多張GPU卡的PCIE地址的獲取順序；

按照所述獲取順序，從所述多張GPU卡中確定當前測試的目標GPU卡。

可選地，在所述獲取所述多張GPU卡的PCIE地址的步驟之后，所述方法還包括：

將所述多張GPU卡的PCIE地址寫入系統(tǒng)目錄文件；

所述保留所述目標GPU卡的PCIE地址，并卸載除所述目標GPU卡之外的GPU卡的PCIE地址，包括：

查看所述系統(tǒng)目錄文件中多張GPU卡的PCIE地址；

保留所述目標GPU卡的PCIE地址，并按照所述獲取順序依次卸載除所述目標GPU卡之外的GPU卡的PCIE地址。

可選地，所述方法還包括：

重新掃描所述多張GPU卡的PCIE地址；

判斷是否已依次遍歷測試各張GPU卡；