| /aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/ |
| H A D | _fsdp_param_group.py | 11 from torch.profiler import record_function
248 with record_function(self._with_fqn("FSDP::all_gather")):
272 with record_function(self._with_fqn("FSDP::all_gather_copy_out")):
314 with record_function(self._with_fqn("FSDP::pre_forward")):
324 with record_function(self._with_fqn("FSDP::post_forward")):
342 with record_function(self._with_fqn("FSDP::pre_backward")):
353 with record_function(self._with_fqn("FSDP::post_backward_accumulate")):
356 with record_function(self._with_fqn("FSDP::post_backward_reshard")):
382 with record_function(self._with_fqn("FSDP::post_backward_reduce")):
446 with record_function(
|
| H A D | _fsdp_state.py | 123 with torch.profiler.record_function("FSDP::root_pre_forward"):
134 with torch.profiler.record_function("FSDP::inputs_to_device"):
226 with torch.profiler.record_function("FSDP::cast_forward_inputs"):
259 with torch.profiler.record_function("FSDP::cast_forward_outputs"):
280 with torch.profiler.record_function("FSDP::root_post_backward_callback"):
|
| /aosp_15_r20/external/pytorch/test/profiler/ |
| H A D | test_execution_trace.py | 33 record_function,
52 with record_function("## TEST 1 ##", "1, 2, 3"):
147 with record_function(f"## LOOP {idx} ##"):
205 with record_function(f"## LOOP {idx} ##"):
262 with record_function(f"## LOOP {idx} ##"):
296 with record_function(f"## LOOP {idx} ##"):
326 with record_function(f"## LOOP {idx} ##"):
|
| H A D | test_record_function.py | 27 from torch.profiler import kineto_available, record_function
40 with record_function("## TEST 1 ##", "1, 2, 3"):
45 with record_function("## TEST 3 ##"):
|
| H A D | test_profiler_tree.py | 321 with torch.autograd.profiler.record_function("Top level Annotation"):
322 with torch.autograd.profiler.record_function("First Annotation"):
327 _ = torch.autograd.profiler.record_function(
332 with torch.autograd.profiler.record_function("Third Annotation"):
|
| H A D | test_profiler.py | 38 record_function,
574 with record_function("test_user_scope_alloc"):
576 with record_function("test_user_scope_dealloc"):
626 with record_function("test_user_scope_alloc"):
628 with record_function("test_user_scope_dealloc"):
1809 with torch.profiler.record_function("test_span"):
1933 with torch.profiler.record_function("test_user_annotation"):
1946 with torch.profiler.record_function("test_user_annotation"):
1958 with torch.profiler.record_function("test_user_annotation"):
1972 with torch.profiler.record_function("test_user_annotation"):
|
| H A D | test_cpp_thread.py | 78 with torch.autograd.profiler.record_function("user_function"):
|
| /aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
| H A D | _runtime_utils.py | 296 with torch.profiler.record_function(
368 with torch.profiler.record_function("FullyShardedDataParallel._pre_forward"):
427 with torch.profiler.record_function(
463 with torch.profiler.record_function("FullyShardedDataParallel._post_forward"):
516 with torch.profiler.record_function("FullyShardedDataParallel._root_pre_forward"):
588 with torch.profiler.record_function("FullyShardedDataParallel._to_kwargs"):
642 with torch.profiler.record_function("FullyShardedDataParallel._pre_backward_hook"):
679 with torch.profiler.record_function(
711 with torch.autograd.profiler.record_function(
771 with torch.profiler.record_function(
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/ |
| H A D | README.md | 26 [/aten/src/ATen/record_function.h](/aten/src/ATen/record_function.h)
34 There is also a python binding for `RecordFunction` in python (`with torch.profiler.record_function…
42 …/pytorch/blob/2e3fce54506ba82eee2c890410bf7a1405a64ec6/aten/src/ATen/record_function.h#L357): Auto…
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | distributed.py | 11 from torch.profiler import profile, ProfilerActivity, record_function
38 with record_function("Forward"):
41 with record_function("Backward"):
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/jit/ |
| H A D | rpc_test.py | 11 from torch.autograd.profiler import record_function
149 with record_function(block):
156 with record_function(block) as rf:
1150 with torch.autograd.profiler.record_function(prof_key) as rf:
1283 with torch.autograd.profiler.record_function("foo") as rf:
1299 with torch.autograd.profiler.record_function("foo") as rf:
|
| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | bench.py | 10 from torch.autograd.profiler import record_function
98 with record_function("## forward ##"):
|
| /aosp_15_r20/external/pytorch/torch/profiler/ |
| H A D | profiler.py | 676 self.step_rec_fn: Optional[prof.record_function] = None
756 self.step_rec_fn = prof.record_function(
780 self.step_rec_fn = prof.record_function(
|
| H A D | __init__.py | 15 from torch.autograd.profiler import KinetoStepTracker, record_function
|
| /aosp_15_r20/external/pytorch/torch/distributed/_symmetric_memory/ |
| H A D | __init__.py | 384 with torch.profiler.record_function("fused_all_gather_matmul"):
495 with torch.profiler.record_function("fused_all_gather_scaled_matmul"):
636 with torch.profiler.record_function("fused_matmul_reduce_scatter"):
712 with torch.profiler.record_function("fused_matmul_reduce_scatter"):
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/interpreter/ |
| H A D | frame.h | 32 std::unique_ptr<at::RecordFunction> record_function; member
|
| /aosp_15_r20/external/pytorch/benchmarks/profiler_benchmark/ |
| H A D | resnet_memory_profiler.py | 20 with profiler.record_function("root"):
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_with.py | 622 with torch.autograd.profiler.record_function("foo"):
624 with torch.autograd.profiler.record_function("nested"):
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | torch.py | 78 torch.autograd.profiler.record_function,
276 torch.profiler.record_function,
278 torch.autograd.profiler.record_function,
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | profiler.py | 152 with torch.profiler.record_function("TORCHDYNAMO"):
|
| /aosp_15_r20/external/pytorch/torch/distributed/pipelining/ |
| H A D | schedules.py | 27 from torch.profiler import record_function
652 with record_function(f"Forward {i}"):
700 with record_function(f"Forward {i}"):
730 with record_function(f"Backward {i}"):
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_padding.py | 148 with torch.profiler.record_function(tag_lhs):
151 with torch.profiler.record_function(tag_rhs):
|
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | optimizer.py | 472 with torch.autograd.profiler.record_function(profile_name):
947 with torch.autograd.profiler.record_function(self._zero_grad_profile_name):
|
| /aosp_15_r20/external/pytorch/torch/utils/data/datapipes/ |
| H A D | _hook_iterator.py | 134 return torch.autograd.profiler.record_function(datapipe._profile_name)
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | torch.compiler_profiling_torch_compile.rst | 112 with torch.profiler.record_function("warmup compile"):
115 with torch.profiler.record_function("resnet18 compile"):
|