From fe4285d0028338cf0e57697c78d5cab7325b8e23 Mon Sep 17 00:00:00 2001
From: gilbertlee-amd <44450918+gilbertlee-amd@users.noreply.github.com>
Date: Wed, 13 Oct 2021 14:56:57 -0600
Subject: [PATCH] [TransferBench] Adding comment echoing to help distinguish
 tests (#438)

[ROCm/rccl commit: f6b7ac693e0e7f1405b093c83dabbe9c316dd8c1]
---
 projects/rccl/tools/TransferBench/TransferBench.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/projects/rccl/tools/TransferBench/TransferBench.cpp b/projects/rccl/tools/TransferBench/TransferBench.cpp
index 4ab2aad2dd..ba7bfae5dc 100644
--- a/projects/rccl/tools/TransferBench/TransferBench.cpp
+++ b/projects/rccl/tools/TransferBench/TransferBench.cpp
@@ -114,6 +114,10 @@ int main(int argc, char **argv)
   char line[2048];
   while(fgets(line, 2048, fp))
   {
+    // Check if line is a comment
+    if (!ev.outputToCsv && line[0] == '#' && line[1] == '#')
+      printf("%s", line);
+
     // Parse links from configuration file
     std::vector<Link> links;
     ParseLinks(line, numCpuDevices, numGpuDevices, links);
@@ -424,6 +428,7 @@ void DisplayUsage(char const* cmdName)
   printf("-2 (G0 G0 G1 4) (G1 G1 G0 2) Runs 2 Links in parallel.  GPU 0 - > GPU 1 using four CUs, and GPU1 -> GPU 0 using two CUs\n");
   printf("\n");
   printf("Round brackets and arrows' ->' may be included for human clarity, but will be ignored and are unnecessary\n");
+  printf("Lines starting with # will be ignored. Lines starting with ## will be echoed to output\n");
   printf("\n");
 
   EnvVars::DisplayUsage();