DTK-203

fix(bug for parse rocm-smi json)

DTK-203
fix(bug for parse rocm-smi json)
30ac210e · xuanbaby · 662457ba · 30ac210e
Commit 30ac210e authored Sep 24, 2020 by xuanbaby
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/nni_trial_tool/gpu.py tools/nni_trial_tool/gpu.py +1 -1

No files found.
--- a/tools/nni_trial_tool/gpu.py
+++ b/tools/nni_trial_tool/gpu.py
@@ -31,7 +31,7 @@ def parse_nvidia_smi_result(smi):
        output["gpuInfos"] = []
        for gpuIndex, gpu in enumerate(gpuList):
            if gpu == 'system':  
-              break
+              continue
            gpuInfo = {}
            gpuInfo['index'] = gpuIndex
            gpuInfo['gpuUtil'] = smi[gpu]["GPU OverDrive value (%)"]