pytorch/torch
2023-10-13 16:33:07 +00:00
..
_awaits
_C Public API for constructing NT with jagged layout from tensor list (#111078) 2023-10-13 03:27:41 +00:00
_C_flatbuffer
_custom_op [custom op] Use canonical API to constrain unbacked values (#108372) 2023-10-10 05:14:28 +00:00
_decomp Revert "[core ATen IR] Add decompositions for max, min, var_mean (#110906)" 2023-10-11 11:41:21 +00:00
_dispatch
_dynamo [dynamo] Forward fix a bunch of distributed collective allow fixes (#111171) 2023-10-13 15:49:04 +00:00
_export [export] Fix issue with internal model (#111140) 2023-10-13 14:07:07 +00:00
_functorch Fix typo under torch/_functorch directory (#111067) 2023-10-11 23:09:36 +00:00
_higher_order_ops Add a side table for triton kernels to avoid using itertools.partial (#110633) 2023-10-08 02:01:59 +00:00
_inductor [pt][group_fusion] fix shape guarding in fusion candidate search (#111174) 2023-10-13 16:28:02 +00:00
_lazy
_library [custom op] Use canonical API to constrain unbacked values (#108372) 2023-10-10 05:14:28 +00:00
_logging [logging] log exceptions when provided (#111164) 2023-10-13 03:52:26 +00:00
_numpy Fix typo under torch/_numpy directory (#110782) 2023-10-07 17:42:35 +00:00
_prims Use torch._check for cat error checking (#111035) 2023-10-12 03:28:27 +00:00
_prims_common Expand NT subclass to support SAM (#109123) 2023-10-12 20:33:22 +00:00
_refs [BE] Enable Ruff's Flake8 PYI030 (#111103) 2023-10-12 13:31:59 +00:00
_subclasses [Inductor] support channel last for xpu conv in inductor layout opt path (#111018) 2023-10-12 15:13:50 +00:00
amp
ao define public API for torch.nn.utils (#111026) 2023-10-12 23:05:23 +00:00
autograd Allow specifiying inputs as GradientEdge in autograd APIs (#110867) 2023-10-12 04:08:44 +00:00
backends typo: add space after cudnn error messages (#110806) 2023-10-08 20:58:40 +00:00
compiler
contrib
cpu Add current_device() to torch.cpu (#110987) 2023-10-11 05:13:10 +00:00
csrc [PyTorch] AOTI: Add aoti_torch_assign_tensors to ABI (#110909) 2023-10-13 16:07:05 +00:00
cuda
distributed [DCP] Remove _shard_tensor() call in load_sharded_optimizer_state_dict in optimizer.py (#111096) 2023-10-12 20:27:06 +00:00
distributions
export [export] Reland "Fix graph signature data model to list of specs." (#111136) 2023-10-13 02:04:29 +00:00
fft
func
futures
fx [inductor] Implement Fx graph caching to improve warm compilation time. (#103453) 2023-10-13 13:33:56 +00:00
jit Nvfuser cpp api deprecation attempt 2 (#110881) 2023-10-10 08:07:03 +00:00
legacy
lib Move at::{Refcounted,}MapAllocator to c10 (#109881) 2023-10-12 10:51:13 +00:00
linalg
masked Make is_sparse a property of MaskedTensor (#110725) 2023-10-09 22:35:38 +00:00
monitor
mps
multiprocessing Multiprocessing support for NT (#110292) 2023-10-10 21:58:19 +00:00
nested Public API for constructing NT with jagged layout from tensor list (#111078) 2023-10-13 03:27:41 +00:00
nn define public API for torch.nn.utils (#111026) 2023-10-12 23:05:23 +00:00
onnx Fix typo under torch/onnx directory (#110697) 2023-10-06 18:21:00 +00:00
optim [BE] Enable Ruff's Flake8 PYI042 (#111114) 2023-10-13 16:33:07 +00:00
package Fix typo under torch directory (#110824) 2023-10-09 19:16:43 +00:00
profiler Reland [Profiler] Improve the docstring for export_memory_timeline (#110983) 2023-10-11 16:42:05 +00:00
quantization Add CUTLASS-based support for mixed dtypes matrix multiplication (#110981) 2023-10-11 21:47:52 +00:00
signal
sparse [sparse] Fix semi-structured sparse shape mismatch bug (#110420) 2023-10-10 03:07:31 +00:00
special
testing add Half support for bernoulli on CPU (#104176) 2023-10-13 01:18:55 +00:00
utils Add set_checkpoint_debug_enabled that overrides local setting (#110728) 2023-10-11 02:12:31 +00:00
__config__.py
__future__.py
__init__.py Move export.constrain_as_* to torch._constrain_as_* (#110757) 2023-10-12 05:37:44 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py [test][docs] Fix doctest warnings for syntax errors (#110517) 2023-10-05 00:00:06 +00:00
_deploy.py
_guards.py [dynamo] Remove SuperSource (#110475) 2023-10-08 04:45:06 +00:00
_jit_internal.py Fix typo under torch directory (#110824) 2023-10-09 19:16:43 +00:00
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py Quant: add weight int4pack mm kernel (#110914) 2023-10-13 01:21:18 +00:00
_namedtensor_internals.py
_ops.py Fix typo under torch directory (#110824) 2023-10-09 19:16:43 +00:00
_python_dispatcher.py
_sources.py
_storage_docs.py
_tensor.py
_tensor_docs.py
_tensor_str.py
_torch_docs.py Fix a typo in cholesky_inverse documentation (#110364) 2023-10-04 12:30:11 +00:00
_utils.py Fix typo under torch directory (#110824) 2023-10-09 19:16:43 +00:00
_utils_internal.py
_VF.py
_vmap_internals.py
_weights_only_unpickler.py
abi-check.cpp
CMakeLists.txt [2/N] Enable Wunused-result, Wunused-variable and Wmissing-braces in torch targets (#110836) 2023-10-11 23:49:15 +00:00
custom_class.h
custom_class_detail.h
extension.h
functional.py Add torch.unravel_index (#110580) 2023-10-12 00:55:51 +00:00
hub.py
library.h [Reland2] Remove calls of c10::either (#110487) 2023-10-06 00:25:15 +00:00
library.py [test][docs] Fix doctest warnings for syntax errors (#110517) 2023-10-05 00:00:06 +00:00
overrides.py Move export.constrain_as_* to torch._constrain_as_* (#110757) 2023-10-12 05:37:44 +00:00
py.typed
quasirandom.py
random.py use torch.xpu.manual_seed_all in torch.seed (#110376) 2023-10-03 13:41:55 +00:00
README.txt
return_types.py
script.h
serialization.py fix get device index if has _utils._get_device_index in privateuse1 (#108123) 2023-10-07 06:18:59 +00:00
storage.py Fix type promotion of float8_e5m2 and float8_e4m3fn (#110279) 2023-10-05 01:28:48 +00:00
torch_version.py
types.py Allow specifiying inputs as GradientEdge in autograd APIs (#110867) 2023-10-12 04:08:44 +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.