pytorch/torch
2024-08-28 07:35:18 +00:00
..
_awaits Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_C Add compile time instruction count metric (#133834) 2024-08-27 23:29:02 +00:00
_C_flatbuffer
_custom_op Tighten torch.library.infer_schema input types (#130705) 2024-07-29 16:01:19 +00:00
_decomp Remove unnecessary expect_true in split_with_sizes (#133439) 2024-08-27 01:34:00 +00:00
_dispatch [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
_dynamo [dynamo][dicts] Support hasattr on dicts (#134590) 2024-08-28 07:35:18 +00:00
_export Back out "[pytorch][PR] [export] Schematize nn_module_stack serialization" (#134628) 2024-08-28 03:45:46 +00:00
_functorch Update partitioner's is_fusible heuristic to respect triton kernels (#134491) 2024-08-27 15:57:32 +00:00
_higher_order_ops [HOO] add hints_wrapper to support passing context hints (#132860) 2024-08-26 18:21:22 +00:00
_inductor [AOTI][Tooling][4/n] Add torch.save() for individual intermediate tensor (#133871) 2024-08-28 04:48:00 +00:00
_lazy Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_library [HOP] support generating schema for hop (#133521) 2024-08-21 17:34:21 +00:00
_logging [easy] rm duplicate definition for inductor in TORCH_LOGS documentation (#134480) 2024-08-27 20:15:10 +00:00
_numpy [test/torch_np] Fix usages of deprecated NumPy 2.0 APIs in numpy_tests (#131909) 2024-08-05 16:21:08 +00:00
_prims Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_prims_common Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_refs Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_strobelight [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_subclasses Move attention kernels back from fake_impls to meta_registrations (#134288) 2024-08-27 21:10:36 +00:00
_vendor
amp Revert "[MPS] Add support for autocast in MPS (#99272)" 2024-08-05 19:59:04 +00:00
ao Update pt2e numeric debugger to use node.meta["custom"] field (#134040) 2024-08-27 19:51:03 +00:00
autograd Move module_tracker to logging for confused hierarchy (#134467) 2024-08-26 19:39:08 +00:00
backends [sparse] Add cuSPARSELt as a backend (#128534) 2024-08-21 22:06:07 +00:00
compiler [dynamo] ensure polyfill function has the same signature as the original function in substitute_in_graph (#133813) 2024-08-22 16:38:06 +00:00
contrib
cpu [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
csrc [AOTI][Tooling][4/n] Add torch.save() for individual intermediate tensor (#133871) 2024-08-28 04:48:00 +00:00
cuda Fix partially initialized module error (#134019) 2024-08-20 22:20:02 +00:00
distributed [dtensor] add test for local_map decorator (#127752) 2024-08-27 18:22:23 +00:00
distributions Add correct typing annotations to rsample() for all distributions (#133516) 2024-08-18 20:31:54 +00:00
export [export] don't duck size for DIM.AUTO (#134486) 2024-08-27 23:00:26 +00:00
fft
func
futures
fx [export] enumerate unsupported sympy.Functions (#134271) (#134598) 2024-08-28 00:34:38 +00:00
jit [BE]: Update ruff to 0.6.0 (#133609) 2024-08-16 14:11:01 +00:00
legacy
lib
linalg Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
masked Revert "[BE]: Update Typeguard to TypeIs for better type inference (#133814)" 2024-08-21 16:13:34 +00:00
monitor [pytorch][counters] Pybind for WaitCounter (#132357) 2024-08-02 16:08:10 +00:00
mps [BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769) 2024-08-04 10:24:09 +00:00
mtia Improve error msg on _lazy_init() error (#134159) 2024-08-22 15:10:50 +00:00
multiprocessing Revert "Allow mp.start_processes to create processes in parallel (#133707)" 2024-08-26 17:31:10 +00:00
nested [BE] typing for decorators - fx/_compatibility (#134054) 2024-08-26 04:00:27 +00:00
nn [Doc] Fix rendering of the unicode characters (#134597) 2024-08-27 19:52:46 +00:00
onnx [BE] typing for decorators - fx/_compatibility (#134054) 2024-08-26 04:00:27 +00:00
optim Avoid Host & Device Sync In LR Scheduler (#133663) 2024-08-22 03:52:43 +00:00
package [torch/package_importer] add compatibility name mapping (#134376) 2024-08-25 19:34:46 +00:00
profiler [Profiler/CPU] Add Test for Dynamic Activity Toggling [4/n] (#134149) 2024-08-23 01:13:42 +00:00
quantization [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
signal Fix docstring for torch.signal.windows.nuttall (#134512) 2024-08-26 22:51:43 +00:00
sparse [sparse] Update cuSPARSELt to v0.6.2 (#134022) 2024-08-23 19:34:53 +00:00
special
testing Move attention kernels back from fake_impls to meta_registrations (#134288) 2024-08-27 21:10:36 +00:00
utils [export] enumerate unsupported sympy.Functions (#134271) (#134598) 2024-08-28 00:34:38 +00:00
xpu [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
__config__.py
__future__.py
__init__.py Revert "[BE]: Update Typeguard to TypeIs for better type inference (#133814)" 2024-08-21 16:13:34 +00:00
_appdirs.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_classes.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
_compile.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_custom_ops.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_deploy.py Remove unused imported names in python files (#134438) 2024-08-27 20:44:04 +00:00
_guards.py [PT2/Profiler] Add Context Info to Torch-Compiled Regions (#132765) 2024-08-27 04:55:04 +00:00
_jit_internal.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_linalg_utils.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_lobpcg.py [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
_lowrank.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_meta_registrations.py Move attention kernels back from fake_impls to meta_registrations (#134288) 2024-08-27 21:10:36 +00:00
_namedtensor_internals.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_ops.py [hop] ban creating hop by directly instantiating HigherOrderOperator. (#133645) 2024-08-23 17:28:02 +00:00
_python_dispatcher.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
_size_docs.py
_sources.py
_storage_docs.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_streambase.py
_tensor_docs.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_tensor_str.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_tensor.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_torch_docs.py [docs] improve torch.stack example code to be reproducible (#133857) 2024-08-21 14:07:02 +00:00
_utils_internal.py Create a JustknobConfig for use in config (#134161) 2024-08-27 16:07:33 +00:00
_utils.py Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-08-07 00:05:20 +00:00
_VF.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
_vmap_internals.py
_weights_only_unpickler.py Clarify error messages for NEWOBJ and BUILD in weights_only unpickler (#134346) 2024-08-27 14:45:39 +00:00
abi-check.cpp
CMakeLists.txt [split build] fix distributed problems (#134502) 2024-08-27 21:12:58 +00:00
custom_class_detail.h Enable clang-tidy coverage on torch/*.h (#133422) 2024-08-15 18:52:08 +00:00
custom_class.h Enable clang-tidy coverage on torch/*.h (#133422) 2024-08-15 18:52:08 +00:00
extension.h
functional.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
hub.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
library.h Enable clang-tidy coverage on torch/*.h (#133422) 2024-08-15 18:52:08 +00:00
library.py [custom_ops] torch.library.{custom_op, register_kernel} disable Dynamo (#133125) 2024-08-12 18:29:18 +00:00
overrides.py Add new ops wrapped_linear_prepack and wrapped_quantized_linear_prepacked (#134232) 2024-08-23 04:54:26 +00:00
py.typed
quasirandom.py [BE] Format uncategorized Python files with ruff format (#132576) 2024-08-04 17:13:31 +00:00
random.py
README.txt
return_types.py
script.h
serialization.py Make torch.serialization.set_default_mmap_options usable as a context manager (#134371) 2024-08-27 14:45:29 +00:00
storage.py Fix public API tests (#131386) 2024-07-30 18:42:54 +00:00
torch_version.py Add mypy typing to torch_version.py (#131447) 2024-07-23 17:31:07 +00:00
types.py FakeTensor cache SymInt support (#127596) 2024-07-21 19:26:38 +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.