Searched refs:dedup_gpu_user_annotation (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 198 def get_profiling_event(event_name, profiler, dedup_gpu_user_annotation=False): argument
208 and (not dedup_gpu_user_annotation or event.device_type != DeviceType.CUDA)
1631 … events = get_profiling_event(event_name, prof, dedup_gpu_user_annotation=True)
6914 … events = get_profiling_event(all_reduce_event_name, prof, dedup_gpu_user_annotation=True)
6922 … broadcast_events = get_profiling_event(broadcast_event_name, prof, dedup_gpu_user_annotation=True)
6945 … events = get_profiling_event(all_reduce_event_name, prof, dedup_gpu_user_annotation=True)
|