update parser

ROCm · Feb 1, 2024 · 6476454 · 6476454
1 parent 10cb6b0
commit 6476454
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)