pytorch/torch
2023-09-13 17:42:46 +00:00
..
_awaits
_C Add NestedTensor python subclass (#108314) 2023-09-11 18:29:20 +00:00
_C_flatbuffer
_custom_op Extend impl_backward to be usable with torch.library operators (#106817) 2023-08-14 14:33:46 +00:00
_decomp Check index size during decomp of index_add (#108826) 2023-09-13 13:06:26 +00:00
_dispatch Fix some fake mode confusion between inner/outer fake mode in export (#106515) 2023-08-04 15:42:23 +00:00
_dynamo Back out "Dynamo support for autograd.Function w/ once_differentiable (#108686)" (#109199) 2023-09-13 15:43:20 +00:00
_export [export] Lift constant tensors as buffes (reland) (#109040) 2023-09-12 15:23:00 +00:00
_functorch fix 'Node' object is not iterable in functorch.compile.minifier (#103011) 2023-09-12 23:47:40 +00:00
_higher_order_ops Make make_fx cond preserve node meta (#108356) 2023-09-01 22:43:55 +00:00
_inductor [inductor] Enable Mypy Checking for torch/_inductor/codecache.py (#108789) 2023-09-13 14:05:35 +00:00
_lazy
_logging [Easy] log graphs in compiled_autograd if TORCH_LOGS=compiled_autograd (#108991) 2023-09-12 00:15:02 +00:00
_numpy Fix torch._numpy.random (#108944) 2023-09-13 05:08:19 +00:00
_prims Don't set requires_grad inside meta function (#108988) 2023-09-12 12:24:13 +00:00
_prims_common optests improvements based on torchvision usage on nms (#108929) 2023-09-13 13:26:15 +00:00
_refs Introduce Tensor overload to linspace and logspace (#104889) 2023-09-11 23:30:40 +00:00
_subclasses optests improvements based on torchvision usage on nms (#108929) 2023-09-13 13:26:15 +00:00
amp Allow float dtype when Autocast CPU Disabled (#107348) 2023-09-01 00:49:44 +00:00
ao [quant][be] Reduce warnings in tests (#108922) 2023-09-12 21:54:33 +00:00
autograd Back out "Dynamo support for autograd.Function w/ once_differentiable (#108686)" (#109199) 2023-09-13 15:43:20 +00:00
backends [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
compiler
contrib
cpu
csrc Use global variables to register the return_types namedtuples (#108832) 2023-09-13 17:42:46 +00:00
cuda Improve torch.cuda.amp type hints (#108630) 2023-09-08 06:06:25 +00:00
distributed [dtensor][4/n] don't use make_fx for strategy propagation (#108262) 2023-09-13 04:08:02 +00:00
distributions Spelling fix (#108490) 2023-09-04 16:59:35 +00:00
export Add torch.export.register_dataclass API (#109152) 2023-09-13 04:17:12 +00:00
fft
func [pt2] support vmap (#101707) 2023-08-09 03:39:33 +00:00
futures
fx optests improvements based on torchvision usage on nms (#108929) 2023-09-13 13:26:15 +00:00
jit removing some redundant str splits (#106089) 2023-09-01 00:22:58 +00:00
legacy
lib [Reland] fix c10:TempFile APIs on Windows (#108508) 2023-09-10 16:58:41 +00:00
linalg fix matrix_power documentation bug (#108585) 2023-09-05 22:08:46 +00:00
masked
monitor
mps [MPS] Introduce torch.mps.Event() APIs (#102121) 2023-08-08 03:45:45 +00:00
multiprocessing
nested Add NestedTensor python subclass (#108314) 2023-09-11 18:29:20 +00:00
nn Small fix in SDPA docstring codeblock (#109086) 2023-09-12 16:48:46 +00:00
onnx [ONNX] Support None in fx.args as torchlib inputs (#108708) 2023-09-13 02:47:16 +00:00
optim Add capturable ASGD impl (#107857) 2023-09-07 06:30:30 +00:00
package removing some redundant str splits (#106089) 2023-09-01 00:22:58 +00:00
profiler [profiler] move _enable_dynamo_cache_lookup_profiler (#107720) 2023-08-23 23:41:35 +00:00
quantization
signal
sparse Improve triton bsr_dense_mm performance on column-major ordered inputs with float32 dtype (#108512) 2023-09-06 17:30:06 +00:00
special
testing add Half support for BatchNorm on CPU (#102070) 2023-09-13 17:30:16 +00:00
utils Revert "[ROCm] Add ROCm AMDGPU support for inductor cpp codegen (#105141)" 2023-09-12 12:29:55 +00:00
__config__.py
__future__.py
__init__.py Revert "Re-land: Break graph on manual_seed. (#108647)" 2023-09-08 21:18:00 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py Extend impl_backward to be usable with torch.library operators (#106817) 2023-08-14 14:33:46 +00:00
_deploy.py
_guards.py Revert "Force synced KJT to trace unbacked SymInt (#108960)" 2023-09-12 14:37:36 +00:00
_jit_internal.py
_linalg_utils.py
_lobpcg.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
_lowrank.py
_meta_registrations.py Flash Attention v2 (#105602) 2023-09-13 13:59:05 +00:00
_namedtensor_internals.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
_ops.py Support cond and out_dtype for predispatch (#107941) 2023-08-25 17:37:16 +00:00
_python_dispatcher.py
_sources.py
_storage_docs.py
_tensor_docs.py Modify signature for tensor.tile in doc (#106295) 2023-08-01 19:51:52 +00:00
_tensor_str.py
_tensor.py [PyTorch][Tensor] Introduce tensor.dim_order (#106835) 2023-08-25 00:06:03 +00:00
_torch_docs.py Introduce Tensor overload to linspace and logspace (#104889) 2023-09-11 23:30:40 +00:00
_utils_internal.py [Dynamo] Improve PT2 fbcode logging observability (#106932) 2023-08-11 20:46:04 +00:00
_utils.py Add optional is_coalesced argument to sparse coo tensor factory function. (#107638) 2023-08-26 07:24:29 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py
abi-check.cpp
CMakeLists.txt fix a CMake syntax warning (#108849) 2023-09-08 16:10:12 +00:00
custom_class_detail.h
custom_class.h Revert "Remove some unnecessary <iostream> includes from headers (#106914)" 2023-08-29 02:22:33 +00:00
extension.h reduce header file to boost cpp_wrapper build. (#107585) 2023-08-22 11:58:47 +00:00
functional.py Improve docs for torch.unique dim argument (#108292) 2023-09-02 11:09:09 +00:00
hub.py Default permissions for torch.hub downloads (#82869) 2023-08-24 15:48:24 +00:00
library.h
library.py
overrides.py Add torch.export.register_dataclass API (#109152) 2023-09-13 04:17:12 +00:00
py.typed
quasirandom.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
random.py Revert "Re-land: Break graph on manual_seed. (#108647)" 2023-09-08 21:18:00 +00:00
README.txt
return_types.py
script.h
serialization.py Fix error message concatanation (#108581) 2023-09-05 19:46:52 +00:00
storage.py
torch_version.py
types.py [BE]: Apply PYI autofixes to various types (#107521) 2023-08-20 02:42:21 +00:00
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.