in train/comms/pt/comms.py [0:0]
def printPreamble(self, commsParams):
logger.debug(f"\tcommsParams: {str(commsParams.__dict__)}")
header = "\n\tCOMMS-RES"
if self.collectiveArgs.collective == "pt2pt":
header += "{:>15}{:>20}{:>10}{:>10}{:>25}{:>10}{:>10}{:>15}{:>15}{:>18}{:>18}".format(
"size (B)",
"pingLatency(us):p50",
"p75",
"p95",
"pingPongLatency(us):p50",
"p75",
"p95",
"avgUniBW(GB/s)",
"avgBiBW(GB/s)",
"totalUniBW(GB/s)",
"totalBiBW(GB/s)",
)
else:
if commsParams.bitwidth < 32:
header += "-QUANT\t{:>15}{:>18}{:>25}{:>15}{:>15}{:>15}".format(
"size (B)",
"nElementsPerRank",
"P95 Latency(us): Quant",
"Comms",
"De-Quant",
"Overall",
)
elif not self.collectiveArgs.pair:
header += (
"{:>15}{:>18}{:>18}{:>12}{:>12}{:>12}{:>12}{:>15}{:>12}".format(
"size (B)",
"nElementsPerRank",
"Latency(us):p50",
"p75",
"p95",
"Min",
"Max",
"AlgBW(GB/s)",
"BusBW(GB/s)",
)
)
else:
header += "{:>15}{:>18}{:>22}{:>18}{:>12}{:>12}{:>12}{:>12}{:>15}{:>12}".format(
"total-size (B)",
"nElementsPerRank",
"nElementsPairPerRank",
"Latency(us):p50",
"p75",
"p95",
"Min",
"Max",
"AlgBW(GB/s)",
"BusBW(GB/s)",
)
print(header)