Home
last modified time | relevance | path

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 Ddistributed_test.py198 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)