mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43898 Adding with_source parameter to enable tracking source code (filename and line) in profiler for eager, torchscript and autograd modes Test Plan: python test/test_profiler.py ``` Name Self CPU total % Self CPU total CPU total % CPU total CPU time avg Number of Calls Source Location ----------------------------------- --------------- --------------- --------------- --------------- --------------- --------------- -------------------------------------------- ts_method_1 10.43% 235.364us 36.46% 822.920us 822.920us 1 test/test_profiler.py(70): test_source aten::add 7.52% 169.833us 8.88% 200.439us 200.439us 1 test/test_profiler.py(69): test_source aten::normal_ 6.26% 141.380us 6.26% 141.380us 141.380us 1 test/test_profiler.py(67): test_source aten::add 5.80% 130.830us 8.41% 189.800us 63.267us 3 test/test_profiler.py(72): test_source aten::sum 5.02% 113.340us 8.39% 189.475us 189.475us 1 test/test_profiler.py(64): ts_method_1 aten::add 4.58% 103.346us 6.33% 142.847us 142.847us 1 test/test_profiler.py(62): ts_method_1 aten::mul 4.05% 91.498us 9.62% 217.113us 217.113us 1 test/test_profiler.py(71): test_source aten::add 4.03% 90.880us 5.60% 126.405us 126.405us 1 test/test_profiler.py(58): ts_method_2 aten::empty 3.49% 78.735us 3.49% 78.735us 19.684us 4 test/test_profiler.py(72): test_source ``` Reviewed By: ngimel Differential Revision: D23432664 Pulled By: ilia-cher fbshipit-source-id: 83ad7ebe0c2502494d3b48c4e687802db9c77615 |
||
|---|---|---|
| .. | ||
| backward_compatibility | ||
| bottleneck_test | ||
| cpp | ||
| cpp_api_parity | ||
| cpp_extensions | ||
| custom_backend | ||
| custom_operator | ||
| distributed | ||
| error_messages | ||
| expect | ||
| fx | ||
| jit | ||
| mobile | ||
| namespace_b | ||
| onnx | ||
| optim | ||
| package_a | ||
| quantization | ||
| scripts | ||
| test_img | ||
| type_hint_tests | ||
| HowToWriteTestsUsingFileCheck.md | ||
| module_a.py | ||
| print_test_stats.py | ||
| run_test.py | ||
| simulate_nccl_errors.py | ||
| test_autograd.py | ||
| test_bundled_images.py | ||
| test_bundled_inputs.py | ||
| test_complex.py | ||
| test_cpp_api_parity.py | ||
| test_cpp_extensions_aot.py | ||
| test_cpp_extensions_jit.py | ||
| test_cuda.py | ||
| test_cuda_primary_ctx.py | ||
| test_dataloader.py | ||
| test_determination.py | ||
| test_dispatch.py | ||
| test_distributions.py | ||
| test_expecttest.py | ||
| test_foreach.py | ||
| test_function_schema.py | ||
| test_functional_autograd_benchmark.py | ||
| test_futures.py | ||
| test_fx.py | ||
| test_indexing.py | ||
| test_jit.py | ||
| test_jit_cuda_fuser.py | ||
| test_jit_cuda_fuser_legacy.py | ||
| test_jit_cuda_fuser_profiling.py | ||
| test_jit_disabled.py | ||
| test_jit_fuser.py | ||
| test_jit_fuser_legacy.py | ||
| test_jit_fuser_te.py | ||
| test_jit_legacy.py | ||
| test_jit_profiling.py | ||
| test_jit_py3.py | ||
| test_jit_simple.py | ||
| test_jit_string.py | ||
| test_linalg.py | ||
| test_logging.py | ||
| test_mkldnn.py | ||
| test_mobile_optimizer.py | ||
| test_multiprocessing.py | ||
| test_multiprocessing_spawn.py | ||
| test_namedtensor.py | ||
| test_namedtuple_return_api.py | ||
| test_native_functions.py | ||
| test_nn.py | ||
| test_numba_integration.py | ||
| test_op_aliases.py | ||
| test_openmp.py | ||
| test_ops.py | ||
| test_optim.py | ||
| test_overrides.py | ||
| test_package.py | ||
| test_profiler.py | ||
| test_pruning_op.py | ||
| test_quantization.py | ||
| test_serialization.py | ||
| test_show_pickle.py | ||
| test_sparse.py | ||
| test_spectral_ops.py | ||
| test_static_runtime.py | ||
| test_tensor_creation_ops.py | ||
| test_tensorboard.py | ||
| test_tensorexpr.py | ||
| test_throughput_benchmark.py | ||
| test_torch.py | ||
| test_type_hints.py | ||
| test_type_info.py | ||
| test_type_promotion.py | ||
| test_unary_ufuncs.py | ||
| test_utils.py | ||
| test_vmap.py | ||
| test_vulkan.py | ||
| test_xnnpack_integration.py | ||