Merge pull request #9 from ROCm/cl/update-parser

Update the parser to use the latest log.
ROCm · Feb 1, 2024 · e632f8d · e632f8d
2 parents 10cb6b0 + 6476454
commit e632f8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rccl_nccl_parser.py b/rccl_nccl_parser.py
@@ -87,7 +87,7 @@ def parse_nccl_log(nccl_lines):
 
         total_bytes = int(count) * data_type_bytes_map[datatype]
 
-        test_cmd = "./build/" + coll_op_map[comm] + " -d " + data_types_map[datatype] + \
+        test_cmd = "./build/" + coll_op_map[comm.replace("mscclFunc", "")] + " -d " + data_types_map[datatype] + \
                        " -b " + str(total_bytes) + " -e " + str(total_bytes) + \
                        " -o " + reduction_op_map[op_type] + " -g " + str(nnranks)
         #print (test_cmd)