colossalai.utils.profiler.pcie_profiler

class colossalai.utils.profiler.pcie_profiler.PcieEvent(count=0, pcie_vol=0, cuda_time=0)

Pcie Event.

class colossalai.utils.profiler.pcie_profiler.PcieProfiler(dtype='fp32', depth=1)

Pcie profiler. Records all data transmission between CPU and GPU.

TODO: Merge pcie profiler into communication profiler