[TransferBench] Restore some previous fixes - memory leak, PCIe address (#314)

[ROCm/rccl commit: 62e0447e9a]
2021-02-01 09:48:09 -07:00
parent 7f9c15b843
commit 60c74f63fa
1 changed files with 6 additions and 3 deletions
@@ -540,11 +540,12 @@ void DisplayTopology()
  printf("        |");
  for (int j = 0; j < numGpuDevices; j++)
    printf(" GPU %02d |", j);
-  printf("\n");
+  printf(" PCIe Bus ID\n");
  for (int j = 0; j <= numGpuDevices; j++)
    printf("--------+");
-  printf("\n");
+  printf("-------------\n");

+  char pciBusId[20];
  for (int i = 0; i < numGpuDevices; i++)
  {
    printf(" GPU %02d |", i);
@@ -565,7 +566,8 @@ void DisplayTopology()
               hopCount);
      }
    }
-    printf("\n");
+    HIP_CALL(hipDeviceGetPCIBusId(pciBusId, 20, i));
+    printf(" %s\n", pciBusId);
  }
 }

@@ -865,6 +867,7 @@ void CheckOrFill(ModeType mode, int N, bool isMemset, bool isHipCall, float* ptr
        exit(1);
      }
    }
+    free(hostBuffer);
  }

  free(refBuffer);