Mike Guo d4f8313497 Add low level torch.profiler.kineto_profile base class (#63302) ... Summary: Refactor torch.profiler.profile by separate it into one low level class and one high level wrapper. The PR include the following change: 1. separate class torch.profiler.profile into two separated class: kineto_profiler and torch.profiler.profile. 2. The former class has the low-level functionality exposed in C++ level like: prepare_profiler, start_profiler, stop_profiler. 3. The original logics in torch.profiler.profile including export_chrome_trace, export_stacks, key_averages, events, add_metadata are all moved into kineto_profiler since they are all exposed by the torch.autograd.profiler. 4. The new torch.profiler.profile is fully back-compatible with original class since it inherit from torch.profiler.kineto_profiler. Its only responsibility in new implementation is the maintenance of the finite state machine of ProfilerAction. With the refactoring, the responsibility boundary is clear and the new logic is simple to understand. Pull Request resolved: https://github.com/pytorch/pytorch/pull/63302 Reviewed By: albanD Differential Revision: D33006442 Pulled By: robieta fbshipit-source-id: 30d7c9f5c101638703f1243fb2fcc6ced47fb690		2021-12-14 14:47:43 -08:00
..
_static	clarify the documentation of torch.meshgrid (#62977)	2021-08-18 04:01:22 -07:00
_templates	Remove master documentation from being indexable by search engines (#58056)	2021-05-18 06:20:09 -07:00
community	Update contribution_guide.rst (#64142)	2021-08-30 19:26:59 -07:00
elastic	(torchelastic) make --max_restarts explicit in the quickstart and runner docs (#65838)	2021-09-29 19:29:01 -07:00
notes	Remove remaining THC code (#69039)	2021-12-08 12:18:08 -08:00
rpc	Support Union in TorchScript (#64234)	2021-09-03 06:12:24 -07:00
scripts	[docs] Add images to some activation functions (#65415)	2021-09-22 11:05:29 -07:00
__config__.rst
amp.rst	rebase for autocast updates to include device_type and dtype flags (#61002)	2021-08-10 20:03:12 -07:00
autograd.rst	Update extending doc to cover forward mode AD (#66962)	2021-10-27 14:18:38 -07:00
backends.rst	[Linalg] Add a runtime switch to let pytorch prefer a backend impl in linalg functions on GPU (#67980)	2021-12-03 19:06:30 -08:00
benchmark_utils.rst
bottleneck.rst
checkpoint.rst
complex_numbers.rst	Grammatical update of tech docs (#61547)	2021-07-14 14:01:59 -07:00
conf.py	[Quant] Add dynamic QAT Linear module (#67325)	2021-11-08 10:24:25 -08:00
cpp_extension.rst
cpp_index.rst
cuda.rst	Fixes mem_get_info when querying on a device other than the current device (#69640)	2021-12-08 23:04:30 -08:00
cudnn_persistent_rnn.rst	Remove orphan from cuDNN persistent note (#65160)	2021-09-21 11:09:47 -07:00
cudnn_rnn_determinism.rst
data.rst	[DateLoader] more clearly expose 'default_collate' and 'default_convert' to users (#69862)	2021-12-14 11:18:26 -08:00
ddp_comm_hooks.rst	[DDP Comm Hook] Add debugging communication hooks to ddp_comm_hooks.rst (#64352)	2021-09-01 17:37:19 -07:00
deploy.rst	[deploy] docs (#69251)	2021-12-01 21:55:18 -08:00
distributed.algorithms.join.rst	Add tutorial link (#62785)	2021-08-05 17:28:02 -07:00
distributed.elastic.rst	[1/n][torch/elastic] Move torchelastic docs *.rst (#148)	2021-05-04 00:57:56 -07:00
distributed.optim.rst	[distributed][docs] Delete distributed optimimzer section from RPC and add reference to namespace docs page (#68068)	2021-11-09 15:01:54 -08:00
distributed.rst	Remove autograd-enabled collective APIs from distributed docs (#69011)	2021-11-29 18:14:50 -08:00
distributions.rst
dlpack.rst
docutils.conf
fft.rst	C++ API and docs for hfftn (#66127)	2021-10-07 12:48:36 -07:00
futures.rst	Update docs to mention CUDA support for Future (#50048)	2021-05-11 08:26:33 -07:00
fx.rst	[FX][docs] Document gotcha about training flag (#68915)	2021-11-29 16:13:32 -08:00
hub.rst
index.rst	[deploy] docs (#69251)	2021-12-01 21:55:18 -08:00
jit_builtin_functions.rst
jit_language_reference_v2.rst	Add Union type to TorchScript Language Ref (#69514)	2021-12-07 12:53:54 -08:00
jit_language_reference.rst	fix typos in jit_language_reference.rst (#68706)	2021-11-22 19:09:06 -08:00
jit_python_reference.rst	[JIT] improve documentation (#57991)	2021-05-19 11:47:32 -07:00
jit_unsupported.rst
jit.rst	Back out "D30740897 Add fusion enabled apis" (#64500)	2021-09-04 20:55:58 -07:00
linalg.rst	Revert D32521980: Add linalg.lu_factor	2021-11-28 17:22:15 -08:00
math-quantizer-equation.png
mobile_optimizer.rst
model_zoo.rst
multiprocessing.rst
name_inference.rst	Abladawood patch 1 (#58496)	2021-05-20 10:32:18 -07:00
named_tensor.rst
nn.functional.rst	Add mish activation function (#58648)	2021-05-25 10:36:21 -07:00
nn.init.rst
nn.rst	Implements the orthogonal parametrization (#62089)	2021-08-30 13:12:07 -07:00
onnx.rst	[Doc] [ONNX]Fix a broken url for ONNXRuntime custom op (#67944)	2021-11-08 15:51:02 -08:00
optim.rst	To add SequentialLR to PyTorch Core Schedulers (#64037)	2021-09-09 09:36:32 -07:00
package.rst	Minor changes in documentation (#68557)	2021-11-18 17:57:16 -08:00
pipeline.rst	Minor changes in documentation (#68557)	2021-11-18 17:57:16 -08:00
profiler.rst	Add low level torch.profiler.kineto_profile base class (#63302)	2021-12-14 14:47:43 -08:00
quantization-support.rst	Updating quantization documentation (#68907)	2021-12-08 08:32:33 -08:00
quantization.rst	[quant][docs] quantized model save/load instructions (#69789)	2021-12-13 20:23:59 -08:00
random.rst
rpc.rst	[distributed][docs] Delete distributed optimimzer section from RPC and add reference to namespace docs page (#68068)	2021-11-09 15:01:54 -08:00
sparse.rst	Sparse CSR: add convert_indices_from_csr_to_coo (#66774)	2021-11-17 22:28:30 -08:00
special.rst	[special] special alias for softmax (#62251)	2021-10-01 03:55:32 -07:00
storage.rst
tensor_attributes.rst
tensor_view.rst	Minor changes in documentation (#68557)	2021-11-18 17:57:16 -08:00
tensorboard.rst
tensors.rst	ammend tensors.rst and torch.rst for doc generation (#69030)	2021-11-30 12:04:13 -08:00
testing.rst	[Doc] make_tensor to torch.testing module (#63925)	2021-08-30 12:25:40 -07:00
torch.ao.ns._numeric_suite_fx.rst	Quantization docs: add pages for Numeric Suite (Eager and FX) (#66380)	2021-10-11 18:47:58 -07:00
torch.ao.ns._numeric_suite.rst	Quantization docs: add pages for Numeric Suite (Eager and FX) (#66380)	2021-10-11 18:47:58 -07:00
torch.overrides.rst
torch.rst	Porting index_add to structured kernels, add an out variant (#65993)	2021-12-14 11:57:13 -08:00
type_info.rst	[Docs] Mention torch.bfloat16 in torch.finfo (#68496)	2021-11-18 17:52:41 -08:00