Ke Wen 3090667cf9 [pipelining] pipeline() taking microbatch as example input (#128163) ... Changed the API of `pipeline()` to take microbatch instead of full batch as example args. Main purpose is to: - make this API more atomic; - decouple tracing frontend from runtime info like `num_chunks`. Side effects: - Creates opportunity for varying `num_chunks` of schedules with the same `pipe` object. - User has to create example microbatch input. - Chunk spec stuff are now all moved to runtime side. Pull Request resolved: https://github.com/pytorch/pytorch/pull/128163 Approved by: https://github.com/H-Huang		2024-06-07 15:51:53 +00:00
..
_static	[docs] Update PT2+Profiler docs (#122272)	2024-03-28 17:52:28 +00:00
_templates
community	Fix broken link of scikit-learn (#120972)	2024-05-16 11:46:34 +00:00
elastic	Reapply "distributed debug handlers (#126601)" (#127805)	2024-06-04 19:44:30 +00:00
notes	Fixed typos in the complex numbers portion of the autograd docs (#127948)	2024-06-06 22:47:04 +00:00
rpc
scripts	[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)	2024-05-27 14:49:57 +00:00
amp.rst	generalize custom_fwd&custom_bwd to be device-agnostic (#126531)	2024-05-25 06:48:16 +00:00
autograd.rst	Add torch.library.register_autograd (#124071)	2024-04-18 12:47:59 +00:00
backends.rst	preferred blas library; cublaslt gemm implementation (#122106)	2024-04-22 15:38:22 +00:00
benchmark_utils.rst	Adding Compare in torch.utils.benchmark documentation (#125009)	2024-05-03 00:50:54 +00:00
bottleneck.rst
checkpoint.rst	Add missing words to torch.utils.checkpoint doc (#120196)	2024-02-20 20:18:42 +00:00
complex_numbers.rst	Document complex optimizer semantic behavior (#121667)	2024-03-16 00:43:47 +00:00
cond.rst
conf.py	Retire torch.distributed.pipeline (#127354)	2024-06-07 08:11:58 +00:00
config_mod.rst
cpp_extension.rst
cpp_index.rst
cpu.rst
cuda._sanitizer.rst
cuda.rst	[ROCm] TunableOp improvements (#124362)	2024-06-03 22:30:11 +00:00
cuda.tunable.rst	[ROCm] TunableOp improvements (#124362)	2024-06-03 22:30:11 +00:00
cuda_environment_variables.rst	Add doc page for environment variables that effect PyTorch Runtime (#119087)	2024-02-15 21:41:38 +00:00
cudnn_persistent_rnn.rst
cudnn_rnn_determinism.rst
data.rst
ddp_comm_hooks.rst
debugging_environment_variables.rst	Add doc page for environment variables that effect PyTorch Runtime (#119087)	2024-02-15 21:41:38 +00:00
deploy.rst
deterministic.rst
distributed.algorithms.join.rst
distributed.checkpoint.rst	[DCP] Provides default AsyncStager (#124939)	2024-05-02 19:48:54 +00:00
distributed.elastic.rst	Reapply "distributed debug handlers (#126601)" (#127805)	2024-06-04 19:44:30 +00:00
distributed.optim.rst
distributed.pipelining.rst	[pipelining] pipeline() taking microbatch as example input (#128163)	2024-06-07 15:51:53 +00:00
distributed.rst	Retire torch.distributed.pipeline (#127354)	2024-06-07 08:11:58 +00:00
distributed.tensor.parallel.rst	[tp] doc fixes (#121431)	2024-03-08 17:46:44 +00:00
distributions.rst
dlpack.rst
docutils.conf
export.ir_spec.rst
export.rst	[export] provide refine function for automatically accepting dynamic shapes suggested fixes (#127436)	2024-06-07 03:29:06 +00:00
fft.rst
fsdp.rst
func.api.rst
func.batch_norm.rst
func.migrating.rst
func.rst
func.ux_limitations.rst
func.whirlwind_tour.rst
future_mod.rst
futures.rst
fx.experimental.rst	Traceable wrapper subclass support for deferred runtime asserts (#126198)	2024-05-21 01:21:46 +00:00
fx.rst	Implement Graph Transform Observer (#127427)	2024-06-02 06:49:47 +00:00
hub.rst
index.rst	Retire torch.distributed.pipeline (#127354)	2024-06-07 08:11:58 +00:00
jit.rst
jit_builtin_functions.rst
jit_language_reference.rst
jit_language_reference_v2.rst
jit_python_reference.rst
jit_unsupported.rst
jit_utils.rst
library.rst	New Custom Ops Documentation landing page (#127400)	2024-05-30 01:06:04 +00:00
linalg.rst
logging.rst
masked.rst
math-quantizer-equation.png
meta.rst
miscellaneous_environment_variables.rst	Add doc page for environment variables that effect PyTorch Runtime (#119087)	2024-02-15 21:41:38 +00:00
mobile_optimizer.rst
model_zoo.rst
module_tracker.rst	Add module tracker (#125352)	2024-05-04 18:33:35 +00:00
monitor.rst
mps.rst	Conform torch.mps to device module interface (#124676)	2024-04-23 18:38:48 +00:00
mtia.rst	torch.mtia module for MTIA device backend (#123612)	2024-04-26 16:17:54 +00:00
multiprocessing.rst
name_inference.rst
named_tensor.rst
nested.rst
nn.attention.bias.rst
nn.attention.rst
nn.functional.rst	Add RMSNorm module (#121364)	2024-03-29 18:05:28 +00:00
nn.init.rst
nn.rst	Cleanup some duplicated placeholder py:module docs (#123244)	2024-04-05 03:18:53 +00:00
onnx.rst	fix pytorch version for onnx in doc (#124182)	2024-04-17 18:05:15 +00:00
onnx_dynamo.rst
onnx_dynamo_onnxruntime_backend.rst
onnx_torchscript.rst
onnx_torchscript_supported_aten_ops.rst
optim.rst
package.rst
profiler.rst
quantization-accuracy-debugging.rst
quantization-backend-configuration.rst
quantization-support.rst	[quant][pt2e] Add model_is_exported util function (#119726)	2024-02-16 19:29:36 +00:00
quantization.rst	Cleanup some duplicated placeholder py:module docs (#123244)	2024-04-05 03:18:53 +00:00
random.rst
rpc.rst
signal.rst
size.rst	Added a docstring for torch.Size.numel. (#124186)	2024-04-19 09:23:02 +00:00
sparse.rst	Fix typo in sparse.rst (#121826)	2024-03-19 00:17:19 +00:00
special.rst
storage.rst
tensor_attributes.rst	Include the scalar tensor auto-transfer in the doc (#119967)	2024-02-15 22:37:39 +00:00
tensor_view.rst
tensorboard.rst
tensors.rst	Document the legacy constructor for Tensor (#122625)	2024-05-29 23:23:19 +00:00
testing.rst
threading_environment_variables.rst	Add doc page for environment variables that effect PyTorch Runtime (#119087)	2024-02-15 21:41:38 +00:00
torch.ao.ns._numeric_suite.rst
torch.ao.ns._numeric_suite_fx.rst
torch.compiler.rst	Fix links rendering when surrounding code in Dynamo deepdive (#123427)	2024-04-13 04:55:15 +00:00
torch.compiler_aot_inductor.rst	[AOTI] docs: add suggestion to turn on freezing on CPU (#128010)	2024-06-07 08:57:02 +00:00
torch.compiler_api.rst	[torch.export] Support is_compiling() flag for non-strict mode (#119602)	2024-02-29 05:52:51 +00:00
torch.compiler_best_practices_for_backends.rst
torch.compiler_cudagraph_trees.rst
torch.compiler_custom_backends.rst	Fix a link in the compiler backend doc (#126079)	2024-05-21 20:16:04 +00:00
torch.compiler_dynamic_shapes.rst	feat: Add min, max ranges to mark_dynamic API (#119737)	2024-03-07 23:26:03 +00:00
torch.compiler_dynamo_deepdive.rst	Fix links rendering when surrounding code in Dynamo deepdive (#123427)	2024-04-13 04:55:15 +00:00
torch.compiler_dynamo_overview.rst	Rename TorchDynamo -> Dyanamo in the dynamo tutorial doc (#123431)	2024-05-07 05:07:00 +00:00
torch.compiler_fake_tensor.rst
torch.compiler_faq.rst	Fixed broken link and removed unfinished sentence from issue #126367 (#127938)	2024-06-05 07:37:32 +00:00
torch.compiler_fine_grain_apis.rst	[torch.export] Support is_compiling() flag for non-strict mode (#119602)	2024-02-29 05:52:51 +00:00
torch.compiler_get_started.rst	[Inductor] Add 0 initialization to Triton masked loads (#127311)	2024-05-30 04:50:54 +00:00
torch.compiler_inductor_profiling.rst
torch.compiler_ir.rst
torch.compiler_nn_module.rst
torch.compiler_performance_dashboard.rst
torch.compiler_profiling_torch_compile.rst	[docs] Update PT2+Profiler docs (#122272)	2024-03-28 17:52:28 +00:00
torch.compiler_transformations.rst
torch.compiler_troubleshooting.rst	Add force_disable_caches to the docs (#126184)	2024-05-15 07:16:08 +00:00
torch.overrides.rst
torch.rst	torch.mtia module for MTIA device backend (#123612)	2024-04-26 16:17:54 +00:00
torch_cuda_memory.rst
torch_environment_variables.rst	Add doc page for environment variables that effect PyTorch Runtime (#119087)	2024-02-15 21:41:38 +00:00
type_info.rst
utils.rst
xpu.rst	[2/2] Intel GPU Runtime Upstreaming for Generator (#118613)	2024-02-28 05:28:11 +00:00