Skip to content

GitLab

Commit 30ac210e authored by xuanbaby's avatar xuanbaby
Browse files

DTK-203 

fix(bug for parse rocm-smi json)
parent 662457ba
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/nni_trial_tool/gpu.py
View file @ 30ac210e
...@@ -31,7 +31,7 @@ def parse_nvidia_smi_result(smi): ...@@ -31,7 +31,7 @@ def parse_nvidia_smi_result(smi):
output["gpuInfos"] = [] output["gpuInfos"] = []
for gpuIndex, gpu in enumerate(gpuList): for gpuIndex, gpu in enumerate(gpuList):
if gpu == 'system': if gpu == 'system':
break continue
gpuInfo = {} gpuInfo = {}
gpuInfo['index'] = gpuIndex gpuInfo['index'] = gpuIndex
gpuInfo['gpuUtil'] = smi[gpu]["GPU OverDrive value (%)"] gpuInfo['gpuUtil'] = smi[gpu]["GPU OverDrive value (%)"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment