pytorch/torch
2024-07-25 04:35:26 +00:00
..
_awaits
_C [dynamo] Rename TENSOR_ALIASING to OBJECT_ALIASING. Permit OBJECT_ALIASING for dict guards (#131480) 2024-07-24 00:06:53 +00:00
_C_flatbuffer
_custom_op Revert "Tighten torch.library.infer_schema input types (#130705)" 2024-07-16 12:57:11 +00:00
_decomp [BE] typing for decorators - jit/_decompositions (#131566) 2024-07-24 20:28:28 +00:00
_dispatch
_dynamo Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
_export [ts_convert] Update conversion for aten.tensor (#131549) 2024-07-25 01:11:03 +00:00
_functorch [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
_higher_order_ops [triton_op] fix autotuning (#131363) 2024-07-24 14:03:37 +00:00
_inductor [BE] Annotate subgraph_lowering (#131545) 2024-07-25 04:35:26 +00:00
_lazy [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
_library [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
_logging [pt2] Increase dynamo/inductor default log level to info (#131311) 2024-07-22 17:33:29 +00:00
_numpy Make hashing a SymInt raise an error again (#130548) 2024-07-16 18:30:30 +00:00
_prims Fix out_wrapper, _make_copy_from_view to handle all signatures (#130937) 2024-07-21 20:39:24 +00:00
_prims_common Fix out_wrapper, _make_copy_from_view to handle all signatures (#130937) 2024-07-21 20:39:24 +00:00
_refs [BE] typing for decorators - jit/_decompositions (#131566) 2024-07-24 20:28:28 +00:00
_strobelight
_subclasses Fix fake tensor SymInt caching when there's a SymInt storage_offset (#131500) 2024-07-23 23:37:04 +00:00
_vendor
amp
ao [BE] enable UFMT for torch/ao/quantization/ (#128863) 2024-07-25 04:17:54 +00:00
autograd [Profiler] exclude gpu_user_annotation when accumulating cuda time total (#130733) 2024-07-22 04:35:21 +00:00
backends [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
compiler
contrib
cpu
csrc [Distributed] [10/N] Fix clang-tidy warnings in torch/csrc/distributed/c10d/control_plane (#131671) 2024-07-25 03:46:55 +00:00
cuda Revert "Add wrappers for synchronous GPUDirect Storage APIs (#130633)" 2024-07-23 17:19:34 +00:00
distributed [BE] typing for decorators - jit/_decompositions (#131566) 2024-07-24 20:28:28 +00:00
distributions
export [export] Fix graph_break log registration error when importing export/_trace.py (#131523) 2024-07-24 22:40:24 +00:00
fft
func
futures
fx [triton_op] fix autotuning (#131363) 2024-07-24 14:03:37 +00:00
jit [BE] typing for decorators - jit/_decompositions (#131566) 2024-07-24 20:28:28 +00:00
legacy
lib
linalg
masked [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
monitor
mps
mtia [torch.mtia] Noop set_rng_state and get_rng_state APIs (#130873) 2024-07-24 01:52:21 +00:00
multiprocessing Enable sharing meta tensors between processes (#129520) 2024-07-04 20:29:48 +00:00
nested [NestedTensor] Integrate the mean operator along the jagged dimension into NestedTensor (#131132) 2024-07-23 18:48:34 +00:00
nn [MTIA] Support module.mtia() (#131499) 2024-07-25 04:23:48 +00:00
onnx [ONNX] Update torch.onnx.export API (#131501) 2024-07-24 20:03:17 +00:00
optim [BE] remove unnecessary _dispatch_sqrt by using ** 0.5 (#131358) 2024-07-24 14:58:57 +00:00
package
profiler [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
quantization
signal [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
sparse [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
special
testing [3.13] support 3.13 multiline traces in munge_exc (#131207) 2024-07-24 18:22:30 +00:00
utils Fully type torch/utils/_config_module.py (#131676) 2024-07-24 19:36:09 +00:00
xpu
__config__.py
__future__.py
__init__.py Revert "Add wrappers for synchronous GPUDirect Storage APIs (#130633)" 2024-07-23 17:19:34 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py Revert "Tighten torch.library.infer_schema input types (#130705)" 2024-07-16 12:57:11 +00:00
_deploy.py
_guards.py [BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199) 2024-07-11 17:30:28 +00:00
_jit_internal.py [torchscript] Add logging for model id. (#130118) 2024-07-09 22:24:16 +00:00
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
_namedtensor_internals.py
_ops.py [HOP] Don't send HOPs to torch_dispatch (#131370) 2024-07-23 13:41:08 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor.py [easy] Small rendering fix in Tensor.module_load doc (#130489) 2024-07-12 22:12:53 +00:00
_tensor_docs.py [MTIA] Support module.mtia() (#131499) 2024-07-25 04:23:48 +00:00
_tensor_str.py fix tensor print behavior for XPU (#130523) 2024-07-17 02:03:32 +00:00
_torch_docs.py Introduce the concept of Accelerators to PyTorch doc (#129363) 2024-07-15 14:24:46 +00:00
_utils.py
_utils_internal.py Write trace_structured events to scuba (#130955) 2024-07-19 06:02:47 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py Blocklist certain modules for weights_only load (#131259) 2024-07-22 18:23:21 +00:00
abi-check.cpp
CMakeLists.txt Revert "Add wrappers for synchronous GPUDirect Storage APIs (#130633)" 2024-07-23 17:19:34 +00:00
custom_class.h [2/N] Fix some violations of unused-function and unused-variable checks in torch_cpu (#129878) 2024-07-04 00:39:28 +00:00
custom_class_detail.h [1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301) 2024-07-08 07:03:53 +00:00
extension.h
functional.py
hub.py
library.h [3/N] Fix Wunused-parameter warnings (#131271) 2024-07-20 23:31:03 +00:00
library.py [custom ops] Add register_vmap for custom ops (#130589) 2024-07-23 17:48:38 +00:00
overrides.py [MTIA] Support module.mtia() (#131499) 2024-07-25 04:23:48 +00:00
py.typed
quasirandom.py
random.py
README.txt
return_types.py
script.h
serialization.py Updating Types in torch/_dynamo/utils.py (#131001) 2024-07-23 18:25:52 +00:00
storage.py typing: storage (#130669) 2024-07-16 14:31:35 +00:00
torch_version.py Add mypy typing to torch_version.py (#131447) 2024-07-23 17:31:07 +00:00
types.py FakeTensor cache SymInt support (#127596) 2024-07-21 19:26:38 +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.