pytorch/torch
2024-05-09 01:34:19 +00:00
..
_awaits
_C add uuid in cudaDeviceProperties (#125083) 2024-05-08 19:15:55 +00:00
_C_flatbuffer
_custom_op
_decomp Add a code comment about torch._check_is_size in tensor_split (#125292) 2024-05-02 02:25:38 +00:00
_dispatch
_dynamo [dynamo][disable] Move disable impl to its own __call__ method (#125486) 2024-05-09 01:03:12 +00:00
_export [export] disable_forced_specializations (#124949) 2024-05-08 18:42:39 +00:00
_functorch Retry: Low mem max_pool2d_with_indices (#122832) 2024-05-08 19:37:08 +00:00
_higher_order_ops Add support for multiple flexattention calls in a single compile (#125516) 2024-05-07 21:37:37 +00:00
_inductor Revert "[Inductor] [Quant] Enable lowering of quant per tensor and refactor quant pattern (#124041)" 2024-05-09 01:34:19 +00:00
_lazy
_library [dynamo] support torchbind object input (#124978) 2024-05-07 03:02:00 +00:00
_logging [dynamo] add trace_bytecode logging artifact (#125360) 2024-05-02 22:01:00 +00:00
_numpy [BE]: Try TCH autofixes on torch/ (#125536) 2024-05-05 23:13:59 +00:00
_prims Make min(stride, strides[idx]) in collapse_view_helper size oblivious (#125301) 2024-05-02 02:39:58 +00:00
_prims_common Fix 'Could not infer dtype of SymBool' on torch.tensor call (#125656) 2024-05-07 22:41:49 +00:00
_refs Fix 'Could not infer dtype of SymBool' on torch.tensor call (#125656) 2024-05-07 22:41:49 +00:00
_subclasses [BE]: Improve exception typing. Remove NOQAs (#125535) 2024-05-08 14:07:13 +00:00
_vendor
amp make torch.amp.autocast more generic (#125103) 2024-05-08 12:13:26 +00:00
ao [BE]: Improve exception typing. Remove NOQAs (#125535) 2024-05-08 14:07:13 +00:00
autograd Fix torch.profiler Schedule Function (Function Event only) (#125510) 2024-05-08 23:32:50 +00:00
backends Better core binding in torch.backends.xeon.run_cpu when launced from torchrun with --nproc-per-node (#123711) 2024-05-09 00:32:11 +00:00
compiler Fix broken docs (#124940) 2024-04-26 19:24:52 +00:00
contrib
cpu Revert "[inductor] add cpp builder code. (#124045)" 2024-05-08 15:33:20 +00:00
csrc [compiled autograd][dynamo] improve lifted autograd.Function.backward handling and fallback to pseudo-eager (#125661) 2024-05-08 21:00:37 +00:00
cuda [BE]: Apply ruff FURB 118. (#124743) 2024-04-26 14:34:52 +00:00
distributed [DeviceMesh] Make _validate_tp_mesh_dim support 3D (#125763) 2024-05-08 21:22:11 +00:00
distributions
export [export] disable_forced_specializations (#124949) 2024-05-08 18:42:39 +00:00
fft
func
futures
fx Fix infinite recursion in API BC test (#125706) 2024-05-08 23:07:16 +00:00
jit [dynamo] fix potentially missing _torchdynamo_inline from ScriptFunction (#125447) 2024-05-06 20:36:56 +00:00
legacy
lib [codemod] Fix missing field initializer in caffe2/torch/lib/libshm/core.cpp +2 (#125047) 2024-04-27 15:52:56 +00:00
linalg Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
masked Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
monitor
mps Conform torch.mps to device module interface (#124676) 2024-04-23 18:38:48 +00:00
mtia torch.mtia module for MTIA device backend (#123612) 2024-04-26 16:17:54 +00:00
multiprocessing
nested Revert "[FX] Update type hints in torch.fx._compatibility.py (#125469)" 2024-05-06 18:36:43 +00:00
nn [BE]: Try TCH autofixes on torch/ (#125536) 2024-05-05 23:13:59 +00:00
onnx Revert "[FX] Update type hints in torch.fx._compatibility.py (#125469)" 2024-05-06 18:36:43 +00:00
optim Update LRScheduler to handle tensor LR (#123753) 2024-05-09 00:52:43 +00:00
package [BE]: Try TCH autofixes on torch/ (#125536) 2024-05-05 23:13:59 +00:00
profiler Fix torch.profiler Schedule Function (Function Event only) (#125510) 2024-05-08 23:32:50 +00:00
quantization
signal Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
sparse Fix a bug in retrieving approximate bsr_dense_addmm kernel meta data (#124371) 2024-04-24 13:59:18 +00:00
special Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
testing [Inductor Intel GPU backend Upstream] Reuse inductor test for Intel GPU (PART 1) (#122866) 2024-05-09 00:51:35 +00:00
utils [BE]: Improve exception typing. Remove NOQAs (#125535) 2024-05-08 14:07:13 +00:00
xpu
__config__.py
__future__.py
__init__.py Add userbase library dir to windows dll search path (#125684) 2024-05-07 20:43:40 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py
_deploy.py
_guards.py [BE]: Try TCH autofixes on torch/ (#125536) 2024-05-05 23:13:59 +00:00
_jit_internal.py [BE]: Improve exception typing. Remove NOQAs (#125535) 2024-05-08 14:07:13 +00:00
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py SymInt-ify mem-efficient attention forward op signature (#125418) 2024-05-07 23:59:28 +00:00
_namedtensor_internals.py
_ops.py [Reland] Fakify script object inputs and attributes for non-strict ex… (#125490) 2024-05-04 02:39:42 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor.py Add testing and fix weights_only load for quantized types and nn.Parameters with python attrs (#124330) 2024-04-23 04:13:26 +00:00
_tensor_docs.py
_tensor_str.py
_torch_docs.py Updated docs to add the error case for torch.multinomial Issue#125388 (#125495) 2024-05-07 00:26:27 +00:00
_utils.py Fix to preserve tensor wrapper subclass dtype through multiprocessing serialization (#125615) 2024-05-07 14:35:48 +00:00
_utils_internal.py upload pt2 cprofile stats to manifold (#125162) 2024-04-30 15:05:01 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py Add testing and fix weights_only load for quantized types and nn.Parameters with python attrs (#124330) 2024-04-23 04:13:26 +00:00
abi-check.cpp
CMakeLists.txt
custom_class.h
custom_class_detail.h
extension.h
functional.py
hub.py
library.h Revert "Verify types in custom op schemas (#124520)" 2024-04-26 08:42:11 +00:00
library.py Revert "Add registration API for torch.compile-eager (#121387)" 2024-04-30 22:13:04 +00:00
overrides.py torch.mtia module for MTIA device backend (#123612) 2024-04-26 16:17:54 +00:00
py.typed
quasirandom.py
random.py [Doc] Update docstrings for torch/random.py (#125265) 2024-05-02 20:55:23 +00:00
README.txt
return_types.py
script.h
serialization.py Add MAP_SHARED option for torch.load(mmap=True) (#124889) 2024-04-30 15:02:19 +00:00
storage.py Add testing and fix weights_only load for quantized types and nn.Parameters with python attrs (#124330) 2024-04-23 04:13:26 +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.