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
_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
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
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
linalg
masked
monitor
mps
mtia
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
sparse
special
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
__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
_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
_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
_sources.py
_storage_docs.py
_streambase.py
_tensor.py [BE]: Update ruff to v0.4.4 (#125031) 2024-05-12 20:02:37 +00:00
_tensor_docs.py
_tensor_str.py
_torch_docs.py Fix four misspellings of "its" in documentation (#125681) 2024-05-13 18:14:09 +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
_VF.py
_vmap_internals.py
_weights_only_unpickler.py
abi-check.cpp
CMakeLists.txt
custom_class.h
custom_class_detail.h
extension.h
functional.py
hub.py
library.h
library.py Add registration API for torch.compile-eager (#121387) 2024-05-10 00:30:27 +00:00
overrides.py
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
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.