pytorch/torch
PyTorch MergeBot e363a8a222 Revert "[pipelining] Add pipeline stage test (#126721)"
This reverts commit b948b1ad7a.

Reverted https://github.com/pytorch/pytorch/pull/126721 on behalf of https://github.com/clee2000 due to The test_public_bindings failure is real, you just got unlucky since it was also broken on trunk for a different reason ([comment](https://github.com/pytorch/pytorch/pull/126721#issuecomment-2121725408))
2024-05-21 04:40:05 +00:00
..
_awaits
_C Revert "c10d: add Collectives abstraction (#125978)" 2024-05-20 07:40:41 +00:00
_C_flatbuffer
_custom_op
_decomp [ROCm] Implement forward AD for miopen_batch_norm (#125069) 2024-05-14 19:09:50 +00:00
_dispatch
_dynamo [Dynamo] Treat integers stored on nn.Modules as dynamic (#126466) 2024-05-21 03:31:20 +00:00
_export Tool for scouting exportability in one shot (#126471) 2024-05-18 00:10:46 +00:00
_functorch Prevent partitioner from ever saving views (#126446) 2024-05-20 23:40:56 +00:00
_higher_order_ops Add Lowering for FlexAttention Backwards (#125515) 2024-05-17 00:41:55 +00:00
_inductor Script for choosing template configurations (#126560) 2024-05-21 02:28:39 +00:00
_lazy
_library Fix documentation for register_fake_class (#126422) 2024-05-17 00:45:21 +00:00
_logging Remove dist_ prefix from TORCH_LOGS shortcuts (#126499) 2024-05-18 00:07:30 +00:00
_numpy Remove removed ruff rule TRY200 (#126256) 2024-05-17 16:31:05 +00:00
_prims
_prims_common
_refs Remove removed ruff rule TRY200 (#126256) 2024-05-17 16:31:05 +00:00
_subclasses Make propagate_real_tensor more safe (#126281) 2024-05-15 23:57:02 +00:00
_vendor
amp
ao [Quant][PT2E] enable qlinear post op fusion for dynamic quant & qat (#122667) 2024-05-20 15:55:18 +00:00
autograd Fix public binding to actually traverse modules (#126103) 2024-05-15 19:36:03 +00:00
backends Fix public binding to actually traverse modules (#126103) 2024-05-15 19:36:03 +00:00
compiler
contrib
cpu [Doc] Add deprecated autocast comments for doc (#126062) 2024-05-16 05:26:43 +00:00
csrc [BE] split seq_id to collective_seq_id and p2p_seq_id (#125727) 2024-05-21 03:26:49 +00:00
cuda [ROCm] amdsmi library integration (#119182) 2024-05-21 01:59:26 +00:00
distributed Revert "[pipelining] Add pipeline stage test (#126721)" 2024-05-21 04:40:05 +00:00
distributions
export Refactor variables / function names related to non-strict export (#126458) 2024-05-18 06:05:14 +00:00
fft
func
futures
fx Traceable wrapper subclass support for deferred runtime asserts (#126198) 2024-05-21 01:21:46 +00:00
jit [BE][Easy] Remove usage of deprecated ast.Str, ast.Ellipsis and ast.NameConstant (#125912) 2024-05-10 17:35:35 +00:00
legacy
lib
linalg
masked
monitor
mps
mtia
multiprocessing [BE]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +00:00
nested [NT] Make NestedTensor register as having symbolic sizes/strides (#124687) 2024-05-13 16:50:25 +00:00
nn Fix flexattention not realizing inputs before lowering (also refactored runtime estimation) (#126615) 2024-05-20 23:40:56 +00:00
onnx Replace torch.library.impl_abstract with torch.library.register_fake (#126606) 2024-05-19 13:21:39 +00:00
optim [optim] Fix: wrong ASGD implementation (#126375) 2024-05-17 15:46:39 +00:00
package
profiler [Profiler] Clean up deprecated use_cuda by default (#126180) 2024-05-14 21:23:31 +00:00
quantization
signal
sparse
special
testing Fix flexattention not realizing inputs before lowering (also refactored runtime estimation) (#126615) 2024-05-20 23:40:56 +00:00
utils [cudagraph] fix verbose graph logging (#126694) 2024-05-21 00:55:15 +00:00
xpu Add some type annotations to python stream and event classes (#126171) 2024-05-15 04:58:07 +00:00
__config__.py
__future__.py
__init__.py fixed typo in documentation (#125974) 2024-05-13 04:37:51 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py Replace torch.library.impl_abstract with torch.library.register_fake (#126606) 2024-05-19 13:21:39 +00:00
_deploy.py
_guards.py
_jit_internal.py [BE][Easy] Remove usage of deprecated ast.Str, ast.Ellipsis and ast.NameConstant (#125912) 2024-05-10 17:35:35 +00:00
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py [optim] Fix: wrong ASGD implementation (#126375) 2024-05-17 15:46:39 +00:00
_namedtensor_internals.py
_ops.py [export] add SchemaCheckMode testing for pre-dispatch export, OpInfo (#125481) 2024-05-14 21:07:21 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py Add some type annotations to python stream and event classes (#126171) 2024-05-15 04:58:07 +00:00
_tensor.py [BE]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +00:00
_tensor_docs.py
_tensor_str.py
_torch_docs.py Switched from parameter in can_cast to from_. (#126030) 2024-05-16 20:58:24 +00:00
_utils.py xpu: implement xpu serialization (#125530) 2024-05-16 20:22:17 +00:00
_utils_internal.py [cprofile] log manifold link instead of raw data to trace_structured (#126451) 2024-05-21 00:44:55 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py Allow tensor subclasses and add torch.serialization.add_safe_globals that allows users to allowlist classes for weights_only load (#124331) 2024-05-17 17:56:57 +00:00
abi-check.cpp
CMakeLists.txt
custom_class.h
custom_class_detail.h [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
extension.h
functional.py
hub.py
library.h [codemod] c10:optional -> std::optional (#126135) 2024-05-14 19:35:51 +00:00
library.py Add registration API for torch.compile-eager (#121387) 2024-05-10 00:30:27 +00:00
overrides.py
py.typed
quasirandom.py
random.py
README.txt
return_types.py
script.h
serialization.py Allow tensor subclasses and add torch.serialization.add_safe_globals that allows users to allowlist classes for weights_only load (#124331) 2024-05-17 17:56:57 +00:00
storage.py xpu: implement xpu serialization (#125530) 2024-05-16 20:22:17 +00:00
torch_version.py
types.py
version.py.tpl

Note [TH abstraction violation]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

TH/THC provide some hpp headers, which are proper C++ headers rather than
C headers.  These headers serve double duty as *internal implementation
detail* headers, whose contents should largely not be used by external
clients.

Ideally, we would not install these headers at all; instead, you should
use public functions (in headers like `THTensor.h`, NOT `THTensor.hpp`)
to manipulate these structs.  However, there are a few places
in torch/csrc where we violate this abstraction.  They are marked with
a pointer to this note.  Each of those sites will have to be refactored
when we refactor the guts of THTensor and related structures.