/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"):
|