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

Update DCGMFileReader.py (#50652)

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