pytorch/torch
2024-05-13 18:14:09 +00:00
..
_awaits
_C [onnx.export] Avoid linear look up in env for exist_in_env (#124909) 2024-05-09 22:38:00 +00:00
_C_flatbuffer
_custom_op Move schema inference to torch._library (#124199) 2024-04-19 17:56:30 +00:00
_decomp Add a code comment about torch._check_is_size in tensor_split (#125292) 2024-05-02 02:25:38 +00:00
_dispatch
_dynamo Remove deprecated _aminmax operator (#125995) 2024-05-12 17:50:17 +00:00
_export [export] handle constant aliasing for export (#125509) 2024-05-10 00:14:37 +00:00
_functorch Keep track of ViewMeta with symbolic inputs. (#125876) 2024-05-12 01:41:06 +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 [inductor] Make load_mask() codegen determinstic (#126017) 2024-05-13 17:36:52 +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]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +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 Memoize local_scalar_dense calls, refactor all memos (#125623) 2024-05-11 21:12:35 +00:00
_vendor
amp make torch.amp.autocast more generic (#125103) 2024-05-08 12:13:26 +00:00
ao [BE]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +00:00
autograd Fix torch.profiler Schedule Function (Function Event only) (#125510) 2024-05-08 23:32:50 +00:00
backends [BE]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +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 [NT] Make NestedTensor register as having symbolic sizes/strides (#124687) 2024-05-13 16:50:25 +00:00
cuda Revert "[ROCm] amdsmi library integration (#119182)" 2024-05-09 21:18:21 +00:00
distributed [FSDP] Use generic device handle instead of cuda (#121620) 2024-05-13 18:07:08 +00:00
distributions [BE]: Update ruff to 0.4.1 (#124549) 2024-04-21 14:06:23 +00:00
export [RELAND] Switch default behavoir of export IR to be predispatch (#125860) 2024-05-10 17:36:53 +00:00
fft
func
futures
fx [BE]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +00:00
jit [BE][Easy] Remove usage of deprecated ast.Str, ast.Ellipsis and ast.NameConstant (#125912) 2024-05-10 17:35:35 +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 [BE]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +00:00
nested [NT] Make NestedTensor register as having symbolic sizes/strides (#124687) 2024-05-13 16:50:25 +00:00
nn [BE]: Try TCH autofixes on torch/ (#125536) 2024-05-05 23:13:59 +00:00
onnx [BE]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +00:00
optim [optim] add fused_adagrad support for CPU device (#124905) 2024-05-13 01:16:20 +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 [optim] add fused_adagrad support for CPU device (#124905) 2024-05-13 01:16:20 +00:00
utils use shutil.which in _find_cuda_home (#126060) 2024-05-13 17:38:17 +00:00
xpu [BE]: TRY002 - Ban raising vanilla exceptions (#124570) 2024-04-21 22:26:40 +00:00
__config__.py
__future__.py
__init__.py fixed typo in documentation (#125974) 2024-05-13 04:37:51 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py [BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261) 2024-04-17 19:29:34 +00:00
_deploy.py
_guards.py [BE]: Try TCH autofixes on torch/ (#125536) 2024-05-05 23:13:59 +00:00
_jit_internal.py [BE][Easy] Remove usage of deprecated ast.Str, ast.Ellipsis and ast.NameConstant (#125912) 2024-05-10 17:35:35 +00:00
_linalg_utils.py
_lobpcg.py
_lowrank.py Fix svd_lowrank parameter M (#122681) 2024-03-29 18:06:38 +00:00
_meta_registrations.py Add meta for _embedding_bag_dense_backward and _embedding_bag_per_sample_weights_backward (#125785) 2024-05-09 04:28:16 +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 Added a docstring for torch.Size.numel. (#124186) 2024-04-19 09:23:02 +00:00
_sources.py
_storage_docs.py
_streambase.py [BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261) 2024-04-17 19:29:34 +00:00
_tensor_docs.py Fix doc example of masked_scatter (#123664) 2024-04-09 22:15:12 +00:00
_tensor_str.py
_tensor.py [BE]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +00:00
_torch_docs.py Fix four misspellings of "its" in documentation (#125681) 2024-05-13 18:14:09 +00:00
_utils_internal.py upload pt2 cprofile stats to manifold (#125162) 2024-04-30 15:05:01 +00:00
_utils.py Fix to preserve tensor wrapper subclass dtype through multiprocessing serialization (#125615) 2024-05-07 14:35:48 +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 [rfc] opentelemetry in pytorch (#122999) 2024-04-21 15:20:21 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py
hub.py [BE]: TRY002 - Ban raising vanilla exceptions (#124570) 2024-04-21 22:26:40 +00:00
library.h Revert "Verify types in custom op schemas (#124520)" 2024-04-26 08:42:11 +00:00
library.py Add registration API for torch.compile-eager (#121387) 2024-05-10 00:30:27 +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 Enable UFMT on torch_version.py and types.py (#123131) 2024-04-09 15:03:17 +00:00
types.py Enable UFMT on torch_version.py and types.py (#123131) 2024-04-09 15:03:17 +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.