未验证 提交 c92b1c54 编写于 作者: C chenxujun 提交者: GitHub

Update DCGMFileReader.py (#50652)

上级 8c758313
......@@ -176,7 +176,7 @@ class dcgmFileReader(FileReader):
gpuDcgmData = dcgm_data[dcgm_data['Entity'].isin([gpuId])]
traceEventList = []
for metric, parameteList in dcgmMetricParameterMap.items():
for metric, parameterList in dcgmMetricParameterMap.items():
metaInfo = {}
metaInfo['name'] = 'process_name'
metaInfo['ph'] = 'M'
......@@ -185,7 +185,7 @@ class dcgmFileReader(FileReader):
traceEventList.append(metaInfo)
for index, row in gpuDcgmData.iterrows():
for metric, parameteList in dcgmMetricParameterMap.items():
for metric, parameterList in dcgmMetricParameterMap.items():
trainerId = int(row['trainerId']) % self._groupSize
if trainerId >= self._displaySize:
continue
......@@ -203,7 +203,7 @@ class dcgmFileReader(FileReader):
di['id'] = trainerId
args = {}
for p in parameteList:
for p in parameterList:
args[p[0]] = row[p[1]]
di['args'] = args
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册