pytorch/torch
2024-07-02 18:24:02 +00:00
..
_awaits
_C Add warpSize to Device properties (#128449) 2024-07-01 09:13:32 +00:00
_C_flatbuffer Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
_custom_op Rename impl_abstract to register_fake, part 2/2 (#123938) 2024-06-14 14:37:24 +00:00
_decomp Add decomposition for slice_scatter (#123744) 2024-06-28 17:02:10 +00:00
_dispatch Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
_dynamo [Inductor] Make FlexAttention block_mask argument as tuple (#129831) 2024-07-02 17:18:33 +00:00
_export [ts-migration] support prim::SetAttr and fix prim::GetAttr (#129440) 2024-06-29 05:08:13 +00:00
_functorch [Ez][BE]: Enable new stable ruff rules (#129825) 2024-07-02 14:47:10 +00:00
_higher_order_ops [Inductor] Make FlexAttention block_mask argument as tuple (#129831) 2024-07-02 17:18:33 +00:00
_inductor [Inductor] Make FlexAttention block_mask argument as tuple (#129831) 2024-07-02 17:18:33 +00:00
_lazy Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_library [custom ops] Add unknown arg (#129614) 2024-07-02 16:10:14 +00:00
_logging [ts migration] add logging as part of torch logging system (#129405) 2024-06-27 00:20:20 +00:00
_numpy [Ez][BE]: Enable new stable ruff rules (#129825) 2024-07-02 14:47:10 +00:00
_prims [Traceable FSDP2] Add Dynamo support for run_with_rng_state HOP (#127247) 2024-06-28 01:04:49 +00:00
_prims_common Make are_strides_like_channels_last size oblivious (#129677) 2024-07-02 11:05:20 +00:00
_refs check boolean alpha and beta of Fake tensor impl for Tensor.addr (#129839) 2024-07-02 09:20:49 +00:00
_strobelight Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_subclasses [dynamic-shapes] Dont create symbol if .item() is a nan (#129881) 2024-07-02 16:51:56 +00:00
_vendor
amp Revert "[MPS] Add support for autocast in MPS (#99272)" 2024-07-02 12:29:51 +00:00
ao Fix for HistogramObserver (#129387) 2024-07-02 15:41:44 +00:00
autograd Fix test test_type_hints.py::TestTypeHints::test_doc_examples (#129829) 2024-07-01 13:28:37 +00:00
backends [cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343) 2024-06-30 19:22:16 +00:00
compiler Flip default value for mypy disallow_untyped_defs [4/11] (#127841) 2024-06-08 18:36:48 +00:00
contrib Flip default value for mypy disallow_untyped_defs [4/11] (#127841) 2024-06-08 18:36:48 +00:00
cpu [inductor][cpp] BF16 AMX micro-gemm support (#127195) 2024-06-21 07:21:47 +00:00
csrc Revert "[MPS] Add support for autocast in MPS (#99272)" 2024-07-02 12:29:51 +00:00
cuda Revert "[MPS] Add support for autocast in MPS (#99272)" 2024-07-02 12:29:51 +00:00
distributed [pipelining] [BE] Move pipeline_order validation to schedules.py (#129369) 2024-07-02 18:19:28 +00:00
distributions [BE]: Update mypy to 1.10.0 (#127717) 2024-06-13 15:57:13 +00:00
export [export] make with_effect mark op has_effect to prevent them from DCEed. (#129680) 2024-06-28 02:22:30 +00:00
fft
func
futures Flip default value for mypy disallow_untyped_defs [7/11] (#127844) 2024-06-08 18:49:45 +00:00
fx Fix typo in floordiv solver code that affects flipped relation (#129888) 2024-07-02 11:15:03 +00:00
jit [BE][Easy] replace import pathlib with from pathlib import Path (#129426) 2024-06-30 01:36:07 +00:00
legacy
lib
linalg
masked [BE] update type annotations for basic utilities in torch/__init__.py (#129001) 2024-06-24 18:04:38 +00:00
monitor
mps Add support in Python API for the recommended max working set size. (#128289) 2024-06-12 16:03:57 +00:00
mtia [MTIA] Fix synchronize API (#128714) 2024-06-17 21:58:46 +00:00
multiprocessing expose set_thread_name to Python and set thread names (#128448) 2024-06-13 16:38:23 +00:00
nested Fix layout for *_like() factories on NJTs (#129879) 2024-07-02 14:51:23 +00:00
nn [Inductor] Make FlexAttention block_mask argument as tuple (#129831) 2024-07-02 17:18:33 +00:00
onnx [BE] enable UFMT for torch/storage.py (#127706) 2024-06-27 23:16:24 +00:00
optim Add wraps back (#129933) 2024-07-02 18:24:02 +00:00
package [BE] enforce style for empty lines in import segments (#129751) 2024-06-29 14:15:24 +00:00
profiler [Profiler] Clean up use_mtia to follow standard use_device instead (#126284) 2024-06-18 21:01:03 +00:00
quantization Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
signal Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
sparse Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
special
testing [Ez][BE]: Enable new stable ruff rules (#129825) 2024-07-02 14:47:10 +00:00
utils [BE][Easy] Fix PYI001: unprefixed-type-param in torch/utils/data/datapipes (#129885) 2024-07-02 14:56:27 +00:00
xpu Flip default value for mypy disallow_untyped_defs [10/11] (#127847) 2024-06-08 18:50:06 +00:00
__config__.py Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
__future__.py
__init__.py Make sym_min/sym_max handle Numpy scalars (#129917) 2024-07-02 14:59:20 +00:00
_appdirs.py
_classes.py Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
_compile.py Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
_custom_ops.py Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
_deploy.py Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
_guards.py Evaluate symexprs on load path of cache not write (#128997) 2024-06-20 08:55:12 +00:00
_jit_internal.py [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
_linalg_utils.py [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
_lobpcg.py [BE] enable UFMT for top-level files torch/*.py (#127707) 2024-06-12 20:15:05 +00:00
_lowrank.py [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
_meta_registrations.py [cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343) 2024-06-30 19:22:16 +00:00
_namedtensor_internals.py Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_ops.py Torchbind call method + effects support (#128397) 2024-06-14 21:28:17 +00:00
_python_dispatcher.py Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_size_docs.py Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_sources.py Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_storage_docs.py Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_streambase.py Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_tensor_docs.py Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_tensor_str.py Flip default value for mypy disallow_untyped_defs [3/11] (#127840) 2024-06-08 18:28:01 +00:00
_tensor.py added type hints for __contains__ (#129653) 2024-06-30 11:49:11 +00:00
_torch_docs.py Fix test test_type_hints.py::TestTypeHints::test_doc_examples (#129829) 2024-07-01 13:28:37 +00:00
_utils_internal.py [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
_utils.py Remove dependency on private _compat_pickle in CPython (#129509) 2024-06-26 14:20:27 +00:00
_VF.py
_vmap_internals.py [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
_weights_only_unpickler.py Improve error message for weights_only load (#129705) 2024-06-28 19:36:31 +00:00
abi-check.cpp
CMakeLists.txt Remove outdated CMake code (#129851) 2024-07-02 00:40:37 +00:00
custom_class_detail.h Revert "[1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301)" 2024-06-15 01:58:20 +00:00
custom_class.h
extension.h
functional.py [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
hub.py [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
library.h Revert "[1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301)" 2024-06-15 01:58:20 +00:00
library.py [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
overrides.py [BE] enable UFMT for torch/nn/*.py (#128593) 2024-06-23 16:05:13 +00:00
py.typed
quasirandom.py [BE] enable UFMT for top-level files torch/*.py (#127707) 2024-06-12 20:15:05 +00:00
random.py [BE] enable UFMT for top-level files torch/*.py (#127707) 2024-06-12 20:15:05 +00:00
README.txt
return_types.py [BE] enable UFMT for top-level files torch/*.py (#127707) 2024-06-12 20:15:05 +00:00
script.h
serialization.py Fix test test_type_hints.py::TestTypeHints::test_doc_examples (#129829) 2024-07-01 13:28:37 +00:00
storage.py [BE] enable UFMT for torch/storage.py (#127706) 2024-06-27 23:16:24 +00:00
torch_version.py [BE] enable UFMT for top-level files torch/*.py (#127707) 2024-06-12 20:15:05 +00:00
types.py [BE] enable UFMT for torch/storage.py (#127706) 2024-06-27 23:16:24 +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.