diff --git a/python/paddle/profiler/profiler_statistic.py b/python/paddle/profiler/profiler_statistic.py index 69fd3c15130e3426182b0f7c1dd6e3d53bb1548b..df1cfe49cd9d2b5fe23915bb369dc1a689074ec1 100755 --- a/python/paddle/profiler/profiler_statistic.py +++ b/python/paddle/profiler/profiler_statistic.py @@ -472,7 +472,11 @@ class DistributedSummary: for runtimenode in hostnode.runtime_node: for devicenode in runtimenode.device_node: if devicenode.type == TracerEventType.Kernel: - if 'nccl' in devicenode.name.lower(): + kernel_name = devicenode.name.lower() + if ( + 'nccl' in kernel_name + or 'xccl' in kernel_name + ): self.gpu_communication_range.append( (devicenode.start_ns, devicenode.end_ns) )