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)
|