pytorch/torch
2024-08-07 18:25:33 +00:00
..
_awaits
_C Add padding_side to pad_sequence with "left" and "right" options ("right" as default) (#131884) 2024-08-07 15:53:07 +00:00
_C_flatbuffer
_custom_op
_decomp [traced-graph][sparse] sparsity propagation for all current tests (#132690) 2024-08-06 21:18:13 +00:00
_dispatch
_dynamo Revert "Consolidate SymDispatchMode into ProxyTensorMode (#132674)" 2024-08-07 18:25:33 +00:00
_export Revert "[BE] Simplify code interacting with get_proxy_mode/enable_tracing (#132675)" 2024-08-07 18:25:33 +00:00
_functorch Revert "Only thunkify proxies in some situations (#132421)" 2024-08-07 15:29:54 +00:00
_higher_order_ops Revert "[BE] Simplify code interacting with get_proxy_mode/enable_tracing (#132675)" 2024-08-07 18:25:33 +00:00
_inductor fix autotuning init issues (#132837) 2024-08-07 16:36:47 +00:00
_lazy
_library [Doc] fix some typos (found by codespell and typos) (#132544) 2024-08-05 17:21:56 +00:00
_logging Add cudagraph static inputs logging (#132726) 2024-08-06 12:01:20 +00:00
_numpy [test/torch_np] Fix usages of deprecated NumPy 2.0 APIs in numpy_tests (#131909) 2024-08-05 16:21:08 +00:00
_prims Revert "[BE] Simplify code interacting with get_proxy_mode/enable_tracing (#132675)" 2024-08-07 18:25:33 +00:00
_prims_common
_refs Don't attempt to compute hints for unbacked expressions (#132060) 2024-08-02 16:39:14 +00:00
_strobelight
_subclasses Conversions between strided and jagged layouts for Nested Tensors (#115749) 2024-08-07 14:18:53 +00:00
_vendor
amp Revert "[MPS] Add support for autocast in MPS (#99272)" 2024-08-05 19:59:04 +00:00
ao [export][fx] More robust DCE pass (#132764) 2024-08-06 22:27:22 +00:00
autograd torch.autograd.graph.increment_version: accept List[Tensor], use in AOTDispatcher (#132652) 2024-08-06 17:46:48 +00:00
backends [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
compiler
contrib
cpu [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
csrc Add padding_side to pad_sequence with "left" and "right" options ("right" as default) (#131884) 2024-08-07 15:53:07 +00:00
cuda Allow torch.cuda.memory.mem_get_info to take a device str argument with an unspecified device index. (#132616) 2024-08-06 13:19:46 +00:00
distributed Revert "[BE] Simplify code interacting with get_proxy_mode/enable_tracing (#132675)" 2024-08-07 18:25:33 +00:00
distributions
export [export][fx] More robust DCE pass (#132764) 2024-08-06 22:27:22 +00:00
fft
func
futures
fx Revert "Consolidate SymDispatchMode into ProxyTensorMode (#132674)" 2024-08-07 18:25:33 +00:00
jit [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
legacy
lib
linalg
masked [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
monitor [pytorch][counters] Pybind for WaitCounter (#132357) 2024-08-02 16:08:10 +00:00
mps [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
mtia [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
multiprocessing [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
nested Conversions between strided and jagged layouts for Nested Tensors (#115749) 2024-08-07 14:18:53 +00:00
nn fix autotuning init issues (#132837) 2024-08-07 16:36:47 +00:00
onnx [ONNX] Rename _internal/exporter to _exporter_legacy (#132429) 2024-08-03 04:23:05 +00:00
optim Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-08-07 00:05:20 +00:00
package
profiler
quantization
signal
sparse Add tests to bsr_dense_addmm_meta. Tune bsr_dense_addmm kernel for ViT shapes. (#132646) 2024-08-05 20:22:33 +00:00
special
testing [DeviceMesh] Create new group for 1D mesh when default backend is 'gloo' and 'cuda' is available (#132709) 2024-08-07 16:13:11 +00:00
utils Revert "Loads .pyd instead of .so in MemPool test for windows (#132749)" 2024-08-07 18:08:44 +00:00
xpu
__config__.py
__future__.py
__init__.py Revert "Fix infinite recursion while walking to submodules (#132763)" 2024-08-07 18:20:27 +00:00
_appdirs.py
_classes.py
_compile.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_custom_ops.py
_deploy.py
_guards.py [dynamo][inline_inbuilt_nn_modules] Mark nn module tensor static for cudagraphs (#132736) 2024-08-06 20:13:28 +00:00
_jit_internal.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_linalg_utils.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_lobpcg.py fix dtype mismatch in lobpcg eigen solver (#132762) 2024-08-07 18:20:46 +00:00
_lowrank.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_meta_registrations.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_namedtensor_internals.py
_ops.py Actually report the HOP and subclass/mode when there isn't a registration (#132550) 2024-08-06 21:33:10 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_streambase.py
_tensor.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_tensor_docs.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_tensor_str.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_torch_docs.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_utils.py Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-08-07 00:05:20 +00:00
_utils_internal.py
_VF.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_vmap_internals.py
_weights_only_unpickler.py
abi-check.cpp
CMakeLists.txt
custom_class.h
custom_class_detail.h
extension.h
functional.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
hub.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
library.h
library.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
overrides.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
py.typed
quasirandom.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
random.py
README.txt
return_types.py
script.h
serialization.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
storage.py
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.