pytorch/torch
2024-08-29 20:56:16 +00:00
..
_awaits Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_C Uses MemPoolContext to route allocations from CUDACachingAllocator (#134685) 2024-08-29 03:56:31 +00:00
_C_flatbuffer
_custom_op
_decomp Remove unnecessary expect_true in split_with_sizes (#133439) 2024-08-27 01:34:00 +00:00
_dispatch
_dynamo [dynamo][itertools] refactor itertools.chain and itertools.chain.from_iterable to use polyfills (#133864) 2024-08-29 20:56:16 +00:00
_export hang dim hint constants off Dim (#134702) 2024-08-29 01:02:01 +00:00
_functorch Never CSE aten.empty in the partitioner (#134703) 2024-08-29 13:51:19 +00:00
_higher_order_ops Changed masked out rows logsumexp to be -inf and not zero (#134650) 2024-08-29 17:22:52 +00:00
_inductor [Inductor] Fix error checking for scaled_mm lowering (#134765) 2024-08-29 20:18:42 +00:00
_lazy Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_library Improve custom ops aliasing error message (#134688) 2024-08-28 22:22:04 +00:00
_logging [easy] rm duplicate definition for inductor in TORCH_LOGS documentation (#134480) 2024-08-27 20:15:10 +00:00
_numpy
_prims [hop] require hops to override __call__. (#134352) 2024-08-28 19:56:40 +00:00
_prims_common Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_refs Bucketize fix to include number and tensor inputs (#133652) 2024-08-28 13:35:41 +00:00
_strobelight
_subclasses preserve aten::to device in export training (#134622) 2024-08-29 14:53:30 +00:00
_vendor
amp
ao Update pt2e numeric debugger to use node.meta["custom"] field (#134040) 2024-08-27 19:51:03 +00:00
autograd Move module_tracker to logging for confused hierarchy (#134467) 2024-08-26 19:39:08 +00:00
backends [sparse] Add cuSPARSELt as a backend (#128534) 2024-08-21 22:06:07 +00:00
compiler [dynamo] simplify polyfill registration for builtins.all and builtins.any (#133769) 2024-08-29 20:56:16 +00:00
contrib
cpu
csrc [5/N] Reconcile barrier and NaN checker (#134707) 2024-08-29 18:51:12 +00:00
cuda Uses MemPoolContext to route allocations from CUDACachingAllocator (#134685) 2024-08-29 03:56:31 +00:00
distributed Revert "[reland][dtensor][MTPG] make sharding prop lru cache not shared among threads (#134509)" 2024-08-29 16:39:19 +00:00
distributions Add correct typing annotations to rsample() for all distributions (#133516) 2024-08-18 20:31:54 +00:00
export [export] use single FQN for param_buffer_mapping (#134500) 2024-08-29 17:06:31 +00:00
fft
func
futures
fx [export] enumerate unsupported sympy.Functions (#134271) (#134598) 2024-08-28 00:34:38 +00:00
jit [BE]: Update ruff to 0.6.0 (#133609) 2024-08-16 14:11:01 +00:00
legacy
lib
linalg Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
masked Revert "Add MaskedTensor support to *_like API (#128637)" 2024-08-29 01:42:52 +00:00
monitor
mps
mtia Improve error msg on _lazy_init() error (#134159) 2024-08-22 15:10:50 +00:00
multiprocessing Create processes in parallel in mp.start_processes for forkserver (#134629) 2024-08-28 21:34:32 +00:00
nested [NJT] Support NJT SDPA + meta-device flop counting (#134289) 2024-08-29 03:43:42 +00:00
nn [Dynamo] Support inspect.signature.Parameter getattr (#134636) 2024-08-28 09:59:41 +00:00
onnx [BE] typing for decorators - fx/_compatibility (#134054) 2024-08-26 04:00:27 +00:00
optim Avoid Host & Device Sync In LR Scheduler (#133663) 2024-08-22 03:52:43 +00:00
package [torch/package_importer] add compatibility name mapping (#134376) 2024-08-25 19:34:46 +00:00
profiler [Profiler/CPU] Add Test for Dynamic Activity Toggling [4/n] (#134149) 2024-08-23 01:13:42 +00:00
quantization
signal Fix docstring for torch.signal.windows.nuttall (#134512) 2024-08-26 22:51:43 +00:00
sparse [sparse] Update cuSPARSELt to v0.6.2 (#134022) 2024-08-23 19:34:53 +00:00
special
testing Revert "Add MaskedTensor passthrough: unfold, F.Unfold, F.Fold, stack (#125262)" 2024-08-28 23:10:07 +00:00
utils [export] enumerate unsupported sympy.Functions (#134271) (#134598) 2024-08-28 00:34:38 +00:00
xpu
__config__.py
__future__.py
__init__.py Revert "[BE]: Update Typeguard to TypeIs for better type inference (#133814)" 2024-08-21 16:13:34 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py
_deploy.py Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_guards.py [PT2/Profiler] Add Context Info to Torch-Compiled Regions (#132765) 2024-08-27 04:55:04 +00:00
_jit_internal.py
_linalg_utils.py
_lobpcg.py [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
_lowrank.py
_meta_registrations.py [CPU][flash attention] make the stride of output align with input (#134656) 2024-08-29 16:04:25 +00:00
_namedtensor_internals.py
_ops.py [hop] require hops to override __call__. (#134352) 2024-08-28 19:56:40 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py
_tensor_str.py
_tensor.py Add torch.serialization.skip_data context manager (#134504) 2024-08-29 04:52:52 +00:00
_torch_docs.py [docs] improve torch.stack example code to be reproducible (#133857) 2024-08-21 14:07:02 +00:00
_utils_internal.py Create a JustknobConfig for use in config (#134161) 2024-08-27 16:07:33 +00:00
_utils.py Add torch.serialization.skip_data context manager (#134504) 2024-08-29 04:52:52 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py Clarify error messages for NEWOBJ and BUILD in weights_only unpickler (#134346) 2024-08-27 14:45:39 +00:00
abi-check.cpp
CMakeLists.txt [split build] fix distributed problems (#134502) 2024-08-27 21:12:58 +00:00
custom_class_detail.h Enable clang-tidy coverage on torch/*.h (#133422) 2024-08-15 18:52:08 +00:00
custom_class.h Enable clang-tidy coverage on torch/*.h (#133422) 2024-08-15 18:52:08 +00:00
extension.h
functional.py
hub.py
library.h Enable clang-tidy coverage on torch/*.h (#133422) 2024-08-15 18:52:08 +00:00
library.py Improve opcheck docs. (#134692) 2024-08-28 22:22:04 +00:00
overrides.py Add new ops wrapped_linear_prepack and wrapped_quantized_linear_prepacked (#134232) 2024-08-23 04:54:26 +00:00
py.typed
quasirandom.py
random.py
README.txt
return_types.py
script.h
serialization.py Add torch.serialization.skip_data context manager (#134504) 2024-08-29 04:52:52 +00:00
storage.py Add torch.serialization.skip_data context manager (#134504) 2024-08-29 04:52:52 +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.