Skip to content

Commit 95d38d0

Browse files
committed
More robust results parsing for NCCL
1 parent 65b8d8d commit 95d38d0

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/cloudai/workloads/nccl_test/performance_report_generation_strategy.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,9 @@ def _parse_data_rows(file: TextIOWrapper) -> List[List[str]]:
3939
for line in file:
4040
line: str = line.strip()
4141
if re.match(r"^\d", line):
42-
parsed_data_rows.append(re.split(r"\s+", line))
42+
parts = re.split(r"\s+", line)
43+
if len(parts) == 13:
44+
parsed_data_rows.append(parts)
4345
return parsed_data_rows
4446

4547

0 commit comments

Comments
 (0)