pytorch/torch
2024-04-23 06:26:11 +00:00
..
_awaits
_C rename ort to maia (#123265) 2024-04-23 00:33:25 +00:00
_C_flatbuffer
_custom_op Move schema inference to torch._library (#124199) 2024-04-19 17:56:30 +00:00
_decomp Add out wrappers to some decompositions (#115437) 2024-04-23 06:26:11 +00:00
_dispatch
_dynamo Add support for capturing tensors with score_mod (#124444) 2024-04-23 06:20:13 +00:00
_export [export] handle Dim.lower = 0, 1 for ep.run_decompositions() (#123602) 2024-04-19 21:29:36 +00:00
_functorch Dynamo x autograd.Function supports non-{Tensor, symnode, constant} inputs (#124360) 2024-04-22 23:32:54 +00:00
_higher_order_ops Add support for capturing tensors with score_mod (#124444) 2024-04-23 06:20:13 +00:00
_inductor Add support for capturing tensors with score_mod (#124444) 2024-04-23 06:20:13 +00:00
_lazy
_library [BE]: FURB142 - Remove set mutations. Use set update (#124551) 2024-04-21 14:12:33 +00:00
_logging
_numpy [BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261) 2024-04-17 19:29:34 +00:00
_prims [BE]: Update ruff to 0.4.1 (#124549) 2024-04-21 14:06:23 +00:00
_prims_common Add out wrappers to some decompositions (#115437) 2024-04-23 06:26:11 +00:00
_refs [BE]: Update ruff to 0.4.1 (#124549) 2024-04-21 14:06:23 +00:00
_subclasses [BE]: TRY002 - Ban raising vanilla exceptions (#124570) 2024-04-21 22:26:40 +00:00
_vendor
amp [BE][Ez]: Fix minor potential perf regression from #123960 (#124013) 2024-04-15 16:51:45 +00:00
ao Add Matmul recipe into x86_inductor_quantizer (#122776) 2024-04-23 00:25:41 +00:00
autograd [Profiler] Unify the device(CUDA, XPU, PrivateUse1) in torch profiler post processing (#123247) 2024-04-22 01:26:55 +00:00
backends preferred blas library; cublaslt gemm implementation (#122106) 2024-04-22 15:38:22 +00:00
compiler
contrib
cpu
csrc Add missing aten::sort.any op for assistant lm models (#123982) 2024-04-23 01:35:07 +00:00
cuda [BE]: TRY002 - Ban raising vanilla exceptions (#124570) 2024-04-21 22:26:40 +00:00
distributed [tp] add kwargs support to prepare_module_input (#124114) 2024-04-22 21:46:31 +00:00
distributions [BE]: Update ruff to 0.4.1 (#124549) 2024-04-21 14:06:23 +00:00
export [export] handle Dim.lower = 0, 1 for ep.run_decompositions() (#123602) 2024-04-19 21:29:36 +00:00
fft
func
futures
fx [sym_shapes][perf] Cache ShapeEnv constrain_symbol_range calls (#124610) 2024-04-22 21:49:08 +00:00
jit [BE]: TRY002 - Ban raising vanilla exceptions (#124570) 2024-04-21 22:26:40 +00:00
legacy
lib [codemod][lowrisk] Remove unused exception parameter from caffe2/caffe2/image/image_input_op.h (#123056) 2024-04-04 17:24:43 +00:00
linalg
masked
monitor
mps
multiprocessing
nested [NT] Fix typo in declared strides variable (#123856) 2024-04-13 19:55:57 +00:00
nn [BE]: Update ruff to 0.4.1 (#124549) 2024-04-21 14:06:23 +00:00
onnx Do not import transformers when import torch._dynamo (#124634) 2024-04-23 00:25:20 +00:00
optim Better Error Message in ChainedScheduler and SequentialLR (#121633) 2024-04-19 13:37:41 +00:00
package [BE]: TRY002 - Ban raising vanilla exceptions (#124570) 2024-04-21 22:26:40 +00:00
profiler [Profiler] Unify the device(CUDA, XPU, PrivateUse1) in torch profiler post processing (#123247) 2024-04-22 01:26:55 +00:00
quantization
signal
sparse [sparse] Add fast semi-structured spasification kernels (#122350) 2024-04-19 13:31:58 +00:00
special
testing preferred blas library; cublaslt gemm implementation (#122106) 2024-04-22 15:38:22 +00:00
utils [EZ] Update pillow to 10.3.0 (#124614) 2024-04-23 03:22:23 +00:00
xpu [BE]: TRY002 - Ban raising vanilla exceptions (#124570) 2024-04-21 22:26:40 +00:00
__config__.py
__future__.py
__init__.py Revert "torch.mtia module for MTIA device backend (#123612)" 2024-04-19 22:44:26 +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] enable ruff rule RSE and remove useless parentheses in raise statements (#124261) 2024-04-17 19:29:34 +00:00
_jit_internal.py [BE]: TRY002 - Ban raising vanilla exceptions (#124570) 2024-04-21 22:26:40 +00:00
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py Add meta function for torch.histc (#124548) 2024-04-23 00:24:59 +00:00
_namedtensor_internals.py
_ops.py Support aot_export torchbind op (#123370) 2024-04-19 17:17:27 +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 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_docs.py
_utils_internal.py rename sl to strobelight (#124455) 2024-04-19 22:50:13 +00:00
_utils.py Revert "torch.mtia module for MTIA device backend (#123612)" 2024-04-19 22:44:26 +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 rename ort to maia (#123265) 2024-04-23 00:33:25 +00:00
library.py Change register_autograd to reflect ordering of setup_context and backward (#124403) 2024-04-19 17:56:30 +00:00
overrides.py rename ort to maia (#123265) 2024-04-23 00:33:25 +00:00
py.typed
quasirandom.py
random.py
README.txt
return_types.py
script.h
serialization.py
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.