Home
last modified time | relevance | path

Searched refs:record_function (Results 1 – 25 of 44) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A D_fsdp_param_group.py11 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.py123 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 Dtest_execution_trace.py33 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 Dtest_record_function.py27 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 Dtest_profiler_tree.py321 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 Dtest_profiler.py38 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 Dtest_cpp_thread.py78 with torch.autograd.profiler.record_function("user_function"):
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_runtime_utils.py296 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 DREADME.md26 [/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 Ddistributed.py11 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 Drpc_test.py11 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 Dbench.py10 from torch.autograd.profiler import record_function
98 with record_function("## forward ##"):
/aosp_15_r20/external/pytorch/torch/profiler/
H A Dprofiler.py676 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__.py15 from torch.autograd.profiler import KinetoStepTracker, record_function
/aosp_15_r20/external/pytorch/torch/distributed/_symmetric_memory/
H A D__init__.py384 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 Dframe.h32 std::unique_ptr<at::RecordFunction> record_function; member
/aosp_15_r20/external/pytorch/benchmarks/profiler_benchmark/
H A Dresnet_memory_profiler.py20 with profiler.record_function("root"):
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_with.py622 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 Dtorch.py78 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 Dprofiler.py152 with torch.profiler.record_function("TORCHDYNAMO"):
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A Dschedules.py27 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 Dtest_padding.py148 with torch.profiler.record_function(tag_lhs):
151 with torch.profiler.record_function(tag_rhs):
/aosp_15_r20/external/pytorch/torch/optim/
H A Doptimizer.py472 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.py134 return torch.autograd.profiler.record_function(datapipe._profile_name)
/aosp_15_r20/external/pytorch/docs/source/
H A Dtorch.compiler_profiling_torch_compile.rst112 with torch.profiler.record_function("warmup compile"):
115 with torch.profiler.record_function("resnet18 compile"):

12